def test_flumine_market_stream(self, mock_on_process):
mock_listener = mock.Mock(stream_unique_id=0)
stream = datastream.FlumineMarketStream(mock_listener, 0)
market_books = [{"id": "1.123"}, {"id": "1.456"}, {"id": "1.123"}]
stream._process(market_books, 123)
self.assertEqual(len(stream._caches), 2)
self.assertEqual(stream._updates_processed, 3)
mock_on_process.assert_called_with(
[mock_listener.stream_unique_id, 123, market_books])
def test_flumine_market_stream_market_closed(self, mock_on_process):
mock_listener = mock.Mock()
stream = datastream.FlumineMarketStream(mock_listener)
stream._caches = {"1.123": object}
market_books = [{"id": "1.123", "marketDefinition": {"status": "CLOSED"}}]
stream._process(market_books, 123)
self.assertEqual(stream._lookup, "mc")
self.assertEqual(len(stream._caches), 0)
self.assertEqual(stream._updates_processed, 1)
mock_on_process.assert_called_with(
[mock_listener.stream_unique_id, 123, market_books]
)
