1
0

Fix tests

This commit is contained in:
Erik Johnston
2025-01-22 14:05:04 +00:00
parent 192fd59943
commit 3f2e5730f3
2 changed files with 4 additions and 3 deletions

View File

@@ -807,6 +807,7 @@ class FederationEventHandlerTests(unittest.FederatingHomeserverTestCase):
OTHER_USER = f"@user:{self.OTHER_SERVER_NAME}"
main_store = self.hs.get_datastores().main
epoch_store = self.hs.get_datastores().state_epochs
# Create the room.
kermit_user_id = self.register_user("kermit", "test")
@@ -958,7 +959,7 @@ class FederationEventHandlerTests(unittest.FederatingHomeserverTestCase):
bert_member_event.event_id: bert_member_event,
rejected_kick_event.event_id: rejected_kick_event,
},
state_res_store=StateResolutionStore(main_store),
state_res_store=StateResolutionStore(main_store, epoch_store),
)
),
[bert_member_event.event_id, rejected_kick_event.event_id],
@@ -1003,7 +1004,7 @@ class FederationEventHandlerTests(unittest.FederatingHomeserverTestCase):
rejected_power_levels_event.event_id,
],
event_map={},
state_res_store=StateResolutionStore(main_store),
state_res_store=StateResolutionStore(main_store, epoch_store),
full_conflicted_set=set(),
)
),

View File

@@ -441,7 +441,7 @@ class EventChainStoreTestCase(HomeserverTestCase):
persist_events_store._store_event_txn(
txn,
[
(e, EventContext(self.hs.get_storage_controllers(), {}))
(e, EventContext(self.hs.get_storage_controllers(), 1, {}))
for e in events
],
)