From cb4b6c844a0c9e2d4a96165958ff5680ed82e160 Mon Sep 17 00:00:00 2001
From: Mark Haines <mark.haines@matrix.org>
Date: Fri, 14 Nov 2014 21:25:02 +0000
Subject: Merge PDUs and Events into one object

---
 tests/handlers/test_federation.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'tests/handlers')

diff --git a/tests/handlers/test_federation.py b/tests/handlers/test_federation.py
index 3f17ca8fb0..e19b073817 100644
--- a/tests/handlers/test_federation.py
+++ b/tests/handlers/test_federation.py
@@ -19,9 +19,10 @@ from tests import unittest
 from synapse.api.events.room import (
     MessageEvent,
 )
+
+from synapse.api.events import SynapseEvent
 from synapse.handlers.federation import FederationHandler
 from synapse.server import HomeServer
-from synapse.federation.units import Pdu
 
 from mock import NonCallableMock, ANY, Mock
 
@@ -74,7 +75,7 @@ class FederationTestCase(unittest.TestCase):
 
     @defer.inlineCallbacks
     def test_msg(self):
-        pdu = Pdu(
+        pdu = SynapseEvent(
             type=MessageEvent.TYPE,
             room_id="foo",
             content={"msgtype": u"fooo"},
-- 
cgit 1.5.1