Fix tests.

Erethon · Nov 26, 2014 · 822b15e · 822b15e
1 parent 3598c11
commit 822b15e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/handlers/test_federation.py b/tests/handlers/test_federation.py
@@ -98,7 +98,7 @@ def annotate(ev, old_state=None):
         yield self.handlers.federation_handler.on_receive_pdu(pdu, False)
 
         self.datastore.persist_event.assert_called_once_with(
-            ANY, is_new_state=False, backfilled=False
+            ANY, is_new_state=False, backfilled=False, current_state=None
         )
 
         self.state_handler.annotate_event_with_state.assert_called_once_with(