Fix an invalid comparison of UserPresenceState to str

changelog.d/14393.bugfix

@@ -0,0 +1 @@
+Fix a bug introduced in 1.58.0 where a user with presence state 'org.matrix.msc3026.busy' would mistakenly be set to 'online' when calling `/sync` or `/events` on a worker process.

synapse/handlers/presence.py

@@ -478,7 +478,7 @@ async def user_syncing(
             return _NullContextManager()
 
         prev_state = await self.current_state_for_user(user_id)
-        if prev_state != PresenceState.BUSY:
+        if prev_state.state != PresenceState.BUSY:
             # We set state here but pass ignore_status_msg = True as we don't want to
             # cause the status message to be cleared.
             # Note that this causes last_active_ts to be incremented which is not

tests/handlers/test_presence.py

@@ -15,6 +15,7 @@
 from typing import Optional
 from unittest.mock import Mock, call
 
+from parameterized import parameterized
 from signedjson.key import generate_signing_key
 
 from synapse.api.constants import EventTypes, Membership, PresenceState
@@ -37,6 +38,7 @@
 from synapse.types import UserID, get_domain_from_id
 
 from tests import unittest
+from tests.replication._base import BaseMultiWorkerStreamTestCase
 
 
 class PresenceUpdateTestCase(unittest.HomeserverTestCase):
@@ -505,7 +507,7 @@ def test_last_active(self):
         self.assertEqual(state, new_state)
 
 
-class PresenceHandlerTestCase(unittest.HomeserverTestCase):
+class PresenceHandlerTestCase(BaseMultiWorkerStreamTestCase):
     def prepare(self, reactor, clock, hs):
         self.presence_handler = hs.get_presence_handler()
         self.clock = hs.get_clock()
@@ -716,19 +718,41 @@ def test_set_presence_from_syncing_keeps_status(self):
         # our status message should be the same as it was before
         self.assertEqual(state.status_msg, status_msg)
 
-    def test_set_presence_from_syncing_keeps_busy(self):
+    @parameterized.expand([(False,), (True,)])
+    @unittest.override_config(
+        {
+            "experimental_features": {
+                "msc3026_enabled": True,
+            },
+        }
+    )
+    def test_set_presence_from_syncing_keeps_busy(self, test_with_workers: bool):
         """Test that presence set by syncing doesn't affect busy status"""
-        # while this isn't the default
-        self.presence_handler._busy_presence_enabled = True
-
         user_id = "@test:server"
         status_msg = "I'm busy!"
 
+        worker_hs = None
+        if test_with_workers:
+            # Create a worker that will handle /sync traffic
+            worker_hs = self.make_worker_hs(
+                "synapse.app.generic_worker", {"worker_name": "presence_writer"}
+            )
+
         self._set_presencestate_with_status_msg(user_id, PresenceState.BUSY, status_msg)
 
-        self.get_success(
-            self.presence_handler.user_syncing(user_id, True, PresenceState.ONLINE)
-        )
+        if test_with_workers and worker_hs is not None:
+            # Sync against a worker instead of the main process. Busy state should still
+            # be preserved in this instance.
+            self.get_success(
+                worker_hs.get_presence_handler().user_syncing(
+                    user_id, True, PresenceState.ONLINE
+                )
+            )
+        else:
+            # Sync against the main process.
+            self.get_success(
+                self.presence_handler.user_syncing(user_id, True, PresenceState.ONLINE)
+            )
 
         state = self.get_success(
             self.presence_handler.get_state(UserID.from_string(user_id))

tests/module_api/test_api.py

@@ -762,7 +762,7 @@ def test_send_local_online_presence_to_workers(self):
 
 
 def _test_sending_local_online_presence_to_local_user(
-    test_case: HomeserverTestCase, test_with_workers: bool = False
+    test_case: BaseMultiWorkerStreamTestCase, test_with_workers: bool = False
 ):
     """Tests that send_local_presence_to_users sends local online presence to local users.
 

tests/replication/_base.py

@@ -542,8 +542,13 @@ def handle_command(self, command, *args):
             self.send("OK")
         elif command == b"GET":
             self.send(None)
+
+        # Connection keep-alives.
+        elif command == b"PING":
+            self.send("PONG")
+
         else:
-            raise Exception("Unknown command")
+            raise Exception(f"Unknown command: {command}")
 
     def send(self, msg):
         """Send a message back to the client."""