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"],
)
|
