Updates tests
This commit is contained in:
parent
3fa362502c
commit
25c0a020f4
|
@ -36,8 +36,7 @@ class RedactionTestCase(unittest.TestCase):
|
|||
|
||||
self.store = hs.get_datastore()
|
||||
self.event_builder_factory = hs.get_event_builder_factory()
|
||||
self.handlers = hs.get_handlers()
|
||||
self.message_handler = self.handlers.message_handler
|
||||
self.event_creation_handler = hs.get_event_creation_handler()
|
||||
|
||||
self.u_alice = UserID.from_string("@alice:test")
|
||||
self.u_bob = UserID.from_string("@bob:test")
|
||||
|
@ -59,7 +58,7 @@ class RedactionTestCase(unittest.TestCase):
|
|||
"content": content,
|
||||
})
|
||||
|
||||
event, context = yield self.message_handler._create_new_client_event(
|
||||
event, context = yield self.event_creation_handler._create_new_client_event(
|
||||
builder
|
||||
)
|
||||
|
||||
|
@ -79,7 +78,7 @@ class RedactionTestCase(unittest.TestCase):
|
|||
"content": {"body": body, "msgtype": u"message"},
|
||||
})
|
||||
|
||||
event, context = yield self.message_handler._create_new_client_event(
|
||||
event, context = yield self.event_creation_handler._create_new_client_event(
|
||||
builder
|
||||
)
|
||||
|
||||
|
@ -98,7 +97,7 @@ class RedactionTestCase(unittest.TestCase):
|
|||
"redacts": event_id,
|
||||
})
|
||||
|
||||
event, context = yield self.message_handler._create_new_client_event(
|
||||
event, context = yield self.event_creation_handler._create_new_client_event(
|
||||
builder
|
||||
)
|
||||
|
||||
|
|
|
@ -37,8 +37,7 @@ class RoomMemberStoreTestCase(unittest.TestCase):
|
|||
# storage logic
|
||||
self.store = hs.get_datastore()
|
||||
self.event_builder_factory = hs.get_event_builder_factory()
|
||||
self.handlers = hs.get_handlers()
|
||||
self.message_handler = self.handlers.message_handler
|
||||
self.event_creation_handler = hs.get_event_creation_handler()
|
||||
|
||||
self.u_alice = UserID.from_string("@alice:test")
|
||||
self.u_bob = UserID.from_string("@bob:test")
|
||||
|
@ -58,7 +57,7 @@ class RoomMemberStoreTestCase(unittest.TestCase):
|
|||
"content": {"membership": membership},
|
||||
})
|
||||
|
||||
event, context = yield self.message_handler._create_new_client_event(
|
||||
event, context = yield self.event_creation_handler._create_new_client_event(
|
||||
builder
|
||||
)
|
||||
|
||||
|
|
Loading…
Reference in New Issue