m-chrzan.xyz
aboutsummaryrefslogtreecommitdiff
path: root/src/test/java/pl/edu/mimuw/cloudatlas/agent/EventBusTest.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/test/java/pl/edu/mimuw/cloudatlas/agent/EventBusTest.java')
-rw-r--r--src/test/java/pl/edu/mimuw/cloudatlas/agent/EventBusTest.java71
1 files changed, 27 insertions, 44 deletions
diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/agent/EventBusTest.java b/src/test/java/pl/edu/mimuw/cloudatlas/agent/EventBusTest.java
index f3df887..34b80bb 100644
--- a/src/test/java/pl/edu/mimuw/cloudatlas/agent/EventBusTest.java
+++ b/src/test/java/pl/edu/mimuw/cloudatlas/agent/EventBusTest.java
@@ -7,6 +7,7 @@ import org.junit.runners.JUnit4;
import pl.edu.mimuw.cloudatlas.agent.messages.AgentMessage;
import pl.edu.mimuw.cloudatlas.agent.messages.TimerSchedulerMessage;
import pl.edu.mimuw.cloudatlas.agent.modules.Module;
+import pl.edu.mimuw.cloudatlas.agent.modules.ModuleType;
import pl.edu.mimuw.cloudatlas.agent.modules.TimerScheduledTask;
import java.util.ArrayList;
@@ -18,29 +19,25 @@ public class EventBusTest {
public abstract class MessageCounterModule extends Module {
public int counter = 0;
- MessageCounterModule(AgentMessage.AgentModule moduleType) {
+ MessageCounterModule(ModuleType moduleType) {
super(moduleType);
}
}
- public HashMap<AgentMessage.AgentModule, Module> initializeTwoModules() {
- HashMap<AgentMessage.AgentModule, Module> modules = new HashMap<AgentMessage.AgentModule, Module>();
- modules.put(AgentMessage.AgentModule.RMI, new MessageCounterModule(AgentMessage.AgentModule.RMI) {
+ /*
+ public HashMap<ModuleType, Module> initializeTwoModules() {
+ HashMap<ModuleType, Module> modules = new HashMap<ModuleType, Module>();
+ modules.put(ModuleType.RMI, new MessageCounterModule(ModuleType.RMI) {
@Override
public void handle(AgentMessage event) throws InterruptedException {
System.out.println("Module 1 handle called");
// TODO correct message subclass
- sendMessage(new AgentMessage("1", AgentMessage.AgentModule.UDP) {
- @Override
- public AgentModule getCorrectMessageType() {
- return AgentModule.UDP;
- }
- });
+ sendMessage(new AgentMessage("1", ModuleType.UDP) {});
counter ++;
}
});
- modules.put(AgentMessage.AgentModule.UDP, new MessageCounterModule(AgentMessage.AgentModule.UDP) {
+ modules.put(ModuleType.UDP, new MessageCounterModule(ModuleType.UDP) {
@Override
public void handle(AgentMessage event) {
System.out.println("Module 2 handle called");
@@ -51,10 +48,10 @@ public class EventBusTest {
return modules;
}
- public HashMap<AgentMessage.AgentModule, Module> initializeModule() {
- HashMap<AgentMessage.AgentModule, Module> modules = new HashMap<AgentMessage.AgentModule, Module>();
+ public HashMap<ModuleType, Module> initializeModule() {
+ HashMap<ModuleType, Module> modules = new HashMap<ModuleType, Module>();
- modules.put(AgentMessage.AgentModule.RMI, new MessageCounterModule(AgentMessage.AgentModule.RMI) {
+ modules.put(ModuleType.RMI, new MessageCounterModule(ModuleType.RMI) {
@Override
public void handle(AgentMessage event) {
System.out.println("Module 1 handle called");
@@ -69,55 +66,45 @@ public class EventBusTest {
@Ignore
// TODO correct message subclass
public void messageModule() throws InterruptedException {
- HashMap<AgentMessage.AgentModule, Module> modules = initializeModule();
- HashMap<AgentMessage.AgentModule, Executor> executors = Agent.initializeExecutors(modules);
+ HashMap<ModuleType, Module> modules = initializeModule();
+ HashMap<ModuleType, Executor> executors = Agent.initializeExecutors(modules);
ArrayList<Thread> executorThreads = Agent.initializeExecutorThreads(executors);
EventBus eventBus = new EventBus(executors);
Thread eventBusThread = new Thread(eventBus);
eventBusThread.start();
- eventBus.addMessage(new AgentMessage("0", AgentMessage.AgentModule.RMI) {
- @Override
- public AgentModule getCorrectMessageType() {
- return AgentModule.RMI;
- }
- });
+ eventBus.addMessage(new AgentMessage("0", ModuleType.RMI) {});
Thread.sleep(1000);
eventBusThread.interrupt();
Agent.closeExecutors(executorThreads);
- assertEquals(1, ((MessageCounterModule) modules.get(AgentMessage.AgentModule.RMI)).counter);
+ assertEquals(1, ((MessageCounterModule) modules.get(ModuleType.RMI)).counter);
}
@Test
@Ignore
// TODO correct message subclass
public void messagingBetweenModules() throws InterruptedException {
- HashMap<AgentMessage.AgentModule, Module> modules = initializeTwoModules();
- HashMap<AgentMessage.AgentModule, Executor> executors = Agent.initializeExecutors(modules);
+ HashMap<ModuleType, Module> modules = initializeTwoModules();
+ HashMap<ModuleType, Executor> executors = Agent.initializeExecutors(modules);
ArrayList<Thread> executorThreads = Agent.initializeExecutorThreads(executors);
EventBus eventBus = new EventBus(executors);
Thread eventBusThread = new Thread(eventBus);
eventBusThread.start();
- eventBus.addMessage(new AgentMessage("0", AgentMessage.AgentModule.RMI) {
- @Override
- public AgentModule getCorrectMessageType() {
- return AgentModule.RMI;
- }
- });
+ eventBus.addMessage(new AgentMessage("0", ModuleType.RMI) {});
Thread.sleep(1000);
eventBusThread.interrupt();
Agent.closeExecutors(executorThreads);
- assertEquals(1, ((MessageCounterModule) modules.get(AgentMessage.AgentModule.RMI)).counter);
- assertEquals(1, ((MessageCounterModule) modules.get(AgentMessage.AgentModule.UDP)).counter);
+ assertEquals(1, ((MessageCounterModule) modules.get(ModuleType.RMI)).counter);
+ assertEquals(1, ((MessageCounterModule) modules.get(ModuleType.UDP)).counter);
}
@Test
public void sendWrongMessageType1() throws InterruptedException {
- HashMap<AgentMessage.AgentModule, Module> modules = initializeModule();
- HashMap<AgentMessage.AgentModule, Executor> executors = Agent.initializeExecutors(modules);
+ HashMap<ModuleType, Module> modules = initializeModule();
+ HashMap<ModuleType, Executor> executors = Agent.initializeExecutors(modules);
ArrayList<Thread> executorThreads = Agent.initializeExecutorThreads(executors);
EventBus eventBus = new EventBus(executors);
Thread eventBusThread = new Thread(eventBus);
@@ -127,7 +114,7 @@ public class EventBusTest {
try {
eventBus.addMessage(new TimerSchedulerMessage(
"0",
- AgentMessage.AgentModule.RMI,
+ ModuleType.RMI,
System.currentTimeMillis() / 1000L,
"1",
10,
@@ -154,8 +141,8 @@ public class EventBusTest {
@Test
public void sendWrongMessageType2() throws InterruptedException {
- HashMap<AgentMessage.AgentModule, Module> modules = initializeModule();
- HashMap<AgentMessage.AgentModule, Executor> executors = Agent.initializeExecutors(modules);
+ HashMap<ModuleType, Module> modules = initializeModule();
+ HashMap<ModuleType, Executor> executors = Agent.initializeExecutors(modules);
ArrayList<Thread> executorThreads = Agent.initializeExecutorThreads(executors);
EventBus eventBus = new EventBus(executors);
Thread eventBusThread = new Thread(eventBus);
@@ -163,12 +150,7 @@ public class EventBusTest {
eventBusThread.start();
try {
- eventBus.addMessage(new AgentMessage("0", AgentMessage.AgentModule.RMI) {
- @Override
- public AgentModule getCorrectMessageType() {
- return AgentModule.QUERY;
- }
- });
+ eventBus.addMessage(new AgentMessage("0", ModuleType.RMI) {});
Thread.sleep(1000);
} catch (AssertionError e) {
System.out.println("Wrong timer-scheduler message type error caught");
@@ -182,4 +164,5 @@ public class EventBusTest {
Assert.fail("Routing not detected as faulty");
}
}
+ */
}