PullMessagesUtilityTest.java

/*
 *        JacORB - a free Java ORB
 *
 *   Copyright (C) 1999-2014 Gerald Brose / The JacORB Team.
 *
 *   This library is free software; you can redistribute it and/or
 *   modify it under the terms of the GNU Library General Public
 *   License as published by the Free Software Foundation; either
 *   version 2 of the License, or (at your option) any later version.
 *
 *   This library is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *   Library General Public License for more details.
 *
 *   You should have received a copy of the GNU Library General Public
 *   License along with this library; if not, write to the Free
 *   Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */

package org.jacorb.test.notification.servant;

import java.util.concurrent.ScheduledFuture;
import org.easymock.AbstractMatcher;
import org.easymock.MockControl;
import org.jacorb.notification.engine.TaskProcessor;
import org.jacorb.notification.interfaces.MessageSupplier;
import org.jacorb.notification.servant.PullMessagesUtility;
import org.junit.Before;
import org.junit.Test;

public class PullMessagesUtilityTest
{
    private PullMessagesUtility objectUnderTest_;

    private MockControl controlTaskProcessor_;

    private TaskProcessor mockTaskProcessor_;

    private MockControl controlMessageSupplier_;

    private MessageSupplier mockMessageSupplier_;

    private ScheduledFuture mockScheduledFuture_;

    private MockControl controlScheduledFuture_;

    @Before
    public void setUp() throws Exception
    {
        controlScheduledFuture_ = MockControl.createControl(ScheduledFuture.class);
        mockScheduledFuture_ = (ScheduledFuture) controlScheduledFuture_.getMock();
        controlTaskProcessor_ = MockControl.createControl(TaskProcessor.class);
        mockTaskProcessor_ = (TaskProcessor) controlTaskProcessor_.getMock();
        controlMessageSupplier_ = MockControl.createControl(MessageSupplier.class);
        mockMessageSupplier_ = (MessageSupplier) controlMessageSupplier_.getMock();
        objectUnderTest_ = new PullMessagesUtility(mockTaskProcessor_, mockMessageSupplier_);
    }

    private void replayAll()
    {
        controlScheduledFuture_.replay();
        controlTaskProcessor_.replay();
        controlMessageSupplier_.replay();
    }

    private void verifyAll()
    {
        controlScheduledFuture_.verify();
        controlTaskProcessor_.verify();
        controlMessageSupplier_.verify();
    }

    @Test
    public void testIllegalArgument()
    {
        replayAll();
        try
        {
            objectUnderTest_.startTask(0);
        } catch (IllegalArgumentException e)
        {
            // expected
        }

        try
        {
            objectUnderTest_.startTask(-1);
        } catch (IllegalArgumentException e)
        {
            // expected
        }
        verifyAll();
    }

    @Test
    public void testStartTaskRegistersTask()
    {
        mockTaskProcessor_.executeTaskPeriodically(1000, null, true);
        controlTaskProcessor_.setMatcher(new AbstractMatcher()
        {
            public boolean matches(Object[] expected, Object[] actual)
            {
                return expected[0].equals(actual[0]);
            }
        });
        controlTaskProcessor_.setReturnValue(mockScheduledFuture_);
        replayAll();
        objectUnderTest_.startTask(1000);
        verifyAll();
    }

    @Test
    public void testMultipleStartsAreIgnored()
    {
        mockTaskProcessor_.executeTaskPeriodically(0, null, false);
        controlTaskProcessor_.setMatcher(MockControl.ALWAYS_MATCHER);
        controlTaskProcessor_.setReturnValue(mockScheduledFuture_);
        replayAll();

        objectUnderTest_.startTask(2000);
        objectUnderTest_.startTask(2000);

        verifyAll();
    }

    @Test
    public void testStopNotStartedTask()
    {
        replayAll();
        objectUnderTest_.stopTask();
        verifyAll();
    }

    @Test
    public void testStopTask()
    {
        mockTaskProcessor_.executeTaskPeriodically(0, null, false);
        controlTaskProcessor_.setMatcher(MockControl.ALWAYS_MATCHER);
        controlTaskProcessor_.setReturnValue(mockScheduledFuture_);
        controlScheduledFuture_.expectAndReturn(mockScheduledFuture_.cancel(true), true);
        replayAll();

        objectUnderTest_.startTask(2000);
        objectUnderTest_.stopTask();

        verifyAll();
    }

    @Test
    public void testRestartNotStarted()
    {
        replayAll();
        try
        {
            objectUnderTest_.restartTask(1000);
        } catch (IllegalStateException e)
        {
            // expected
        }
        verifyAll();
    }

    @Test
    public void testRestartStarted()
    {
        mockTaskProcessor_.executeTaskPeriodically(1000, null, true);
        controlTaskProcessor_.setMatcher(new AbstractMatcher()
        {
            public boolean matches(Object[] expected, Object[] actual)
            {
                return expected[0].equals(actual[0]);
            }
        });
        controlTaskProcessor_.setReturnValue(mockScheduledFuture_);
        mockScheduledFuture_.cancel(true);
        controlScheduledFuture_.setReturnValue(true);

        mockTaskProcessor_.executeTaskPeriodically(2000, null, true);
        controlTaskProcessor_.setReturnValue(mockScheduledFuture_);

        replayAll();
        objectUnderTest_.startTask(1000);
        objectUnderTest_.restartTask(2000);
        verifyAll();
    }
}