diff options
Diffstat (limited to 'src/test/java/pl')
-rw-r--r-- | src/test/java/pl/edu/mimuw/cloudatlas/agent/ExecutorTest.java | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/agent/ExecutorTest.java b/src/test/java/pl/edu/mimuw/cloudatlas/agent/ExecutorTest.java new file mode 100644 index 0000000..445a5b4 --- /dev/null +++ b/src/test/java/pl/edu/mimuw/cloudatlas/agent/ExecutorTest.java @@ -0,0 +1,54 @@ +package pl.edu.mimuw.cloudatlas.agent; + +import org.junit.Test; +import static org.junit.Assert.*; + +import pl.edu.mimuw.cloudatlas.agent.message.AgentMessage; +import pl.edu.mimuw.cloudatlas.agent.message.AgentMessage.AgentModule; + +public class ExecutorTest { + public class MessageCounterModule extends Module { + public int counter = 0; + public void handle(AgentMessage m) { + counter++; + } + } + + @Test + public void testDoesntExecuteWhenNoMessages() throws Exception { + MessageCounterModule module = new MessageCounterModule(); + Executor executor = new Executor(module); + Thread thread = new Thread(executor); + thread.start(); + Thread.sleep(100); + thread.interrupt(); + assertEquals(0, module.counter); + } + + @Test + public void testExecutesHandlerOnce() throws Exception { + MessageCounterModule module = new MessageCounterModule(); + Executor executor = new Executor(module); + executor.addMessage(new AgentMessage("", AgentModule.UDP, 0) {}); + Thread thread = new Thread(executor); + thread.start(); + Thread.sleep(100); + thread.interrupt(); + assertEquals(1, module.counter); + } + + @Test + public void testExecutesHandlerMultipleTimes() throws Exception { + MessageCounterModule module = new MessageCounterModule(); + Executor executor = new Executor(module); + executor.addMessage(new AgentMessage("", AgentModule.UDP, 0) {}); + executor.addMessage(new AgentMessage("", AgentModule.UDP, 0) {}); + Thread thread = new Thread(executor); + thread.start(); + Thread.sleep(100); + executor.addMessage(new AgentMessage("", AgentModule.UDP, 0) {}); + Thread.sleep(100); + thread.interrupt(); + assertEquals(3, module.counter); + } +} |