def test_snap_inplay(self):
# inPlay
self.stream = historicalstream.Stream(
mock.Mock(inplay=True, seconds_to_start=None)
)
self.stream._caches = {
"1.123": mock.Mock(market_definition={"status": "OPEN", "inPlay": False})
}
self.assertEqual(len(self.stream.snap()), 0)
self.stream._caches = {
"1.123": mock.Mock(market_definition={"status": "OPEN", "inPlay": True})
}
self.assertEqual(len(self.stream.snap()), 1)
self.stream._caches = {
"1.123": mock.Mock(market_definition={"status": "CLOSED", "inPlay": False})
}
self.assertEqual(len(self.stream.snap()), 1)
self.stream = historicalstream.Stream(
mock.Mock(inplay=False, seconds_to_start=None)
)
self.stream._caches = {
"1.123": mock.Mock(market_definition={"status": "OPEN", "inPlay": False})
}
self.assertEqual(len(self.stream.snap()), 1)
self.stream._caches = {
"1.123": mock.Mock(market_definition={"status": "OPEN", "inPlay": True})
}
self.assertEqual(len(self.stream.snap()), 0)
def test_snap_seconds_to_start(self):
# secondsToStart
self.stream = historicalstream.Stream(
mock.Mock(inplay=None, seconds_to_start=600)
)
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.Stream(self.listener)