MOD: Allow Live client reuse with new dataset

Author: nmacholl

CHANGELOG.md

@@ -1,5 +1,10 @@
 # Changelog
 
+## 0.36.4 - TBD
+
+#### Enhancements
+- A disconnected `Live` client can now be reused with a different dataset
+
 ## 0.36.3 - 2024-07-02
 
 #### Enhancements

databento/live/client.py

@@ -605,6 +605,7 @@ def _cleanup_client(self) -> None:
         Cleanup any stateful client data.
         """
         self._symbology_map.clear()
+        self._dataset = ""
 
         to_remove = []
         for stream in self._user_streams:

tests/test_live_client.py

@@ -460,7 +460,7 @@ async def test_live_subscribe(
     mock_live_server: MockLiveServerInterface,
     schema: Schema,
     stype_in: SType,
-    symbols: str,
+    symbols: str | None,
     start: str,
 ) -> None:
     """
@@ -635,6 +635,35 @@ def cb_sub(_: DBNRecord) -> None:
     assert second_sub.symbols == "TEST1"
 
 
+async def test_live_subscribe_different_dataset(
+    live_client: client.Live,
+    mock_live_server: MockLiveServerInterface,
+) -> None:
+    """
+    Test that once a Live client is disconnected, it can be used with a
+    different subscription dataset.
+    """
+    # Arrange
+    live_client.subscribe(
+        dataset=Dataset.GLBX_MDP3,
+        schema=Schema.MBO,
+    )
+
+    # Act
+    _ = await mock_live_server.wait_for_message_of_type(
+        message_type=gateway.SubscriptionRequest,
+    )
+
+    live_client.start()
+    await live_client.wait_for_close()
+
+    # Assert
+    live_client.subscribe(
+        dataset=Dataset.XNAS_ITCH,
+        schema=Schema.MBO,
+    )
+
+
 @pytest.mark.usefixtures("mock_live_server")
 def test_live_stop(
     live_client: client.Live,