def test_snap_inplay(self): # inPlay self.stream = historicalstream.FlumineMarketStream( mock.Mock(inplay=True, seconds_to_start=None), 0) self.stream._caches = { "1.123": mock.Mock(_definition_status="OPEN", _definition_in_play=False), } self.assertEqual(len(self.stream.snap()), 0) self.stream._caches = { "1.123": mock.Mock(_definition_status="OPEN", _definition_in_play=True), } self.assertEqual(len(self.stream.snap()), 1) self.stream._caches = { "1.123": mock.Mock(_definition_status="CLOSED", _definition_in_play=False), } self.assertEqual(len(self.stream.snap()), 1) self.stream = historicalstream.FlumineMarketStream( mock.Mock(inplay=False, seconds_to_start=None), 0) self.stream._caches = { "1.123": mock.Mock(_definition_status="OPEN", _definition_in_play=False), } self.assertEqual(len(self.stream.snap()), 1) self.stream._caches = { "1.123": mock.Mock(_definition_status="OPEN", _definition_in_play=True), } self.assertEqual(len(self.stream.snap()), 0)
def test_snap_seconds_to_start(self): # secondsToStart self.stream = historicalstream.FlumineMarketStream( mock.Mock(inplay=None, seconds_to_start=600), 0) self.stream._caches = { "1.123": mock.Mock( publish_time=123, market_definition={ "status": "OPEN", "inPlay": False, "marketTime": 456, }, ) } self.assertEqual(len(self.stream.snap()), 1) self.stream._caches = { "1.123": mock.Mock( publish_time=123, market_definition={ "status": "OPEN", "inPlay": False, "marketTime": 1234567, }, ) } self.assertEqual(len(self.stream.snap()), 0)
def setUp(self) -> None: self.listener = mock.Mock() self.stream = historicalstream.FlumineMarketStream(self.listener, 0)