Rename annotate_state_groups to annotate_event_with_state

6 files changed, 22 insertions, 22 deletions

diff --git a/synapse/handlers/_base.py b/synapse/handlers/_base.py
index 07a8464107..30c6733063 100644
--- a/synapse/handlers/_base.py
+++ b/synapse/handlers/_base.py
@@ -64,7 +64,7 @@ class BaseHandler(object):
 
         snapshot.fill_out_prev_events(event)
 
-        yield self.state_handler.annotate_state_groups(event)
+        yield self.state_handler.annotate_event_with_state(event)
 
         yield self.auth.add_auth_events(event)
 
diff --git a/synapse/handlers/federation.py b/synapse/handlers/federation.py
index c2cd91bb39..d8d5730b65 100644
--- a/synapse/handlers/federation.py
+++ b/synapse/handlers/federation.py
@@ -111,7 +111,7 @@ class FederationHandler(BaseHandler):
         if state:
             state = [self.pdu_codec.event_from_pdu(p) for p in state]
 
-        is_new_state = yield self.state_handler.annotate_state_groups(
+        is_new_state = yield self.state_handler.annotate_event_with_state(
             event,
             old_state=state
         )
@@ -202,7 +202,7 @@ class FederationHandler(BaseHandler):
             event = self.pdu_codec.event_from_pdu(pdu)
 
             # FIXME (erikj): Not sure this actually works :/
-            yield self.state_handler.annotate_state_groups(event)
+            yield self.state_handler.annotate_event_with_state(event)
 
             events.append(event)
 
@@ -268,7 +268,7 @@ class FederationHandler(BaseHandler):
 
             logger.debug("do_invite_join state: %s", state)
 
-            is_new_state = yield self.state_handler.annotate_state_groups(
+            is_new_state = yield self.state_handler.annotate_event_with_state(
                 event,
                 old_state=state
             )
@@ -289,7 +289,7 @@ class FederationHandler(BaseHandler):
                 # FIXME: Auth these.
                 e.outlier = True
 
-                yield self.state_handler.annotate_state_groups(
+                yield self.state_handler.annotate_event_with_state(
                     e,
                 )
 
@@ -330,7 +330,7 @@ class FederationHandler(BaseHandler):
         snapshot = yield self.store.snapshot_room(event)
         snapshot.fill_out_prev_events(event)
 
-        yield self.state_handler.annotate_state_groups(event)
+        yield self.state_handler.annotate_event_with_state(event)
         yield self.auth.add_auth_events(event)
         self.auth.check(event, raises=True)
 
@@ -345,7 +345,7 @@ class FederationHandler(BaseHandler):
 
         event.outlier = False
 
-        is_new_state = yield self.state_handler.annotate_state_groups(event)
+        is_new_state = yield self.state_handler.annotate_event_with_state(event)
         self.auth.check(event, raises=True)
 
         # FIXME (erikj):  All this is duplicated above :(
@@ -421,7 +421,7 @@ class FederationHandler(BaseHandler):
             )
         )
 
-        yield self.state_handler.annotate_state_groups(event)
+        yield self.state_handler.annotate_event_with_state(event)
 
         yield self.store.persist_event(
             event,
diff --git a/synapse/state.py b/synapse/state.py
index 11c54fd38c..9c22cf7701 100644
--- a/synapse/state.py
+++ b/synapse/state.py
@@ -45,7 +45,7 @@ class StateHandler(object):
 
     @defer.inlineCallbacks
     @log_function
-    def annotate_state_groups(self, event, old_state=None):
+    def annotate_event_with_state(self, event, old_state=None):
         yield run_on_reactor()
 
         if old_state:
diff --git a/tests/handlers/test_federation.py b/tests/handlers/test_federation.py
index a9d6b2bb17..e386cddb38 100644
--- a/tests/handlers/test_federation.py
+++ b/tests/handlers/test_federation.py
@@ -36,7 +36,7 @@ class FederationTestCase(unittest.TestCase):
         self.mock_config.signing_key = [MockKey()]
 
         self.state_handler = NonCallableMock(spec_set=[
-            "annotate_state_groups",
+            "annotate_event_with_state",
         ])
 
         self.auth = NonCallableMock(spec_set=[
@@ -85,7 +85,7 @@ class FederationTestCase(unittest.TestCase):
         self.datastore.persist_event.return_value = defer.succeed(None)
         self.datastore.get_room.return_value = defer.succeed(True)
 
-        self.state_handler.annotate_state_groups.return_value = (
+        self.state_handler.annotate_event_with_state.return_value = (
             defer.succeed(False)
         )
 
@@ -95,7 +95,7 @@ class FederationTestCase(unittest.TestCase):
             ANY, False, is_new_state=False
         )
 
-        self.state_handler.annotate_state_groups.assert_called_once_with(
+        self.state_handler.annotate_event_with_state.assert_called_once_with(
             ANY,
             old_state=None,
         )
diff --git a/tests/handlers/test_room.py b/tests/handlers/test_room.py
index 55c9f6e142..ee264e5ee2 100644
--- a/tests/handlers/test_room.py
+++ b/tests/handlers/test_room.py
@@ -60,7 +60,7 @@ class RoomMemberHandlerTestCase(unittest.TestCase):
             ]),
             auth=NonCallableMock(spec_set=["check", "add_auth_events"]),
             state_handler=NonCallableMock(spec_set=[
-                "annotate_state_groups",
+                "annotate_event_with_state",
             ]),
             config=self.mock_config,
         )
@@ -251,7 +251,7 @@ class RoomCreationTest(unittest.TestCase):
             ]),
             auth=NonCallableMock(spec_set=["check", "add_auth_events"]),
             state_handler=NonCallableMock(spec_set=[
-                "annotate_state_groups",
+                "annotate_event_with_state",
             ]),
             ratelimiter=NonCallableMock(spec_set=[
                 "send_message",
@@ -282,7 +282,7 @@ class RoomCreationTest(unittest.TestCase):
         def annotate(event):
             event.state_events = {}
             return defer.succeed(None)
-        self.state_handler.annotate_state_groups.side_effect = annotate
+        self.state_handler.annotate_event_with_state.side_effect = annotate
 
         def hosts(room):
             return defer.succeed([])
@@ -311,6 +311,6 @@ class RoomCreationTest(unittest.TestCase):
         self.assertEquals(user_id, join_event.user_id)
         self.assertEquals(user_id, join_event.state_key)
 
-        self.assertTrue(self.state_handler.annotate_state_groups.called)
+        self.assertTrue(self.state_handler.annotate_event_with_state.called)
 
         self.assertTrue(self.federation.handle_new_event.called)
diff --git a/tests/test_state.py b/tests/test_state.py
index 3cc358be32..7979b54a35 100644
--- a/tests/test_state.py
+++ b/tests/test_state.py
@@ -44,7 +44,7 @@ class StateTestCase(unittest.TestCase):
             self.create_event(type="test2", state_key=""),
         ]
 
-        yield self.state.annotate_state_groups(event, old_state=old_state)
+        yield self.state.annotate_event_with_state(event, old_state=old_state)
 
         for k, v in event.old_state_events.items():
             type, state_key = k
@@ -66,7 +66,7 @@ class StateTestCase(unittest.TestCase):
             self.create_event(type="test2", state_key=""),
         ]
 
