File tree Expand file tree Collapse file tree 2 files changed +5
-2
lines changed
Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Original file line number Diff line number Diff 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 :
Original file line number Diff line number Diff 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
194196def 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
518519def test_live_add_stream_invalid (
You can’t perform that action at this time.
0 commit comments