Refactor agent configs nicely

1 files changed, 5 insertions, 0 deletions

diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/EventBus.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/EventBus.java
index a6d3b2d..a9b25ed 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/EventBus.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/EventBus.java
@@ -22,6 +22,7 @@ public class EventBus implements Runnable {
 
     // Allows for testing with a mock EventBus
     protected EventBus() {
+        this.events = new LinkedBlockingQueue<AgentMessage>();
     }
 
     EventBus(HashMap<ModuleType, Executor> executors) {
@@ -30,6 +31,10 @@ public class EventBus implements Runnable {
         this.events = new LinkedBlockingQueue<AgentMessage>();
     }
 
+    public void setEvents(LinkedBlockingQueue<AgentMessage> events) {
+        this.events = events;
+    }
+
     public void run() {
         System.out.println("Event bus running");
         while (!Thread.currentThread().interrupted()) {