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))