import unittest from unittest.mock import Mock import context from oec.display import Dimensions, Display, encode_ascii_character, encode_ebcdic_character, encode_string class DisplayMoveCursorTestCase(unittest.TestCase): def setUp(self): self.interface = Mock() dimensions = Dimensions(24, 80) self.display = Display(self.interface, dimensions) def test(self): # Act self.display.move_cursor(index=815) # Assert self.assertEqual(self.display.address_counter, 895) self.interface.offload_load_address_counter.assert_called_with(895) def test_with_row_and_column(self): # Act self.display.move_cursor(row=10, column=15) # Assert self.assertEqual(self.display.address_counter, 895) self.interface.offload_load_address_counter.assert_called_with(895) def test_no_change(self): # Arrange self.display.move_cursor(index=0) self.interface.offload_load_address_counter.reset_mock() # Act self.display.move_cursor(index=0) # Assert self.assertEqual(self.display.address_counter, 80) self.interface.offload_load_address_counter.assert_not_called() def test_no_change_force(self): # Arrange self.display.move_cursor(index=0) self.interface.offload_load_address_counter.reset_mock() # Act self.display.move_cursor(index=0, force_load=True) # Assert self.assertEqual(self.display.address_counter, 80) self.interface.offload_load_address_counter.assert_called_with(80) class DisplayBufferedWriteTestCase(unittest.TestCase): def setUp(self): self.interface = Mock() dimensions = Dimensions(24, 80) self.display = Display(self.interface, dimensions) def test(self): # Act self.display.buffered_write(0x01, index=15) self.display.buffered_write(0x02, index=97) # Assert self.assertEqual(self.display.buffer[15], 0x01) self.assertEqual(self.display.buffer[97], 0x02) self.assertSequenceEqual(self.display.dirty, [15, 97]) def test_with_row_and_column(self): # Act self.display.buffered_write(0x01, row=0, column=15) self.display.buffered_write(0x02, row=1, column=17) # Assert self.assertEqual(self.display.buffer[15], 0x01) self.assertEqual(self.display.buffer[97], 0x02) self.assertSequenceEqual(self.display.dirty, [15, 97]) def test_change(self): self.assertTrue(self.display.buffered_write(0x01, index=0)) self.assertTrue(self.display.buffered_write(0x02, index=0)) self.assertEqual(self.display.buffer[0], 0x02) self.assertSequenceEqual(self.display.dirty, [0]) def test_no_change(self): self.assertTrue(self.display.buffered_write(0x01, index=0)) self.assertFalse(self.display.buffered_write(0x01, index=0)) self.assertEqual(self.display.buffer[0], 0x01) self.assertSequenceEqual(self.display.dirty, [0]) class DisplayFlushTestCase(unittest.TestCase): def setUp(self): self.interface = Mock() dimensions = Dimensions(24, 80) self.display = Display(self.interface, dimensions) self.display._flush_range = Mock() def test_no_changes(self): # Act self.display.flush() # Assert self.display._flush_range.assert_not_called() def test_single_range(self): # Arrange self.display.buffered_write(0x01, index=0) self.display.buffered_write(0x02, index=1) self.display.buffered_write(0x03, index=2) # Act self.display.flush() # Assert self.display._flush_range.assert_called_with(0, 2) def test_multiple_ranges(self): # Arrange self.display.buffered_write(0x01, index=0) self.display.buffered_write(0x02, index=1) self.display.buffered_write(0x03, index=2) self.display.buffered_write(0x05, index=30) self.display.buffered_write(0x06, index=31) self.display.buffered_write(0x04, index=20) # Act self.display.flush() # Assert self.display._flush_range.assert_called_with(0, 31) class DisplayClearTestCase(unittest.TestCase): def setUp(self): self.interface = Mock() dimensions = Dimensions(24, 80) self.display = Display(self.interface, dimensions) def test_excluding_status_line(self): # Arrange self.display.buffered_write(0x01, index=0) self.assertEqual(self.display.buffer[0], 0x01) self.assertTrue(self.display.dirty) # Act self.display.clear(clear_status_line=False) # Assert self.interface.offload_write.assert_called_with(b'\x00', address=80, repeat=1919) self.interface.offload_load_address_counter.assert_called_with(80) self.assertEqual(self.display.buffer[0], 0x00) self.assertFalse(self.display.dirty) def test_including_status_line(self): # Arrange self.display.buffered_write(0x01, index=0) self.assertEqual(self.display.buffer[0], 0x01) self.assertTrue(self.display.dirty) # Act self.display.clear(clear_status_line=True) # Assert self.interface.offload_write.assert_called_with(b'\x00', address=0, repeat=1999) self.interface.offload_load_address_counter.assert_called_with(80) self.assertEqual(self.display.buffer[0], 0x00) self.assertFalse(self.display.dirty) class DisplayFlushRangeTestCase(unittest.TestCase): def setUp(self): self.interface = Mock() dimensions = Dimensions(24, 80) self.display = Display(self.interface, dimensions) def test_when_start_address_is_current_address_counter(self): # Arrange self.display.move_cursor(index=0) self.display.buffered_write(0x01, index=0) self.display.buffered_write(0x02, index=1) self.display.buffered_write(0x03, index=2) # Act self.display.flush() # Assert self.interface.offload_write.assert_called_with(bytes.fromhex('01 02 03'), address=None) self.assertEqual(self.display.address_counter, 83) self.assertFalse(self.display.dirty) def test_when_start_address_is_not_current_address_counter(self): # Arrange self.display.move_cursor(index=70) self.display.buffered_write(0x01, index=0) self.display.buffered_write(0x02, index=1) self.display.buffered_write(0x03, index=2) # Act self.display.flush() # Assert self.interface.offload_write.assert_called_with(bytes.fromhex('01 02 03'), address=80) self.assertEqual(self.display.address_counter, 83) self.assertFalse(self.display.dirty) class EncodeAsciiCharacterTestCase(unittest.TestCase): def test_mapped_character(self): self.assertEqual(encode_ascii_character(ord('a')), 0x80) def test_unmapped_character(self): self.assertEqual(encode_ascii_character(ord('`')), 0x00) def test_out_of_range(self): self.assertEqual(encode_ascii_character(ord('✓')), 0x00) class EncodeEbcdicCharacterTestCase(unittest.TestCase): def test_mapped_character(self): self.assertEqual(encode_ebcdic_character(129), 0x80) def test_unmapped_character(self): self.assertEqual(encode_ebcdic_character(185), 0x00) def test_out_of_range(self): self.assertEqual(encode_ebcdic_character(256), 0x00) class EncodeStringTestCase(unittest.TestCase): def test_mapped_characters(self): self.assertEqual(encode_string('Hello, world!'), bytes.fromhex('a7 84 8b 8b 8e 33 00 96 8e 91 8b 83 19')) def test_unmapped_characters(self): self.assertEqual(encode_string('Everything ✓'), bytes.fromhex('a4 95 84 91 98 93 87 88 8d 86 00 18'))