From 57f2b7dff68805f5150a551b1e79d1389f4cad97 Mon Sep 17 00:00:00 2001
From: Andrew Kay <git@ajk.me>
Date: Wed, 3 Jul 2019 23:05:07 -0500
Subject: [PATCH] vt100 unit tests

---
 tests/test_display.py |  2 +-
 tests/test_vt100.py   | 88 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 89 insertions(+), 1 deletion(-)
 create mode 100644 tests/test_vt100.py

diff --git a/tests/test_display.py b/tests/test_display.py
index 4583100..cbba3d3 100644
--- a/tests/test_display.py
+++ b/tests/test_display.py
@@ -1,5 +1,5 @@
 import unittest
-from unittest.mock import Mock, patch
+from unittest.mock import Mock
 
 import context
 
diff --git a/tests/test_vt100.py b/tests/test_vt100.py
new file mode 100644
index 0000000..c4df9d2
--- /dev/null
+++ b/tests/test_vt100.py
@@ -0,0 +1,88 @@
+import unittest
+from unittest.mock import Mock, patch
+
+import context
+
+from oec.display import Dimensions
+from oec.keyboard import Key, KeyboardModifiers
+from oec.vt100 import VT100Session, select
+
+class SessionHandleHostTestCase(unittest.TestCase):
+    def test(self):
+        # Arrange
+        terminal = Mock()
+
+        terminal.display.dimensions = Dimensions(24, 80)
+
+        session = VT100Session(terminal, None)
+
+        session.host_process = Mock()
+
+        session.host_process.read = Mock(return_value=b'abc')
+
+        # Act
+        with patch('oec.vt100.select') as select_patch:
+            select_patch.return_value = [[session.host_process]]
+
+            session.handle_host()
+
+        # Assert
+        terminal.display.buffered_write.assert_any_call(0x80, 0, 0)
+        terminal.display.buffered_write.assert_any_call(0x81, 0, 1)
+        terminal.display.buffered_write.assert_any_call(0x82, 0, 2)
+
+        terminal.display.flush.assert_called()
+
+        terminal.display.move_cursor.assert_called_with(0, 3)
+
+class SessionHandleKeyTestCase(unittest.TestCase):
+    def setUp(self):
+        self.terminal = Mock()
+
+        self.terminal.display.dimensions = Dimensions(24, 80)
+
+        self.session = VT100Session(self.terminal, None)
+
+        self.session.host_process = Mock()
+
+    def test_printable(self):
+        # Act
+        self.session.handle_key(Key.LOWER_A, KeyboardModifiers.NONE, None)
+
+        # Assert
+        self.session.host_process.write.assert_called_with(b'a')
+
+    # TODO: Test the unprintable branch.
+
+    def test_mapped(self):
+        # Act
+        self.session.handle_key(Key.ENTER, KeyboardModifiers.NONE, None)
+
+        # Assert
+        self.session.host_process.write.assert_called_with(b'\r')
+
+    def test_mapped_alt_modifier(self):
+        # Act
+        self.session.handle_key(Key.LOWER_C, KeyboardModifiers.LEFT_ALT, None)
+
+        # Assert
+        self.session.host_process.write.assert_called_with(b'\x03')
+
+    def test_unmapped_alt_modifier(self):
+        # Arrange
+        self.session.logger = Mock()
+
+        # Act
+        self.session.handle_key(Key.THREE, KeyboardModifiers.LEFT_ALT, None)
+
+        # Assert
+        self.session.host_process.write.assert_not_called()
+
+        self.session.logger.warning.assert_called()
+
+    def test_alt_modifier_with_modifier_key(self):
+        # Act
+        self.session.handle_key(Key.LEFT_ALT, KeyboardModifiers.LEFT_ALT, None)
+
+        # Assert
+        self.session.host_process.write.assert_not_called()