def test_timeout(self):
with testfixtures.Replacer() as r:
mock_time = testfixtures.test_time(delta=0.1, delta_type='seconds')
r.replace('streamexpect.time.time', mock_time)
r.replace('streamexpect.time.sleep', lambda _: None)
stream = EmptyStream()
adapter = PollingStreamAdapter(stream)
with self.assertRaises(ExpectTimeout):
adapter.poll(1)
def test_poll(self):
stream = PiecewiseStream(b'alpha beta gamma omega', max_chunk=5)
with testfixtures.Replacer() as r:
mock_time = testfixtures.test_time(delta=0.1, delta_type='seconds')
r.replace('streamexpect.time.time', mock_time)
adapter = PollingStreamAdapter(stream)
for chunk in (b'alpha', b' beta', b' gamm', b'a ome'):
self.assertEqual(chunk, adapter.poll(1.0))
