def test_should_restart_driver(self): buf = ScreenBuffer(page_size=2, buffer_size=5) drv = ScreenBufferTest.FakeDriver(self.queue) buf.start(drv) self.queue.push(None) queue2 = ScreenBufferTest.Queue() drv2 = ScreenBufferTest.FakeDriver(queue2) buf.restart(drv2) queue2.push(None) buf.stop()
def test_should_clear_existing_records_on_restart(self): drv = ScreenBufferTest.FakeDriver(self.queue) buf = ScreenBuffer(page_size=2, buffer_size=5) buf.start(drv) self.queue.push_backward_records(2, 2) self.queue.wait() queue2 = ScreenBufferTest.Queue() drv2 = ScreenBufferTest.FakeDriver(queue2) buf.restart(drv2) queue2.push_none_and_wait() try: self.assertEqual(0, len(buf.get_current_lines())) finally: buf.stop()