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] )