snapshot of https://github.com/zulip/zulip.git @ 48c69911c7dd4e03b20552eb3bd77ad8151c1357 (2026-09-15)

zerver/tests/test_devices.py

@@ -0,0 +1,45 @@
+from zerver.lib.test_classes import ZulipTestCase
+from zerver.models.devices import Device
+
+
+class TestDeviceRegistration(ZulipTestCase):
+    def test_register_device(self) -> None:
+        user = self.example_user("hamlet")
+
+        self.assertEqual(Device.objects.count(), 0)
+
+        result = self.api_post(user, "/api/v1/register_client_device")
+        data = self.assert_json_success(result)
+        self.assertIn("device_id", data)
+
+        device = Device.objects.get(id=data["device_id"])
+        self.assertEqual(device.user_id, user.id)
+
+    def test_remove_device(self) -> None:
+        user = self.example_user("hamlet")
+
+        self.assertEqual(Device.objects.count(), 0)
+
+        result = self.api_post(user, "/api/v1/register_client_device")
+        data = self.assert_json_success(result)
+        device = Device.objects.get(id=data["device_id"])
+
+        result = self.api_post(user, "/api/v1/remove_client_device", {"device_id": device.id})
+        self.assert_json_success(result)
+
+        self.assertEqual(Device.objects.count(), 0)
+
+    def test_remove_device_on_api_key_regeneration(self) -> None:
+        user = self.example_user("hamlet")
+
+        for _ in range(3):
+            result = self.api_post(user, "/api/v1/register_client_device")
+            self.assert_json_success(result)
+
+        self.assertEqual(Device.objects.filter(user=user).count(), 3)
+
+        # Verify all of the user's Device records deleted on API key
+        # regeneration, to stop sending E2EE push notifications.
+        result = self.api_post(user, "/api/v1/users/me/api_key/regenerate")
+        self.assert_json_success(result)
+        self.assertEqual(Device.objects.filter(user=user).count(), 0)