From c823d047ee822e48cbdf12bed796389c2a3e25cd Mon Sep 17 00:00:00 2001 From: Marcin Chrzanowski Date: Tue, 31 Dec 2019 13:52:33 +0100 Subject: Test Remik module's state request --- .../mimuw/cloudatlas/agent/modules/RemikTest.java | 61 ++++++++++++++++++++++ 1 file changed, 61 insertions(+) create mode 100644 src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/RemikTest.java (limited to 'src/test/java/pl/edu/mimuw/cloudatlas') diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/RemikTest.java b/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/RemikTest.java new file mode 100644 index 0000000..8ac8bed --- /dev/null +++ b/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/RemikTest.java @@ -0,0 +1,61 @@ +package pl.edu.mimuw.cloudatlas.agent.modules; + +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.TimeUnit; +import java.util.HashMap; + +import pl.edu.mimuw.cloudatlas.agent.messages.AgentMessage; +import pl.edu.mimuw.cloudatlas.agent.messages.GetStateMessage; +import pl.edu.mimuw.cloudatlas.agent.messages.RequestStateMessage; +import pl.edu.mimuw.cloudatlas.agent.messages.ResponseMessage; +import pl.edu.mimuw.cloudatlas.agent.messages.StanikMessage; +import pl.edu.mimuw.cloudatlas.agent.messages.StateMessage; +import pl.edu.mimuw.cloudatlas.agent.MockExecutor; +import pl.edu.mimuw.cloudatlas.model.ValueTime; +import pl.edu.mimuw.cloudatlas.model.ZMI; + +import org.junit.Before; +import org.junit.Test; +import static org.junit.Assert.*; + +public class RemikTest { + private Remik remik; + private MockExecutor executor; + + @Before + public void setupLocals() { + remik = new Remik(); + executor = new MockExecutor(remik); + } + + @Test + public void asksForStateOnStateRequest() throws Exception { + CompletableFuture future = new CompletableFuture(); + RequestStateMessage message = new RequestStateMessage("", 0, future); + remik.handleTyped(message); + AgentMessage receivedMessage = (AgentMessage) executor.messagesToPass.take(); + assertEquals(ModuleType.STATE, receivedMessage.getDestinationModule()); + StanikMessage stanikMessage = (StanikMessage) receivedMessage; + assertEquals(StanikMessage.Type.GET_STATE, stanikMessage.getType()); + GetStateMessage getStateMessage = (GetStateMessage) stanikMessage; + assertEquals(ModuleType.RMI, getStateMessage.getRequestingModule()); + } + + @Test + public void completesFutureOnReceivedState() throws Exception { + CompletableFuture future = new CompletableFuture(); + RequestStateMessage message = new RequestStateMessage("", 0, future); + remik.handleTyped(message); + + ZMI zmi = new ZMI(); + zmi.getAttributes().add("timestamp", new ValueTime(42l)); + StateMessage response = new StateMessage("", ModuleType.RMI, 0, 0, zmi, new HashMap()); + remik.handleTyped(response); + + ResponseMessage passedResponse = future.get(100, TimeUnit.MILLISECONDS); + assertNotNull(passedResponse); + assertEquals(ResponseMessage.Type.STATE, passedResponse.getType()); + StateMessage stateMessage = (StateMessage) passedResponse; + assertEquals(new ValueTime(42l), stateMessage.getZMI().getAttributes().get("timestamp")); + } +} -- cgit v1.2.3