Merge pull request #87 from m-chrzan/gossip-girl

Begin gossip girl

13 files changed, 771 insertions, 2 deletions

diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/AttributesMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/AttributesMessage.java
new file mode 100644
index 0000000..49bda14
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/AttributesMessage.java
@@ -0,0 +1,31 @@
+package pl.edu.mimuw.cloudatlas.agent.messages;
+
+import java.util.Map;
+
+import pl.edu.mimuw.cloudatlas.model.AttributesMap;
+import pl.edu.mimuw.cloudatlas.model.PathName;
+
+public class AttributesMessage extends RemoteGossipGirlMessage {
+    private PathName path;
+    private AttributesMap attributes;
+    private long receiverGossipId;
+
+    public AttributesMessage(String messageId, long timestamp, PathName path, AttributesMap attributes, long receiverGossipId) {
+        super(messageId, timestamp, Type.ATTRIBUTES);
+        this.path = path;
+        this.attributes = attributes;
+        this.receiverGossipId = receiverGossipId;
+    }
+
+    public PathName getPath() {
+        return path;
+    }
+
+    public AttributesMap getAttributes() {
+        return attributes;
+    }
+
+    public long getReceiverGossipId() {
+        return receiverGossipId;
+    }
+}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/GossipGirlMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/GossipGirlMessage.java
new file mode 100644
index 0000000..508fe88
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/GossipGirlMessage.java
@@ -0,0 +1,29 @@
+package pl.edu.mimuw.cloudatlas.agent.messages;
+
+import pl.edu.mimuw.cloudatlas.agent.modules.Module;
+import pl.edu.mimuw.cloudatlas.agent.modules.ModuleType;
+
+public abstract class GossipGirlMessage extends AgentMessage {
+    public enum Type {
+        ATTRIBUTES,
+        HEJKA,
+        INITIATE,
+        NO_CO_TAM,
+        QUERY
+    }
+
+    private Type type;
+
+    public GossipGirlMessage(String messageId, long timestamp, Type type) {
+        super(messageId, ModuleType.GOSSIP, timestamp);
+        this.type = type;
+    }
+
+    public Type getType() {
+        return type;
+    }
+
+    public void callMe(Module module) throws InterruptedException, Module.InvalidMessageType {
+        module.handleTyped(this);
+    }
+}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/HejkaMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/HejkaMessage.java
new file mode 100644
index 0000000..340d939
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/HejkaMessage.java
@@ -0,0 +1,32 @@
+package pl.edu.mimuw.cloudatlas.agent.messages;
+
+import java.util.Map;
+
+import pl.edu.mimuw.cloudatlas.model.Attribute;
+import pl.edu.mimuw.cloudatlas.model.PathName;
+import pl.edu.mimuw.cloudatlas.model.ValueTime;
+
+public class HejkaMessage extends RemoteGossipGirlMessage {
+    private long senderGossipId;
+    private Map<PathName, ValueTime> zoneTimestamps;
+    private Map<Attribute, ValueTime> queryTimestamps;
+
+    public HejkaMessage(String messageId, long timestamp, long senderGossipId, Map<PathName, ValueTime> zoneTimestamps, Map<Attribute, ValueTime> queryTimestamps) {
+        super(messageId, timestamp, Type.HEJKA);
+        this.senderGossipId = senderGossipId;
+        this.zoneTimestamps = zoneTimestamps;
+        this.queryTimestamps = queryTimestamps;
+    }
+
+    public long getSenderGossipId() {
+        return senderGossipId;
+    }
+
+    public Map<PathName, ValueTime> getZoneTimestamps() {
+        return zoneTimestamps;
+    }
+
+    public Map<Attribute, ValueTime> getQueryTimestamps() {
+        return queryTimestamps;
+    }
+}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/InitiateGossipMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/InitiateGossipMessage.java
new file mode 100644
index 0000000..955570e
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/InitiateGossipMessage.java
@@ -0,0 +1,24 @@
+package pl.edu.mimuw.cloudatlas.agent.messages;
+
+import pl.edu.mimuw.cloudatlas.agent.modules.ModuleType;
+import pl.edu.mimuw.cloudatlas.model.PathName;
+import pl.edu.mimuw.cloudatlas.model.ValueContact;
+
+public class InitiateGossipMessage extends GossipGirlMessage {
+    private PathName ourPath;
+    private ValueContact theirContact;
+
+    public InitiateGossipMessage(String messageId, long timestamp, PathName ourPath, ValueContact theirContact) {
+        super(messageId, timestamp, Type.INITIATE);
+        this.ourPath = ourPath;
+        this.theirContact = theirContact;
+    }
+
+    public PathName getOurPath() {
+        return ourPath;
+    }
+
+    public ValueContact getTheirContact() {
+        return theirContact;
+    }
+}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/NoCoTamMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/NoCoTamMessage.java
new file mode 100644
index 0000000..3dd0c4d
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/NoCoTamMessage.java
@@ -0,0 +1,50 @@
+package pl.edu.mimuw.cloudatlas.agent.messages;
+
+import java.util.Map;
+
+import pl.edu.mimuw.cloudatlas.model.Attribute;
+import pl.edu.mimuw.cloudatlas.model.PathName;
+import pl.edu.mimuw.cloudatlas.model.ValueTime;
+
+public class NoCoTamMessage extends RemoteGossipGirlMessage {
+    private long receiverGossipId;
+    private long senderGossipId;
+    private Map<PathName, ValueTime> zoneTimestamps;
+    private Map<Attribute, ValueTime> queryTimestamps;
+    private ValueTime hejkaSendTimestamp;
+    private ValueTime hejkaReceiveTimestamp;
+
+    public NoCoTamMessage(String messageId, long timestamp, long receiverGossipId, long senderGossipId, Map<PathName, ValueTime> zoneTimestamps, Map<Attribute, ValueTime> queryTimestamps, ValueTime hejkaSendTimestamp, ValueTime hejkaReceiveTimestamp) {
+        super(messageId, timestamp, Type.NO_CO_TAM);
+        this.receiverGossipId = receiverGossipId;
+        this.senderGossipId = senderGossipId;
+        this.zoneTimestamps = zoneTimestamps;
+        this.queryTimestamps = queryTimestamps;
+        this.hejkaSendTimestamp = hejkaSendTimestamp;
+        this.hejkaReceiveTimestamp = hejkaReceiveTimestamp;
+    }
+
+    public long getReceiverGossipId() {
+        return receiverGossipId;
+    }
+
+    public long getSenderGossipId() {
+        return senderGossipId;
+    }
+
+    public Map<PathName, ValueTime> getZoneTimestamps() {
+        return zoneTimestamps;
+    }
+
+    public Map<Attribute, ValueTime> getQueryTimestamps() {
+        return queryTimestamps;
+    }
+
+    public ValueTime getHejkaSendTimestamp() {
+        return hejkaSendTimestamp;
+    }
+
+    public ValueTime getHejkaReceiveTimestamp() {
+        return hejkaReceiveTimestamp;
+    }
+}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/QueryMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/QueryMessage.java
new file mode 100644
index 0000000..2b3b064
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/QueryMessage.java
@@ -0,0 +1,29 @@
+package pl.edu.mimuw.cloudatlas.agent.messages;
+
+import pl.edu.mimuw.cloudatlas.model.Attribute;
+import pl.edu.mimuw.cloudatlas.model.ValueQuery;
+
+public class QueryMessage extends RemoteGossipGirlMessage {
+    private Attribute name;
+    private ValueQuery query;
+    private long receiverGossipId;
+
+    public QueryMessage(String messageId, long timestamp, Attribute name, ValueQuery query, long receiverGossipId) {
+        super(messageId, timestamp, Type.QUERY);
+        this.name = name;
+        this.query = query;
+        this.receiverGossipId = receiverGossipId;
+    }
+
+    public Attribute getName() {
+        return name;
+    }
+
+    public ValueQuery getQuery() {
+        return query;
+    }
+
+    public long getReceiverGossipId() {
+        return receiverGossipId;
+    }
+}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/RemoteGossipGirlMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/RemoteGossipGirlMessage.java
new file mode 100644
index 0000000..0a3a868
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/RemoteGossipGirlMessage.java
@@ -0,0 +1,28 @@
+package pl.edu.mimuw.cloudatlas.agent.messages;
+
+import pl.edu.mimuw.cloudatlas.model.ValueTime;
+
+public class RemoteGossipGirlMessage extends GossipGirlMessage {
+    private ValueTime sentTimestamp;
+    private ValueTime receivedTimestamp;
+
+    public RemoteGossipGirlMessage(String messageId, long timestamp, Type type) {
+        super(messageId, timestamp, type);
+    }
+
+    public void setSentTimestamp(ValueTime sentTimestamp) {
+        this.sentTimestamp = sentTimestamp;
+    }
+
+    public void setReceivedTimestamp(ValueTime receivedTimestamp) {
+        this.receivedTimestamp = receivedTimestamp;
+    }
+
+    public ValueTime getSentTimestamp() {
+        return sentTimestamp;
+    }
+
+    public ValueTime getReceivedTimestamp() {
+        return receivedTimestamp;
+    }
+}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/GossipGirl.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/GossipGirl.java
new file mode 100644
index 0000000..73aef8b
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/GossipGirl.java
@@ -0,0 +1,106 @@
+package pl.edu.mimuw.cloudatlas.agent.modules;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import pl.edu.mimuw.cloudatlas.agent.messages.AttributesMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.GetStateMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.GossipGirlMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.HejkaMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.InitiateGossipMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.NoCoTamMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.QueryMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.ResponseMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.StateMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.UDUPMessage;
+import pl.edu.mimuw.cloudatlas.model.Attribute;
+import pl.edu.mimuw.cloudatlas.model.AttributesMap;
+import pl.edu.mimuw.cloudatlas.model.PathName;
+import pl.edu.mimuw.cloudatlas.model.ValueQuery;
+import pl.edu.mimuw.cloudatlas.model.ValueTime;
+import pl.edu.mimuw.cloudatlas.model.ZMI;
+
+public class GossipGirl extends Module {
+    private long nextGossipId = 0;
+
+    private Map<Long, GossipGirlState> gossipStates;
+    public GossipGirl() {
+        super(ModuleType.GOSSIP);
+        gossipStates = new HashMap();
+    }
+
+    public void handleTyped(GossipGirlMessage message) throws InterruptedException, InvalidMessageType {
+        switch(message.getType()) {
+            case INITIATE:
+                initiateGossip((InitiateGossipMessage) message);
+                break;
+            case NO_CO_TAM:
+                handleNoCoTam((NoCoTamMessage) message);
+                break;
+            default:
+                throw new InvalidMessageType("This type of message cannot be handled by GossipGirl");
+        }
+    }
+
+    public void handleTyped(ResponseMessage message) throws InterruptedException, InvalidMessageType {
+        switch(message.getType()) {
+            case STATE:
+                setState((StateMessage) message);
+                break;
+            default:
+                throw new InvalidMessageType("This type of message cannot be handled by GossipGirl");
+        }
+    }
+
+    private void initiateGossip(InitiateGossipMessage message) throws InterruptedException {
+        Long gossipId = nextGossipId;
+        nextGossipId++;
+        gossipStates.put(gossipId, new GossipGirlState(gossipId, message.getOurPath(), message.getTheirContact(), true));
+
+        GetStateMessage getState = new GetStateMessage("", 0, ModuleType.GOSSIP, gossipId);
+        sendMessage(getState);
+    }
+
+    private void setState(StateMessage message) throws InterruptedException {
+        GossipGirlState state = gossipStates.get(message.getRequestId());
+        if (state != null) {
+            state.setState(message.getZMI(), message.getQueries());
+            if (state.state == GossipGirlState.State.SEND_HEJKA) {
+                HejkaMessage hejka = new HejkaMessage(
+                        "",
+                        0,
+                        state.gossipId,
+                        state.getZoneTimestampsToSend(),
+                        state.getQueryTimestampsToSend()
+                );
+                UDUPMessage udupMessage = new UDUPMessage("", 0, state.theirContact, hejka);
+                sendMessage(udupMessage);
+                state.sentHejka();
+            }
+        } else {
+            System.out.println("ERROR: GossipGirl got state for a nonexistent gossip");
+        }
+    }
+
+    private void handleNoCoTam(NoCoTamMessage message) throws InterruptedException {
+        GossipGirlState state = gossipStates.get(message.getReceiverGossipId());
+        if (state != null) {
+            state.handleNoCoTam(message);
+            for (ZMI zmi : state.getZMIsToSend()) {
+                AttributesMessage attributesMessage = new AttributesMessage("", 0, zmi.getPathName(), zmi.getAttributes(), state.theirGossipId);
+                UDUPMessage udupMessage = new UDUPMessage("", 0, state.theirContact, attributesMessage);
+                sendMessage(udupMessage);
+            }
+
+            for (Entry<Attribute, ValueQuery> query : state.getQueriesToSend()) {
+                QueryMessage queryMessage = new QueryMessage("", 0, query.getKey(), query.getValue(), state.theirGossipId);
+                UDUPMessage udupMessage = new UDUPMessage("", 0, state.theirContact, queryMessage);
+                sendMessage(udupMessage);
+            }
+            state.sentInfo();
+        } else {
+            System.out.println("ERROR: GossipGirl got state for a nonexistent gossip");
+        }
+    }
+}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/GossipGirlState.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/GossipGirlState.java
new file mode 100644
index 0000000..df17957
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/GossipGirlState.java
@@ -0,0 +1,197 @@
+package pl.edu.mimuw.cloudatlas.agent.modules;
+
+import java.util.AbstractMap.SimpleImmutableEntry;
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import pl.edu.mimuw.cloudatlas.agent.messages.NoCoTamMessage;
+import pl.edu.mimuw.cloudatlas.model.Attribute;
+import pl.edu.mimuw.cloudatlas.model.PathName;
+import pl.edu.mimuw.cloudatlas.model.ValueContact;
+import pl.edu.mimuw.cloudatlas.model.ValueQuery;
+import pl.edu.mimuw.cloudatlas.model.ValueTime;
+import pl.edu.mimuw.cloudatlas.model.ValueUtils;
+import pl.edu.mimuw.cloudatlas.model.ZMI;
+
+public class GossipGirlState {
+    public enum State {
+        WAIT_FOR_STATE_INITIALIZER,
+        WAIT_FOR_STATE_RESPONDER,
+        SEND_HEJKA,
+        SEND_INFO,
+        WAIT_FOR_NO_CO_TAM,
+        WAIT_FOR_FIRST_INFO,
+        WAIT_FOR_INFO,
+        ERROR
+    }
+    public PathName ourPath;
+    public ValueContact theirContact;
+    public long gossipId;
+    public long theirGossipId;
+    public long timeOffest;
+    public State state;
+    public ZMI hierarchy;
+    public Map<Attribute, Entry<ValueQuery, ValueTime>> queries;
+    public ValueTime hejkaSendTimestamp;
+    public ValueTime hejkaReceiveTimestamp;
+    public ValueTime noCoTamSendTimestamp;
+    public ValueTime noCoTamSendReceiveTimestamp;
+    private Map<PathName, ValueTime> theirZoneTimestamps;
+    private Map<Attribute, ValueTime> theirQueryTimestamps;
+
+    public GossipGirlState(long gossipId, PathName ourPath, ValueContact theirContact, boolean initiating) {
+        this.gossipId = gossipId;
+        this.ourPath = ourPath;
+        this.theirContact = theirContact;
+        if (initiating) {
+            state = State.WAIT_FOR_STATE_INITIALIZER;
+        } else {
+            state = State.WAIT_FOR_STATE_RESPONDER;
+        }
+    }
+
+    public void setState(ZMI hierarchy, Map<Attribute, Entry<ValueQuery, ValueTime>> queries) {
+        switch (state) {
+            case WAIT_FOR_STATE_INITIALIZER:
+                this.hierarchy = hierarchy;
+                this.queries = queries;
+                state = State.SEND_HEJKA;
+                break;
+            case WAIT_FOR_STATE_RESPONDER:
+                this.hierarchy = hierarchy;
+                this.queries = queries;
+                state = State.WAIT_FOR_FIRST_INFO;
+                break;
+            default:
+                System.out.println("ERROR: tried to set gossip state when not expected");
+                state = State.ERROR;
+        }
+    }
+
+    public void sentHejka() {
+        switch (state) {
+            case SEND_HEJKA:
+                state = state.WAIT_FOR_NO_CO_TAM;
+                break;
+            default:
+                System.out.println("ERROR: tried to set gossip state when not expected");
+                state = State.ERROR;
+        }
+    }
+
+    public void handleNoCoTam(NoCoTamMessage message) {
+        switch (state) {
+            case WAIT_FOR_NO_CO_TAM:
+                theirGossipId = message.getSenderGossipId();
+                theirZoneTimestamps = message.getZoneTimestamps();
+                theirQueryTimestamps = message.getQueryTimestamps();
+                hejkaSendTimestamp = message.getHejkaSendTimestamp();
+                hejkaReceiveTimestamp = message.getHejkaReceiveTimestamp();
+                state = State.SEND_INFO;
+                break;
+            default:
+                System.out.println("ERROR: tried to set gossip state when not expected");
+                state = State.ERROR;
+        }
+    }
+
+    public Map<PathName, ValueTime> getZoneTimestampsToSend() {
+        Map<PathName, ValueTime> timestamps = new HashMap();
+        collectZoneTimestamps(timestamps, hierarchy, theirContact.getName());
+        return timestamps;
+    }
+
+    public Map<Attribute, ValueTime> getQueryTimestampsToSend() {
+        Map<Attribute, ValueTime> queryTimestamps= new HashMap();
+        for (Entry<Attribute, Entry<ValueQuery, ValueTime>> query : queries.entrySet()) {
+            queryTimestamps.put(query.getKey(), query.getValue().getValue());
+        }
+
+        return queryTimestamps;
+    }
+
+    public List<ZMI> getZMIsToSend() {
+        List<ZMI> zmis = new LinkedList();
+        for (Entry<PathName, ValueTime> timestampedPath : getZoneTimestampsToSend().entrySet()) {
+            ValueTime theirTimestamp = theirZoneTimestamps.get(timestampedPath.getKey());
+            if (theirTimestamp == null || ValueUtils.valueLower(theirTimestamp, timestampedPath.getValue())) {
+                System.out.println("going to send " + timestampedPath.getKey().toString());
+                try {
+                    zmis.add(hierarchy.findDescendant(timestampedPath.getKey()));
+                } catch (ZMI.NoSuchZoneException e) {
+                    System.out.println("ERROR: didn't find a zone we wanted to send in getZMIsToSend");
+                }
+            }
+        }
+        return zmis;
+    }
+
+    public List<Entry<Attribute, ValueQuery>> getQueriesToSend() {
+        List<Entry<Attribute, ValueQuery>> queryList = new LinkedList();
+        for (Entry<Attribute, ValueTime> timestampedQuery : getQueryTimestampsToSend().entrySet()) {
+            ValueTime theirTimestamp = theirQueryTimestamps.get(timestampedQuery.getKey());
+            if (theirTimestamp == null || ValueUtils.valueLower(theirTimestamp, timestampedQuery.getValue())) {
+                queryList.add(
+                    new SimpleImmutableEntry(
+                        timestampedQuery.getKey(),
+                        queries.get(timestampedQuery.getKey()).getKey()
+                    )
+                );
+            }
+        }
+        return queryList;
+    }
+
+    public void collectZoneTimestamps(Map<PathName, ValueTime> timestamps, ZMI currentZMI, PathName recipientPath) {
+        for (ZMI zmi : currentZMI.getSons()) {
+            if (interestedIn(recipientPath, zmi.getPathName())) {
+                ValueTime timestamp = (ValueTime) zmi.getAttributes().getOrNull("timestamp");
+                if (timestamp != null) {
+                    timestamps.put(zmi.getPathName(), timestamp);
+                } else {
+                    System.out.println("ERROR: collectZoneTimestamps encountered a zone with no timestamp");
+                }
+            } else {
+                collectZoneTimestamps(timestamps, zmi, recipientPath);
+            }
+        }
+    }
+
+    public boolean interestedIn(PathName recipientPath, PathName zmiPath) {
+        return isPrefix(zmiPath.levelUp(), recipientPath) && !isPrefix(zmiPath, recipientPath);
+    }
+
+    public boolean isPrefix(PathName prefix, PathName path) {
+        List<String> prefixComponents = prefix.getComponents();
+        List<String> pathComponents = path.getComponents();
+
+        if (prefixComponents.size() > pathComponents.size()) {
+            return false;
+        }
+
+        Iterator<String> prefixIterator = prefixComponents.iterator();
+        Iterator<String> pathIterator = pathComponents.iterator();
+
+        while (prefixIterator.hasNext()) {
+            if (!prefixIterator.next().equals(pathIterator.next())) {
+                return false;
+            }
+        }
+        return true;
+    }
+
+    public void sentInfo() {
+        switch (state) {
+            case SEND_INFO:
+                state = State.WAIT_FOR_INFO;
+                break;
+            default:
+                System.out.println("ERROR: tried to set gossip state when not expected");
+                state = State.ERROR;
+        }
+    }
+}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Module.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Module.java
index 67fdab9..ec87649 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Module.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Module.java
@@ -28,6 +28,10 @@ public abstract class Module {
         throw new InvalidMessageType("Got a TimerSchedulerMessage in module " + moduleType.toString());
     }
 
