001/* 002 * #%L 003 * Netarchivesuite - common - test 004 * %% 005 * Copyright (C) 2005 - 2014 The Royal Danish Library, the Danish State and University Library, 006 * the National Library of France and the Austrian National Library. 007 * %% 008 * This program is free software: you can redistribute it and/or modify 009 * it under the terms of the GNU Lesser General Public License as 010 * published by the Free Software Foundation, either version 2.1 of the 011 * License, or (at your option) any later version. 012 * 013 * This program is distributed in the hope that it will be useful, 014 * but WITHOUT ANY WARRANTY; without even the implied warranty of 015 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 016 * GNU General Lesser Public License for more details. 017 * 018 * You should have received a copy of the GNU General Lesser Public 019 * License along with this program. If not, see 020 * <http://www.gnu.org/licenses/lgpl-2.1.html>. 021 * #L% 022 */ 023package dk.netarkivet.testutils.preconfigured; 024 025import dk.netarkivet.common.CommonSettings; 026import dk.netarkivet.common.distribute.JMSConnection; 027import dk.netarkivet.common.distribute.JMSConnectionFactory; 028import dk.netarkivet.common.distribute.JMSConnectionMockupMQ; 029import dk.netarkivet.common.utils.Settings; 030 031public class MockupJMS implements TestConfigurationIF { 032 private String originalClass; 033 034 @Override 035 public void setUp() { 036 originalClass = Settings.get(CommonSettings.JMS_BROKER_CLASS); 037 Settings.set(CommonSettings.JMS_BROKER_CLASS, JMSConnectionMockupMQ.class.getName()); 038 JMSConnectionFactory.getInstance().cleanup(); 039 JMSConnectionMockupMQ.clearTestQueues(); 040 } 041 042 @Override 043 public void tearDown() { 044 JMSConnectionMockupMQ.clearTestQueues(); 045 try { 046 JMSConnectionFactory.getInstance().cleanup(); 047 } catch (Exception e) { 048 // just ignore it 049 } 050 Settings.set(CommonSettings.JMS_BROKER_CLASS, originalClass); 051 } 052 053 public JMSConnection getJMSConnection() { 054 return JMSConnectionFactory.getInstance(); 055 } 056}