Fix redaction tests

1 files changed, 8 insertions, 6 deletions

diff --git a/tests/storage/test_redaction.py b/tests/storage/test_redaction.py
index d81f7add1c..f670e154c3 100644
--- a/tests/storage/test_redaction.py
+++ b/tests/storage/test_redaction.py
@@ -102,7 +102,7 @@ class RedactionTestCase(unittest.TestCase):
     @defer.inlineCallbacks
     def inject_redaction(self, room, event_id, user, reason):
         builder = self.event_builder_factory.new({
-            "type": MessageEvent.TYPE,
+            "type": RoomRedactionEvent.TYPE,
             "sender": user.to_string(),
             "state_key": user.to_string(),
             "room_id": room.to_string(),
@@ -170,6 +170,10 @@ class RedactionTestCase(unittest.TestCase):
 
         event = results[0]
 
+        self.assertEqual(msg_event.event_id, event.event_id)
+
+        self.assertTrue("redacted_because" in event.unsigned)
+
         self.assertObjectHasAttributes(
             {
                 "type": MessageEvent.TYPE,
@@ -179,8 +183,6 @@ class RedactionTestCase(unittest.TestCase):
             event,
         )
 
-        self.assertTrue("redacted_because" in event.unsigned)
-
         self.assertObjectHasAttributes(
             {
                 "type": RoomRedactionEvent.TYPE,
@@ -247,6 +249,8 @@ class RedactionTestCase(unittest.TestCase):
 
         event = results[0]
 
+        self.assertTrue("redacted_because" in event.unsigned)
+
         self.assertObjectHasAttributes(
             {
                 "type": RoomMemberEvent.TYPE,
@@ -256,13 +260,11 @@ class RedactionTestCase(unittest.TestCase):
             event,
         )
 
-        self.assertTrue(hasattr(event, "redacted_because"))
-
         self.assertObjectHasAttributes(
             {
                 "type": RoomRedactionEvent.TYPE,
                 "user_id": self.u_alice.to_string(),
                 "content": {"reason": reason},
             },
-            event.redacted_because,
+            event.unsigned["redacted_because"],
         )