From 120b0562b2d811129b641cef05aa20fe76c398f4 Mon Sep 17 00:00:00 2001
From: Jonathan Edey <jedey@google.com>
Date: Mon, 13 Jan 2025 15:35:45 -0500
Subject: [PATCH 1/3] chore: Add unit tests for `Reference.listen()`

---
 tests/test_db.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/tests/test_db.py b/tests/test_db.py
index 4245f65f..f2ba0882 100644
--- a/tests/test_db.py
+++ b/tests/test_db.py
@@ -535,6 +535,49 @@ def callback(_):
         finally:
             testutils.cleanup_apps()
 
+    @pytest.mark.parametrize(
+        'url,emulator_host,expected_base_url,expected_namespace',
+        [
+            # Production URLs with no override:
+            ('https://test.firebaseio.com', None, 'https://test.firebaseio.com/.json', None),
+            ('https://test.firebaseio.com/', None, 'https://test.firebaseio.com/.json', None),
+
+            # Production URLs with emulator_host override:
+            ('https://test.firebaseio.com', 'localhost:9000', 'http://localhost:9000/.json',
+             'test'),
+            ('https://test.firebaseio.com/', 'localhost:9000', 'http://localhost:9000/.json',
+             'test'),
+
+            # Emulator URL with no override.
+            ('http://localhost:8000/?ns=test', None, 'http://localhost:8000/.json', 'test'),
+
+            # emulator_host is ignored when the original URL is already emulator.
+            ('http://localhost:8000/?ns=test', 'localhost:9999', 'http://localhost:8000/.json',
+             'test'),
+        ]
+    )
+    def test_listen_sse_client(self, url, emulator_host, expected_base_url, expected_namespace,
+                               mocker):
+        if emulator_host:
+            os.environ[_EMULATOR_HOST_ENV_VAR] = emulator_host
+
+        try:
+            firebase_admin.initialize_app(testutils.MockCredential(), {'databaseURL' : url})
+            ref = db.reference()
+            mock_sse_client = mocker.patch('firebase_admin._sseclient.SSEClient')
+            mock_callback = mocker.Mock()
+            ref.listen(mock_callback)
+            args, kwargs = mock_sse_client.call_args
+            assert args[0] == expected_base_url
+            if expected_namespace:
+                assert kwargs.get('params') == {'ns': expected_namespace}
+            else:
+                assert kwargs.get('params') == {}
+        finally:
+            if _EMULATOR_HOST_ENV_VAR in os.environ:
+                del os.environ[_EMULATOR_HOST_ENV_VAR]
+            testutils.cleanup_apps()
+
     def test_listener_session(self):
         firebase_admin.initialize_app(testutils.MockCredential(), {
             'databaseURL' : 'https://test.firebaseio.com',

From beaa6d7aa4cda981326c250ccba179e54452680a Mon Sep 17 00:00:00 2001
From: Jonathan Edey <jedey@google.com>
Date: Mon, 20 Jan 2025 10:01:26 -0500
Subject: [PATCH 2/3] Integration test for rtdb listeners

---
 integration/test_db.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/integration/test_db.py b/integration/test_db.py
index c448436d..6d7c9b1d 100644
--- a/integration/test_db.py
+++ b/integration/test_db.py
@@ -16,6 +16,7 @@
 import collections
 import json
 import os
+import time
 
 import pytest
 
@@ -245,6 +246,37 @@ def test_delete(self, testref):
         ref.delete()
         assert ref.get() is None
 
+class TestListenOperations:
+    """Test cases for listening to changes to node values."""
+
+    def test_listen(self, testref):
+        self.events = []
+        def callback(event):
+            self.events.append(event)
+
+        python = testref.parent
+        registration = python.listen(callback)
+        try:
+            ref = python.child('users').push()
+            assert ref.path == '/_adminsdk/python/users/' + ref.key
+            assert ref.get() == ''
+            
+            self.wait_for(self.events, count=2)
+            assert len(self.events) == 2
+            
+            assert self.events[1].event_type == 'put'
+            assert self.events[1].path == '/users/' + ref.key
+            assert self.events[1].data == ''
+        finally:
+            registration.close()
+
+    @classmethod
+    def wait_for(cls, events, count=1, timeout_seconds=5):
+        must_end = time.time() + timeout_seconds
+        while time.time() < must_end:
+            if len(events) >= count:
+                return
+        raise pytest.fail('Timed out while waiting for events')
 
 class TestAdvancedQueries:
     """Test cases for advanced interactions via the db.Query interface."""

From 0f72c6fe4984a2aab6b5f2f1094fcd53c380e753 Mon Sep 17 00:00:00 2001
From: Jonathan Edey <jedey@google.com>
Date: Mon, 20 Jan 2025 10:05:17 -0500
Subject: [PATCH 3/3] fix lint

---
 integration/test_db.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/integration/test_db.py b/integration/test_db.py
index 6d7c9b1d..0170743d 100644
--- a/integration/test_db.py
+++ b/integration/test_db.py
@@ -260,10 +260,10 @@ def callback(event):
             ref = python.child('users').push()
             assert ref.path == '/_adminsdk/python/users/' + ref.key
             assert ref.get() == ''
-            
+
             self.wait_for(self.events, count=2)
             assert len(self.events) == 2
-            
+
             assert self.events[1].event_type == 'put'
             assert self.events[1].path == '/users/' + ref.key
             assert self.events[1].data == ''