/*
* Copyright (c) MuleSoft, Inc. All rights reserved. http://www.mulesoft.com
* The software in this package is published under the terms of the CPAL v1.0
* license, a copy of which has been included with this distribution in the
* LICENSE.txt file.
*/
package org.mule.transport;
import static org.junit.Assert.fail;
import static org.mockito.Mockito.mock;
import org.mule.MessageExchangePattern;
import org.mule.api.construct.FlowConstruct;
import org.mule.api.transport.Connector;
import org.mule.tck.junit4.AbstractMuleContextTestCase;
import org.mule.tck.testmodels.mule.TestMessageReceiver;
import org.mule.util.concurrent.Latch;
import java.util.concurrent.TimeUnit;
import javax.resource.spi.work.Work;
import org.junit.Test;
import org.mockito.Answers;
public class MessageReceiverWorkManagerTestCase extends AbstractMuleContextTestCase
{
@Test
public void workManagerRecreatedCorrectlyAfterRestart() throws Exception
{
final Latch workExecutedLatch = new Latch();
AbstractMessageReceiver receiver = createMessageReceiver();
Connector receiverConnector = receiver.getConnector();
receiverConnector.stop();
receiverConnector.start();
receiver.getWorkManager().scheduleWork(new Work()
{
@Override
public void release()
{
}
@Override
public void run()
{
workExecutedLatch.release();
}
});
if (!workExecutedLatch.await(1000, TimeUnit.MILLISECONDS))
{
fail("Work should be executed and it was not");
}
}
private AbstractMessageReceiver createMessageReceiver() throws Exception
{
return new TestMessageReceiver(getTestConnector(), mock(FlowConstruct.class, Answers.RETURNS_DEEP_STUBS.get()), getTestInboundEndpoint(MessageExchangePattern.ONE_WAY));
}
}