def test_should_not_stop_fetching_if_forward_fetch_returns_fewer_records(
self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
drv = ScreenBufferTest.FakeDriver(self.queue)
self.queue.push_backward_records(14, 7)
buf.get_records(drv)
buf.go_to_previous_line()
self.queue.push(None)
buf.get_records(drv)
buf.go_to_previous_page()
buf.go_to_previous_page()
self.queue.push_backward_records(7, 7)
buf.get_records(drv)
buf.go_to_previous_page()
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('6', cur[0].message)
self.assertEqual('7', cur[1].message)
def test_should_append_record_on_non_empty_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(2))
buf.append_record(self._get_line(3))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('2', cur[0].message)
self.assertEqual('3', cur[1].message)
def test_should_append_multi_line_record_spanning_over_end_of_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(1))
buf.append_record(self._get_line(2, 'a\nb'))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('a', cur[0].message)
self.assertEqual('b', cur[1].message)
def test_should_append_multi_line_record_spanning_over_end_of_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(1))
buf.append_record(self._get_line(2, 'a\nb'))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('a', cur[0].message)
self.assertEqual('b', cur[1].message)
def test_should_append_record_on_non_empty_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(2))
buf.append_record(self._get_line(3))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('2', cur[0].message)
self.assertEqual('3', cur[1].message)
def test_should_append_record_on_empty_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.add_observer(self.observer.notify)
buf.append_record(self._get_line(2))
self.assertEqual(1, self.observer.count)
cur = buf.get_current_lines()
self.assertEqual(1, len(cur))
self.assertEqual('2', cur[0].message)
def test_should_prepend_multi_line_record_on_empty_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.prepend_record(self._get_line(1, 'a\nb'))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('a', cur[0].message)
self.assertEqual(False, cur[0].is_continuation)
self.assertEqual('b', cur[1].message)
self.assertEqual(True, cur[1].is_continuation)
def test_should_change_page_size_at_end_of_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
for i in range(10, 0, -1):
buf.prepend_record(self._get_line(i))
buf.page_size = 3
cur = buf.get_current_lines()
self.assertEqual(3, len(cur))
self.assertEqual('8', cur[0].message)
def test_should_prepend_multi_line_record_on_empty_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.prepend_record(self._get_line(1, 'a\nb'))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('a', cur[0].message)
self.assertEqual(False, cur[0].is_continuation)
self.assertEqual('b', cur[1].message)
self.assertEqual(True, cur[1].is_continuation)
def test_should_go_to_next_page_with_less_lines_than_page_size(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(1))
buf.go_to_next_page()
cur = buf.get_current_lines()
self.assertEqual(1, len(cur))
self.assertEqual('1', cur[0].message)
def test_should_append_record_on_empty_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.add_observer(self.observer.notify)
buf.append_record(self._get_line(2))
self.assertEqual(1, self.observer.count)
cur = buf.get_current_lines()
self.assertEqual(1, len(cur))
self.assertEqual('2', cur[0].message)
def test_should_go_to_next_page_with_less_lines_than_page_size(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(1))
buf.go_to_next_page()
cur = buf.get_current_lines()
self.assertEqual(1, len(cur))
self.assertEqual('1', cur[0].message)
def test_should_change_page_size_at_end_of_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
for i in range(10, 0, -1):
buf.prepend_record(self._get_line(i))
buf.page_size = 3
cur = buf.get_current_lines()
self.assertEqual(3, len(cur))
self.assertEqual('8', cur[0].message)
def test_should_go_to_previous_line(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
for i in range(10, 0, -1):
buf.prepend_record(self._get_line(i))
buf.go_to_previous_line()
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('8', cur[0].message)
def test_should_go_to_previous_line(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
for i in range(10, 0, -1):
buf.prepend_record(self._get_line(i))
buf.go_to_previous_line()
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('8', cur[0].message)
def test_should_prepend_record_on_buffer_not_at_end_of_screen(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(2))
buf.append_record(self._get_line(3))
buf.append_record(self._get_line(4))
buf.go_to_previous_line()
buf.prepend_record(self._get_line(1))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('2', cur[0].message)
self.assertEqual('3', cur[1].message)
def test_should_fetch_records_in_descending_order(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
self.queue.push_backward_records(2, 2)
drv = ScreenBufferTest.FakeDriver(self.queue)
buf.get_records(drv)
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('1', cur[0].message)
self.assertEqual('2', cur[1].message)
def test_should_prepend_record_on_buffer_not_at_end_of_screen(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(2))
buf.append_record(self._get_line(3))
buf.append_record(self._get_line(4))
buf.go_to_previous_line()
buf.prepend_record(self._get_line(1))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('2', cur[0].message)
self.assertEqual('3', cur[1].message)
def test_should_fetch_records_in_descending_order(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
self.queue.push_backward_records(2, 2)
drv = ScreenBufferTest.FakeDriver(self.queue)
buf.get_records(drv)
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('1', cur[0].message)
self.assertEqual('2', cur[1].message)
def test_should_prepend_record_on_off_screen_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.prepend_record(self._get_line(3))
buf.prepend_record(self._get_line(2))
buf.add_observer(self.observer.notify)
buf.prepend_record(self._get_line(1))
self.assertEqual(0, self.observer.count)
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('2', cur[0].message)
self.assertEqual('3', cur[1].message)
def test_should_prepend_record_on_off_screen_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.prepend_record(self._get_line(3))
buf.prepend_record(self._get_line(2))
buf.add_observer(self.observer.notify)
buf.prepend_record(self._get_line(1))
self.assertEqual(0, self.observer.count)
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('2', cur[0].message)
self.assertEqual('3', cur[1].message)
def test_should_auto_scroll_on_append_if_screen_is_at_end_of_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(2))
buf.append_record(self._get_line(3))
buf.add_observer(self.observer.notify)
buf.append_record(self._get_line(4))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('3', cur[0].message)
self.assertEqual('4', cur[1].message)
self.assertEqual(1, self.observer.count)
def test_should_not_auto_scroll_if_starting_driver_with_date(self):
dt = datetime.datetime.utcnow()
buf = ScreenBuffer(page_size=2, buffer_size=5)
self.queue.push(13)
self.queue.push(None)
self.queue.push_forward_records(13, 7)
self.queue.push_forward_records(12, 5)
drv = ScreenBufferTest.FakeDriver(self.queue, dt)
buf.get_records(drv)
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('13', cur[0].message)
self.assertEqual('14', cur[1].message)
def test_should_not_auto_scroll_if_starting_driver_with_date(self):
dt = datetime.datetime.utcnow()
buf = ScreenBuffer(page_size=2, buffer_size=5)
self.queue.push(13)
self.queue.push(None)
self.queue.push_forward_records(13, 7)
self.queue.push_forward_records(12, 5)
drv = ScreenBufferTest.FakeDriver(self.queue, dt)
buf.get_records(drv)
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('13', cur[0].message)
self.assertEqual('14', cur[1].message)
def test_should_auto_scroll_on_append_if_screen_is_at_end_of_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(2))
buf.append_record(self._get_line(3))
buf.add_observer(self.observer.notify)
buf.append_record(self._get_line(4))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('3', cur[0].message)
self.assertEqual('4', cur[1].message)
self.assertEqual(1, self.observer.count)
def test_should_not_auto_scroll_if_not_at_end_of_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(2))
buf.append_record(self._get_line(3))
buf.append_record(self._get_line(4))
buf.go_to_previous_line()
buf.add_observer(self.observer.notify)
buf.append_record(self._get_line(5))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('2', cur[0].message)
self.assertEqual(0, self.observer.count)
def test_should_not_auto_scroll_if_not_at_end_of_buffer(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
buf.append_record(self._get_line(2))
buf.append_record(self._get_line(3))
buf.append_record(self._get_line(4))
buf.go_to_previous_line()
buf.add_observer(self.observer.notify)
buf.append_record(self._get_line(5))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('2', cur[0].message)
self.assertEqual(0, self.observer.count)
def test_should_fetch_records_from_thread(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()
cur = buf.get_current_lines()
buf.stop()
self.assertTrue(drv.stopped)
self.assertFalse(drv.error)
self.assertEqual((None, True, 7), drv.instruction)
self.assertEqual(2, len(cur))
self.assertEqual('1', cur[0].message)
self.assertEqual('2', cur[1].message)
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()
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()
def test_should_fetch_records_from_thread(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()
cur = buf.get_current_lines()
buf.stop()
self.assertTrue(drv.stopped)
self.assertFalse(drv.error)
self.assertEqual((None, True, 7), drv.instruction)
self.assertEqual(2, len(cur))
self.assertEqual('1', cur[0].message)
self.assertEqual('2', cur[1].message)
def test_should_auto_scroll_after_manual_scroll(self):
dt = datetime.datetime.utcnow()
buf = ScreenBuffer(page_size=2, buffer_size=5)
self.queue.push(13)
self.queue.push(None)
self.queue.push_forward_records(13, 7)
self.queue.push_forward_records(12, 5)
drv = ScreenBufferTest.FakeDriver(self.queue, dt)
buf.get_records(drv)
buf.go_to_next_page()
buf.go_to_next_page()
buf.go_to_next_line()
buf.append_record(self._get_line(20))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('19', cur[0].message)
self.assertEqual('20', cur[1].message)
def test_should_stop_fetching_if_fetch_returns_less_records_than_asked(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
drv = ScreenBufferTest.FakeDriver(self.queue)
self.queue.push_backward_records(6, 6)
buf.get_records(drv)
buf.go_to_previous_page()
buf.go_to_previous_page()
buf.get_records(drv)
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('1', cur[0].message)
self.assertEqual('2', cur[1].message)
def test_should_auto_scroll_after_manual_scroll(self):
dt = datetime.datetime.utcnow()
buf = ScreenBuffer(page_size=2, buffer_size=5)
self.queue.push(13)
self.queue.push(None)
self.queue.push_forward_records(13, 7)
self.queue.push_forward_records(12, 5)
drv = ScreenBufferTest.FakeDriver(self.queue, dt)
buf.get_records(drv)
buf.go_to_next_page()
buf.go_to_next_page()
buf.go_to_next_line()
buf.append_record(self._get_line(20))
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('19', cur[0].message)
self.assertEqual('20', cur[1].message)
def test_should_stop_fetching_if_fetch_returns_less_records_than_asked(
self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
drv = ScreenBufferTest.FakeDriver(self.queue)
self.queue.push_backward_records(6, 6)
buf.get_records(drv)
buf.go_to_previous_page()
buf.go_to_previous_page()
buf.get_records(drv)
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('1', cur[0].message)
self.assertEqual('2', cur[1].message)
def test_should_not_stop_fetching_if_fetch_returns_correct_number_of_records(self):
buf = ScreenBuffer(page_size=2, buffer_size=5)
drv = ScreenBufferTest.FakeDriver(self.queue)
self.queue.push_backward_records(14, 7)
buf.get_records(drv)
buf.go_to_previous_page()
buf.go_to_previous_page()
self.queue.push_backward_records(7, 7)
buf.get_records(drv)
buf.go_to_previous_page()
cur = buf.get_current_lines()
self.assertEqual(2, len(cur))
self.assertEqual('7', cur[0].message)
self.assertEqual('8', cur[1].message)
