diff options
author | Andrew Morgan <andrew@amorgan.xyz> | 2021-11-24 15:19:57 +0000 |
---|---|---|
committer | Andrew Morgan <andrew@amorgan.xyz> | 2021-11-24 15:19:57 +0000 |
commit | b4a4b45dffb682c58030468e1aa94a922373086e (patch) | |
tree | c2074e7f9f06c93a39b3ceac0692435484d3b7fe | |
parent | Fix existing unit tests (diff) | |
download | synapse-b4a4b45dffb682c58030468e1aa94a922373086e.tar.xz |
rename set_type_stream_id_for_appservice -> set_appservice_stream_type_pos
-rw-r--r-- | synapse/handlers/appservice.py | 8 | ||||
-rw-r--r-- | synapse/storage/databases/main/appservice.py | 6 | ||||
-rw-r--r-- | tests/handlers/test_appservice.py | 2 | ||||
-rw-r--r-- | tests/storage/test_appservice.py | 10 |
4 files changed, 13 insertions, 13 deletions
diff --git a/synapse/handlers/appservice.py b/synapse/handlers/appservice.py index 816b51e682..c7ef6d2a9c 100644 --- a/synapse/handlers/appservice.py +++ b/synapse/handlers/appservice.py @@ -283,7 +283,7 @@ class ApplicationServicesHandler: with Measure(self.clock, "notify_interested_services_ephemeral"): for service in services: if stream_key == "typing_key": - # Note that we don't persist the token (via set_type_stream_id_for_appservice) + # Note that we don't persist the token (via set_appservice_stream_type_pos) # for typing_key due to performance reasons and due to their highly # ephemeral nature. # @@ -305,7 +305,7 @@ class ApplicationServicesHandler: self.scheduler.submit_ephemeral_events_for_as(service, events) # Persist the latest handled stream token for this appservice - await self.store.set_type_stream_id_for_appservice( + await self.store.set_appservice_stream_type_pos( service, "read_receipt", new_token ) @@ -314,7 +314,7 @@ class ApplicationServicesHandler: self.scheduler.submit_ephemeral_events_for_as(service, events) # Persist the latest handled stream token for this appservice - await self.store.set_type_stream_id_for_appservice( + await self.store.set_appservice_stream_type_pos( service, "presence", new_token ) @@ -329,7 +329,7 @@ class ApplicationServicesHandler: ) # Persist the latest handled stream token for this appservice - await self.store.set_type_stream_id_for_appservice( + await self.store.set_appservice_stream_type_pos( service, "to_device", new_token ) diff --git a/synapse/storage/databases/main/appservice.py b/synapse/storage/databases/main/appservice.py index 92e6a491c9..1986e36d52 100644 --- a/synapse/storage/databases/main/appservice.py +++ b/synapse/storage/databases/main/appservice.py @@ -411,7 +411,7 @@ class ApplicationServiceTransactionWorkerStore( "get_type_stream_id_for_appservice", get_type_stream_id_for_appservice_txn ) - async def set_type_stream_id_for_appservice( + async def set_appservice_stream_type_pos( self, service: ApplicationService, stream_type: str, pos: Optional[int] ) -> None: if stream_type not in ("read_receipt", "presence", "to_device"): @@ -420,7 +420,7 @@ class ApplicationServiceTransactionWorkerStore( % (stream_type,) ) - def set_type_stream_id_for_appservice_txn(txn): + def set_appservice_stream_type_pos_txn(txn): stream_id_type = "%s_stream_id" % stream_type txn.execute( "UPDATE application_services_state SET %s = ? WHERE as_id=?" @@ -429,7 +429,7 @@ class ApplicationServiceTransactionWorkerStore( ) await self.db_pool.runInteraction( - "set_type_stream_id_for_appservice", set_type_stream_id_for_appservice_txn + "set_appservice_stream_type_pos", set_appservice_stream_type_pos_txn ) diff --git a/tests/handlers/test_appservice.py b/tests/handlers/test_appservice.py index 0bbbce71fd..6ca9c7f9f1 100644 --- a/tests/handlers/test_appservice.py +++ b/tests/handlers/test_appservice.py @@ -282,7 +282,7 @@ class AppServiceHandlerTestCase(unittest.TestCase): self.mock_scheduler.submit_ephemeral_events_for_as.assert_called_once_with( interested_service, [event] ) - self.mock_store.set_type_stream_id_for_appservice.assert_called_once_with( + self.mock_store.set_appservice_stream_type_pos.assert_called_once_with( interested_service, "read_receipt", 580, diff --git a/tests/storage/test_appservice.py b/tests/storage/test_appservice.py index f26d5acf9c..de10b69b7c 100644 --- a/tests/storage/test_appservice.py +++ b/tests/storage/test_appservice.py @@ -433,10 +433,10 @@ class ApplicationServiceStoreTypeStreamIds(unittest.HomeserverTestCase): ValueError, ) - def test_set_type_stream_id_for_appservice(self): + def test_set_appservice_stream_type_pos(self): read_receipt_value = 1024 self.get_success( - self.store.set_type_stream_id_for_appservice( + self.store.set_appservice_stream_type_pos( self.service, "read_receipt", read_receipt_value ) ) @@ -446,7 +446,7 @@ class ApplicationServiceStoreTypeStreamIds(unittest.HomeserverTestCase): self.assertEqual(result, read_receipt_value) self.get_success( - self.store.set_type_stream_id_for_appservice( + self.store.set_appservice_stream_type_pos( self.service, "presence", read_receipt_value ) ) @@ -455,9 +455,9 @@ class ApplicationServiceStoreTypeStreamIds(unittest.HomeserverTestCase): ) self.assertEqual(result, read_receipt_value) - def test_set_type_stream_id_for_appservice_invalid_type(self): + def test_set_appservice_stream_type_pos_invalid_type(self): self.get_failure( - self.store.set_type_stream_id_for_appservice(self.service, "foobar", 1024), + self.store.set_appservice_stream_type_pos(self.service, "foobar", 1024), ValueError, ) |