Fix passing messages in server

1 files changed, 3 insertions, 1 deletions

diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPServer.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPServer.java
index 9a64bbd..6807a86 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPServer.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPServer.java
@@ -37,7 +37,9 @@ public class UDUPServer {
             if (packet.getLength() == this.bufSize) {
                 this.addPartialMessageAndCheckSerialization(packet.getAddress(), packet.getData());
             } else  {
-                if (msg.getDestinationModule() != ModuleType.UDP) {
+                if (msg.getContent().getDestinationModule() == ModuleType.TEST) {
+                    System.out.println("UDP server: test message received");
+                } else if (msg.getContent().getDestinationModule() != ModuleType.UDP) {
                     this.udp.sendMessage(msg.getContent());
                 }
             }