Skip to content

Commit e9e9c11

Browse files
committed
FIX: Fix missing map_symbols callback
1 parent 2df1976 commit e9e9c11

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

databento/live/client.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,9 @@ def __init__(
8888
self._dbn_queue: DBNQueue = DBNQueue(maxsize=DEFAULT_QUEUE_SIZE)
8989
self._metadata: SessionMetadata = SessionMetadata()
9090
self._symbology_map: dict[int, str | int] = {}
91-
self._user_callbacks: dict[RecordCallback, ExceptionCallback | None] = {}
91+
self._user_callbacks: dict[RecordCallback, ExceptionCallback | None] = {
92+
self._map_symbol: None,
93+
}
9294
self._user_streams: dict[IO[bytes], ExceptionCallback | None] = {}
9395

9496
def factory() -> _SessionProtocol:

tests/test_live_client.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,7 @@ def test_live_creation(
179179
port=mock_live_server.port,
180180
)
181181

182+
# Subscribe to connect
182183
live_client.subscribe(
183184
dataset=dataset,
184185
schema=Schema.MBO,
@@ -189,6 +190,7 @@ def test_live_creation(
189190
assert live_client._key == test_api_key
190191
assert live_client.dataset == dataset
191192
assert live_client.is_connected() is True
193+
assert live_client._map_symbol in live_client._user_callbacks
192194

193195

194196
def test_live_connect_auth(
@@ -512,7 +514,6 @@ def test_live_add_stream(
512514
live_client.add_stream(stream)
513515
assert stream in live_client._user_streams
514516
assert live_client._user_streams[stream] is None
515-
assert live_client._user_callbacks == {}
516517

517518

518519
def test_live_add_stream_invalid(

0 commit comments

Comments
 (0)