-        yield self.state.annotate_state_groups(event, old_state=old_state)
+        yield self.state.annotate_event_with_state(event, old_state=old_state)
 
         for k, v in event.old_state_events.items():
             type, state_key = k
@@ -99,7 +99,7 @@ class StateTestCase(unittest.TestCase):
             group_name: old_state,
         }
 
-        yield self.state.annotate_state_groups(event)
+        yield self.state.annotate_event_with_state(event)
 
         for k, v in event.old_state_events.items():
             type, state_key = k
@@ -141,7 +141,7 @@ class StateTestCase(unittest.TestCase):
             group_name: old_state,
         }
 
-        yield self.state.annotate_state_groups(event)
+        yield self.state.annotate_event_with_state(event)
 
         for k, v in event.old_state_events.items():
             type, state_key = k
@@ -199,7 +199,7 @@ class StateTestCase(unittest.TestCase):
             group_name_2: old_state_2,
         }
 
-        yield self.state.annotate_state_groups(event)
+        yield self.state.annotate_event_with_state(event)
 
         self.assertEqual(len(event.old_state_events), 5)
 
@@ -235,7 +235,7 @@ class StateTestCase(unittest.TestCase):
             group_name_2: old_state_2,
         }
 
-        yield self.state.annotate_state_groups(event)
+        yield self.state.annotate_event_with_state(event)
 
         self.assertEqual(len(event.old_state_events), 5)