use 2 workers in tests

Author: FrenchGithubUser

synapse/storage/databases/main/registration.py

@@ -2607,7 +2607,7 @@ def user_delete_access_tokens_for_devices_txn(
             # Print after transaction completes (cache invalidation has happened)
             if tokens_and_devices:
                 print(tokens_and_devices)
-                print(self.get_user_by_access_token(tokens_and_devices[0][0]))
+                print(await self.get_user_by_access_token(tokens_and_devices[0][0]))
         return results
 
     async def delete_access_token(self, access_token: str) -> None:

tests/rest/client/test_read_marker.py

@@ -29,12 +29,14 @@
 from synapse.util.clock import Clock
 
 from tests import unittest
+from tests.replication._base import BaseMultiWorkerStreamTestCase
+from tests.server import make_request
 
 ONE_HOUR_MS = 3600000
 ONE_DAY_MS = ONE_HOUR_MS * 24
 
 
-class ReadMarkerTestCase(unittest.HomeserverTestCase):
+class ReadMarkerTestCase(BaseMultiWorkerStreamTestCase):
     servlets = [
         login.register_servlets,
         register.register_servlets,
@@ -155,12 +157,12 @@ def send_message() -> str:
     def test_deleted_device_cannot_set_read_marker_or_send_message(self) -> None:
         """
         Test that a deleted device cannot set a read marker or send a message.
+        This test removes the device on one worker and tries the next requests on another worker.
         """
         # Register a user and login with a specific device_id
         user_id = self.register_user("testuser", "password")
         device_id = "TEST_DEVICE_123"
         access_token = self.login("testuser", "password", device_id=device_id)
-        print("access token: ", access_token)
 
         # Create a room
         room_id = self.helper.create_room_as(user_id, tok=access_token)
@@ -171,8 +173,24 @@ def test_deleted_device_cannot_set_read_marker_or_send_message(self) -> None:
         )
         event_id = message_result["event_id"]
 
-        # Delete the device
-        channel = self.make_request(
+        # Create two workers
+        worker1 = self.make_worker_hs(
+            "synapse.app.generic_worker",
+            extra_config={"worker_name": "worker1"},
+        )
+        worker2 = self.make_worker_hs(
+            "synapse.app.generic_worker",
+            extra_config={"worker_name": "worker2"},
+        )
+
+        # Get the sites for the workers
+        worker1_site = self._hs_to_site[worker1]
+        worker2_site = self._hs_to_site[worker2]
+
+        # Delete the device on worker1
+        channel = make_request(
+            self.reactor,
+            worker1_site,
             "DELETE",
             f"devices/{device_id}",
             access_token=access_token,
@@ -181,7 +199,9 @@ def test_deleted_device_cannot_set_read_marker_or_send_message(self) -> None:
         # If UI auth is required, provide it
         if channel.code == 401:
             session = channel.json_body["session"]
-            channel = self.make_request(
+            channel = make_request(
+                self.reactor,
+                worker1_site,
                 "DELETE",
                 f"devices/{device_id}",
                 access_token=access_token,
@@ -197,8 +217,10 @@ def test_deleted_device_cannot_set_read_marker_or_send_message(self) -> None:
 
         self.assertEqual(channel.code, 200, channel.result)
 
-        # Try to set a read marker with the deleted device's token
-        channel = self.make_request(
+        # Try to set a read marker with the deleted device's token on worker2
+        channel = make_request(
+            self.reactor,
+            worker2_site,
             "POST",
             f"/rooms/{room_id}/read_markers",
             content={
@@ -210,8 +232,10 @@ def test_deleted_device_cannot_set_read_marker_or_send_message(self) -> None:
         self.assertEqual(channel.code, 401, channel.result)
         self.assertEqual(channel.json_body["errcode"], "M_UNKNOWN_TOKEN")
 
-        # Try to send a message with the deleted device's token
-        channel = self.make_request(
+        # Try to send a message with the deleted device's token on worker2
+        channel = make_request(
+            self.reactor,
+            worker2_site,
             "PUT",
             f"/rooms/{room_id}/send/m.room.message/test_msg",
             content={