def test_flumine_listener(self, mock_market_stream, mock_race_stream):
listener = datastream.FlumineListener()
self.assertEqual(listener._add_stream(123, "marketSubscription"),
mock_market_stream())
with self.assertRaises(ListenerError):
listener._add_stream(123, "orderSubscription")
listener = datastream.FlumineListener()
self.assertEqual(listener._add_stream(123, "raceSubscription"),
mock_race_stream())
def test_flumine_listener(
self,
mock_market_stream,
mock_order_stream,
mock_race_stream,
mock_cricket_stream,
):
listener = datastream.FlumineListener()
self.assertEqual(
listener._add_stream(123, "marketSubscription"), mock_market_stream()
)
self.assertEqual(
listener._add_stream(123, "orderSubscription"), mock_order_stream()
)
self.assertEqual(
listener._add_stream(123, "raceSubscription"), mock_race_stream()
)
self.assertEqual(
listener._add_stream(123, "cricketSubscription"), mock_cricket_stream()
)