1 files changed, 11 insertions, 4 deletions
diff --git a/tests/handlers/test_federation.py b/tests/handlers/test_federation.py
index fb06e5e812..aea96a0986 100644
--- a/tests/handlers/test_federation.py
+++ b/tests/handlers/test_federation.py
@@ -280,16 +280,23 @@ class FederationTestCase(unittest.FederatingHomeserverTestCase):
# we poke this directly into _process_received_pdu, to avoid the
# federation handler wanting to backfill the fake event.
- self.get_success(
- federation_event_handler._process_received_pdu(
- self.OTHER_SERVER_NAME,
+ state_handler = self.hs.get_state_handler()
+ context = self.get_success(
+ state_handler.compute_event_context(
event,
- state_ids={
+ state_ids_before_event={
(e.type, e.state_key): e.event_id for e in current_state
},
partial_state=False,
)
)
+ self.get_success(
+ federation_event_handler._process_received_pdu(
+ self.OTHER_SERVER_NAME,
+ event,
+ context,
+ )
+ )
# we should now have 8 backwards extremities.
backwards_extremities = self.get_success(
|