+    public void handleTyped(GossipGirlMessage message) throws InterruptedException, InvalidMessageType {
+        throw new InvalidMessageType("Got a GossipGirlMessage in module " + moduleType.toString());
+    }
+
     public void handleTyped(QurnikMessage message) throws InterruptedException, InvalidMessageType {
         throw new InvalidMessageType("Got a QurnikMessage in module " + moduleType.toString());
     }
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/ModuleType.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/ModuleType.java
index d221f06..40e629a 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/ModuleType.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/ModuleType.java
@@ -5,8 +5,7 @@ public enum ModuleType {
     TIMER_GTP,
     RMI,
     UDP,
-    GOSSIP_IN,
-    GOSSIP_OUT,
+    GOSSIP,
     STATE,
     QUERY,
     // for testing
diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/GossipGirlTest.java b/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/GossipGirlTest.java
new file mode 100644
index 0000000..7ac27e9
--- /dev/null
+++ b/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/GossipGirlTest.java
@@ -0,0 +1,236 @@
+package pl.edu.mimuw.cloudatlas.agent.modules;
+
+import org.junit.Before;
+import org.junit.Test;
+import static org.junit.Assert.*;
+import static org.hamcrest.CoreMatchers.hasItems;
+
+import java.net.InetAddress;
+import java.util.AbstractMap.SimpleImmutableEntry;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+
+import pl.edu.mimuw.cloudatlas.agent.MockExecutor;
+import pl.edu.mimuw.cloudatlas.agent.messages.AgentMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.AttributesMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.HejkaMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.GetStateMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.GossipGirlMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.InitiateGossipMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.QueryMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.NoCoTamMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.StanikMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.StateMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.UDUPMessage;
+import pl.edu.mimuw.cloudatlas.agent.modules.ModuleType;
+import pl.edu.mimuw.cloudatlas.model.AttributesMap;
+import pl.edu.mimuw.cloudatlas.model.Attribute;
+import pl.edu.mimuw.cloudatlas.model.PathName;
+import pl.edu.mimuw.cloudatlas.model.TestUtil;
+import pl.edu.mimuw.cloudatlas.model.ValueContact;
+import pl.edu.mimuw.cloudatlas.model.ValueInt;
+import pl.edu.mimuw.cloudatlas.model.ValueQuery;
+import pl.edu.mimuw.cloudatlas.model.ValueString;
+import pl.edu.mimuw.cloudatlas.model.ValueTime;
+import pl.edu.mimuw.cloudatlas.model.ValueUtils;
+import pl.edu.mimuw.cloudatlas.model.ZMI;
+
+public class GossipGirlTest {
+    private GossipGirl gossipGirl;
+    private MockExecutor executor;
+    private final PathName ourPath = new PathName("/son/grand");
+    private final PathName theirPath = new PathName("/son/bro");
+    private ValueContact theirContact;
+    private InitiateGossipMessage initiateGossipMessage;
+    private ValueTime testTime;
+    private ZMI initiatorHierarchy;
+    private Map<Attribute, Entry<ValueQuery, ValueTime>> initiatorQueries;
+    private StateMessage initiatorStateMessage;
+    private NoCoTamMessage noCoTamMessage;
+
+    @Before
+    public void setupLocals() throws Exception {
+        gossipGirl = new GossipGirl();
+        executor = new MockExecutor(gossipGirl);
+
+        theirContact = new ValueContact(
+            theirPath,
+            InetAddress.getByAddress("localhost", new byte[] { 127, 0, 0, 1 })
+        );
+        initiateGossipMessage = new InitiateGossipMessage(
+            "test_msg",
+            0,
+            ourPath,
+            theirContact
+        );
+
+        testTime = ValueUtils.currentTime();
+        setupHierarchy();
+        setupQueries();
+        initiatorStateMessage = new StateMessage("", ModuleType.GOSSIP, 0, 0, initiatorHierarchy, initiatorQueries);
+
+        Map<PathName, ValueTime> otherZoneTimestamps = makeOtherZoneTimestamps();
+        Map<Attribute, ValueTime> otherQueryTimestamps = makeOtherQueryTimestamps();
+
+        noCoTamMessage = new NoCoTamMessage("", 0, 0, 42, otherZoneTimestamps, otherQueryTimestamps, TestUtil.addToTime(testTime, 10), TestUtil.addToTime(testTime, 22));
+    }
+
+    public Map<PathName, ValueTime> makeOtherZoneTimestamps() {
+        Map<PathName, ValueTime> zoneTimestamps = new HashMap();
+        addOtherZoneTimestamp(zoneTimestamps, "/son/sis", -100);
+        addOtherZoneTimestamp(zoneTimestamps, "/son/bro", 0);
+        addOtherZoneTimestamp(zoneTimestamps, "/son/whodis", -300);
+
+        return zoneTimestamps;
+    }
+
+    public Map<Attribute, ValueTime> makeOtherQueryTimestamps() {
+        Map<Attribute, ValueTime> queryTimestamps = new HashMap();
+        addOtherQueryTimestamp(queryTimestamps, "&one", 10);
+        addOtherQueryTimestamp(queryTimestamps, "&query", -400);
+        addOtherQueryTimestamp(queryTimestamps, "&three", 0);
+        return queryTimestamps;
+    }
+
+    public void addOtherQueryTimestamp(Map<Attribute, ValueTime> timestamps, String name, long offset) {
+        timestamps.put(new Attribute(name), TestUtil.addToTime(testTime, offset));
+    }
+
+    public void addOtherZoneTimestamp(Map<PathName, ValueTime> timestamps, String path, long offset) {
+        timestamps.put(new PathName(path), TestUtil.addToTime(testTime, offset));
+    }
+
+    public void setupHierarchy() {
+        initiatorHierarchy = makeZMI(null, null, 13l, "hello", testTime);
+        ZMI son = makeZMI(initiatorHierarchy, "son", 42l, "world", testTime);
+        ZMI daughter = makeZMI(initiatorHierarchy, "daughter", 24l, "kebab", testTime);
+        ZMI grand = makeZMI(son, "grand", 1337l, "ok", testTime);
+        ZMI bro = makeZMI(son, "bro", 3451434l, "whazzup", testTime);
+        ZMI sis = makeZMI(son, "sis", 420l, "hey", testTime);
+    }
+
+    public void setupQueries() throws Exception {
+        initiatorQueries = new HashMap();
+        addQuery(initiatorQueries, "&one", "SELECT 1 AS one", testTime);
+        addQuery(initiatorQueries, "&two", "SELECT 2 AS two", testTime);
+        addQuery(initiatorQueries, "&query", "SELECT sum(foo) AS foo", testTime);
+    }
+
+    public void addQuery(Map<Attribute, Entry<ValueQuery, ValueTime>> queries, String name, String query, ValueTime timestamp) throws Exception {
+        queries.put(
+                new Attribute(name),
+                new SimpleImmutableEntry(new ValueQuery(query), timestamp)
+        );
+    }
+
+    private ZMI makeZMI(ZMI parent, String name, Long foo, String bar, ValueTime timestamp) {
+        ZMI zmi = new ZMI(parent);
+        if (parent != null) {
+            parent.addSon(zmi);
+        }
+        AttributesMap attributes = zmi.getAttributes();
+        attributes.add("name", new ValueString(name));
+        attributes.add("foo", new ValueInt(foo));
+        attributes.add("bar", new ValueString(bar));
+        attributes.add("timestamp", timestamp);
+        return zmi;
+    }
+
+    @Test
+    public void initiateGossipRequestsState() throws Exception {
+        gossipGirl.handleTyped(initiateGossipMessage);
+
+        AgentMessage receivedMessage = executor.messagesToPass.poll();
+        assertNotNull(receivedMessage);
+        assertEquals(ModuleType.STATE, receivedMessage.getDestinationModule());
+        StanikMessage stanikMessage = (StanikMessage) receivedMessage;
+        assertEquals(StanikMessage.Type.GET_STATE, stanikMessage.getType());
+        GetStateMessage getStateMessage = (GetStateMessage) stanikMessage;
+        assertEquals(ModuleType.GOSSIP, getStateMessage.getRequestingModule());
+    }
+
+    @Test
+    public void initiatorSendsHejkaOnState() throws Exception {
+        gossipGirl.handleTyped(initiateGossipMessage);
+        executor.messagesToPass.take();
+        gossipGirl.handleTyped(initiatorStateMessage);
+
+        AgentMessage receivedMessage = executor.messagesToPass.poll();
+        assertUDUPMessage(
+                receivedMessage,
+                new PathName("/son/bro"),
+                GossipGirlMessage.Type.HEJKA
+        );
+        HejkaMessage hejkaMessage = (HejkaMessage) ((UDUPMessage) receivedMessage).getContent();
+        assertEquals(0, hejkaMessage.getSenderGossipId());
+        System.out.println(hejkaMessage.getZoneTimestamps().keySet());
+        assertEquals(3, TestUtil.iterableSize(hejkaMessage.getZoneTimestamps().keySet()));
+        Set<PathName> zoneSet = hejkaMessage.getZoneTimestamps().keySet();
+        assertThat(zoneSet, hasItems(new PathName("/daughter")));
+        assertThat(zoneSet, hasItems(new PathName("/son/sis")));
+        assertThat(zoneSet, hasItems(new PathName("/son/grand")));
+
+        assertEquals(3, TestUtil.iterableSize(hejkaMessage.getQueryTimestamps().keySet()));
+        Set<Attribute> querySet = hejkaMessage.getQueryTimestamps().keySet();
+        assertThat(querySet, hasItems(new Attribute("&one")));
+        assertThat(querySet, hasItems(new Attribute("&two")));
+        assertThat(querySet, hasItems(new Attribute("&query")));
+    }
+
+    @Test
+    public void initiatorSendsZonesAndQueriesOnNoCoTam() throws Exception {
+        gossipGirl.handleTyped(initiateGossipMessage);
+        executor.messagesToPass.take();
+        gossipGirl.handleTyped(initiatorStateMessage);
+        executor.messagesToPass.take();
+        gossipGirl.handleTyped(noCoTamMessage);
+
+        // 3 ZMIs, 2 queries
+        assertEquals(5, executor.messagesToPass.size());
+
+        AgentMessage receivedMessage1 = executor.messagesToPass.poll();
+        assertAttributeMessage(receivedMessage1, "/son/bro", "/daughter");
+        AgentMessage receivedMessage2 = executor.messagesToPass.poll();
+        assertAttributeMessage(receivedMessage2, "/son/bro", "/son/sis");
+        AgentMessage receivedMessage3 = executor.messagesToPass.poll();
+        assertAttributeMessage(receivedMessage3, "/son/bro", "/son/grand");
+
+        AgentMessage receivedMessage4 = executor.messagesToPass.poll();
+        assertQueryMessage(receivedMessage4, "/son/bro", "&two", "SELECT 2 AS two");
+        AgentMessage receivedMessage5 = executor.messagesToPass.poll();
+        assertQueryMessage(receivedMessage5, "/son/bro", "&query", "SELECT sum(foo) AS foo");
+    }
+
+    private void assertQueryMessage(AgentMessage message, String recipientPath, String name, String query) throws Exception {
+        assertUDUPMessage(
+                message,
+                new PathName(recipientPath),
+                GossipGirlMessage.Type.QUERY
+        );
+        QueryMessage queryMessage = (QueryMessage) ((UDUPMessage) message).getContent();
+        assertEquals(new Attribute(name), queryMessage.getName());
+        assertEquals(new ValueQuery(query), queryMessage.getQuery());
+    }
+
+    private void assertAttributeMessage(AgentMessage message, String recipientPath, String zonePath) throws Exception {
+        assertUDUPMessage(
+                message,
+                new PathName(recipientPath),
+                GossipGirlMessage.Type.ATTRIBUTES
+        );
+        AttributesMessage attributesMessage = (AttributesMessage) ((UDUPMessage) message).getContent();
+        assertEquals(new PathName(zonePath), attributesMessage.getPath());
+    }
+
+    private void assertUDUPMessage(AgentMessage message, PathName destinationName, GossipGirlMessage.Type type) throws Exception {
+        assertNotNull(message);
+        assertEquals(ModuleType.UDP, message.getDestinationModule());
+        UDUPMessage udupMessage = (UDUPMessage) message;
+        assertEquals(destinationName, udupMessage.getContact().getName());
+        assertEquals(ModuleType.GOSSIP, udupMessage.getContent().getDestinationModule());
+        GossipGirlMessage gossipMessage = (GossipGirlMessage) udupMessage.getContent();
+        assertEquals(type, gossipMessage.getType());
+    }
+}
diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/model/TestUtil.java b/src/test/java/pl/edu/mimuw/cloudatlas/model/TestUtil.java
index ddc61a9..0fef6b4 100644
--- a/src/test/java/pl/edu/mimuw/cloudatlas/model/TestUtil.java
+++ b/src/test/java/pl/edu/mimuw/cloudatlas/model/TestUtil.java
@@ -9,4 +9,8 @@ public class TestUtil {
 
         return count;
     }
+
+    public static ValueTime addToTime(ValueTime time, long millis) {
+        return time.addValue(new ValueDuration(millis));
+    }
 }