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)
