1 files changed, 146 insertions, 0 deletions
diff --git a/tests/federation/test_pdu_codec.py b/tests/federation/test_pdu_codec.py
new file mode 100644
index 0000000000..688182fa5b
--- /dev/null
+++ b/tests/federation/test_pdu_codec.py
@@ -0,0 +1,146 @@
+# -*- coding: utf-8 -*-
+from twisted.trial import unittest
+
+from synapse.federation.pdu_codec import (
+    PduCodec, encode_event_id, decode_event_id
+)
+from synapse.federation.units import Pdu
+#from synapse.api.events.room import MessageEvent
+
+from synapse.server import HomeServer
+
+from mock import Mock
+
+
+class PduCodecTestCase(unittest.TestCase):
+    def setUp(self):
+        self.hs = HomeServer("blargle.net")
+        self.event_factory = self.hs.get_event_factory()
+
+        self.codec = PduCodec(self.hs)
+
+    def test_decode_event_id(self):
+        self.assertEquals(
+            ("foo", "bar.com"),
+            decode_event_id("foo@bar.com", "A")
+        )
+
+        self.assertEquals(
+            ("foo", "bar.com"),
+            decode_event_id("foo", "bar.com")
+        )
+
+    def test_encode_event_id(self):
+        self.assertEquals("A@B", encode_event_id("A", "B"))
+
+    def test_codec_event_id(self):
+        event_id = "aa@bb.com"
+
+        self.assertEquals(
+            event_id,
+            encode_event_id(*decode_event_id(event_id, None))
+        )
+
+        pdu_id = ("aa", "bb.com")
+
+        self.assertEquals(
+            pdu_id,
+            decode_event_id(encode_event_id(*pdu_id), None)
+        )
+
+    def test_event_from_pdu(self):
+        pdu = Pdu(
+            pdu_id="foo",
+            context="rooooom",
+            pdu_type="m.room.message",
+            origin="bar.com",
+            ts=12345,
+            depth=5,
+            prev_pdus=[("alice", "bob.com")],
+            is_state=False,
+            content={"msgtype": u"test"},
+        )
+
+        event = self.codec.event_from_pdu(pdu)
+
+        self.assertEquals("foo@bar.com", event.event_id)
+        self.assertEquals(pdu.context, event.room_id)
+        self.assertEquals(pdu.is_state, event.is_state)
+        self.assertEquals(pdu.depth, event.depth)
+        self.assertEquals(["alice@bob.com"], event.prev_events)
+        self.assertEquals(pdu.content, event.content)
+
+    def test_pdu_from_event(self):
+        event = self.event_factory.create_event(
+            etype="m.room.message",
+            event_id="gargh_id",
+            room_id="rooom",
+            user_id="sender",
+            content={"msgtype": u"test"},
+        )
+
+        pdu = self.codec.pdu_from_event(event)
+
+        self.assertEquals(event.event_id, pdu.pdu_id)
+        self.assertEquals(self.hs.hostname, pdu.origin)
+        self.assertEquals(event.room_id, pdu.context)
+        self.assertEquals(event.content, pdu.content)
+        self.assertEquals(event.type, pdu.pdu_type)
+
+        event = self.event_factory.create_event(
+            etype="m.room.message",
+            event_id="gargh_id@bob.com",
+            room_id="rooom",
+            user_id="sender",
+            content={"msgtype": u"test"},
+        )
+
+        pdu = self.codec.pdu_from_event(event)
+
+        self.assertEquals("gargh_id", pdu.pdu_id)
+        self.assertEquals("bob.com", pdu.origin)
+        self.assertEquals(event.room_id, pdu.context)
+        self.assertEquals(event.content, pdu.content)
+        self.assertEquals(event.type, pdu.pdu_type)
+
+    def test_event_from_state_pdu(self):
+        pdu = Pdu(
+            pdu_id="foo",
+            context="rooooom",
+            pdu_type="m.room.topic",
+            origin="bar.com",
+            ts=12345,
+            depth=5,
+            prev_pdus=[("alice", "bob.com")],
+            is_state=True,
+            content={"topic": u"test"},
+            state_key="",
+        )
+
+        event = self.codec.event_from_pdu(pdu)
+
+        self.assertEquals("foo@bar.com", event.event_id)
+        self.assertEquals(pdu.context, event.room_id)
+        self.assertEquals(pdu.is_state, event.is_state)
+        self.assertEquals(pdu.depth, event.depth)
+        self.assertEquals(["alice@bob.com"], event.prev_events)
+        self.assertEquals(pdu.content, event.content)
+        self.assertEquals(pdu.state_key, event.state_key)
+
+    def test_pdu_from_state_event(self):
+        event = self.event_factory.create_event(
+            etype="m.room.topic",
+            event_id="gargh_id",
+            room_id="rooom",
+            user_id="sender",
+            content={"topic": u"test"},
+        )
+
+        pdu = self.codec.pdu_from_event(event)
+
+        self.assertEquals(event.event_id, pdu.pdu_id)
+        self.assertEquals(self.hs.hostname, pdu.origin)
+        self.assertEquals(event.room_id, pdu.context)
+        self.assertEquals(event.content, pdu.content)
+        self.assertEquals(event.type, pdu.pdu_type)
+        self.assertEquals(event.state_key, pdu.state_key)