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}