From 628c65e1c256d8817f9753251290e2a31b87fd95 Mon Sep 17 00:00:00 2001
From: Andrew Kay <git@ajk.me>
Date: Sat, 14 Sep 2019 07:57:15 -0500
Subject: [PATCH] Add assertion for resetting dirty tracking on flush

---
 tests/test_tn3270.py | 2 ++
 tests/test_vt100.py  | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/tests/test_tn3270.py b/tests/test_tn3270.py
index cd8a25c..c97209a 100644
--- a/tests/test_tn3270.py
+++ b/tests/test_tn3270.py
@@ -63,6 +63,8 @@ class SessionHandleHostTestCase(unittest.TestCase):
 
         self.assertEqual(self.terminal.display.cursor_index, 8)
 
+        self.assertFalse(self.session.emulator.dirty)
+
     def test_eof(self):
         # Arrange
         self.session.emulator.update = Mock(side_effect=EOFError)
diff --git a/tests/test_vt100.py b/tests/test_vt100.py
index 8b10ce7..64781c4 100644
--- a/tests/test_vt100.py
+++ b/tests/test_vt100.py
@@ -35,6 +35,8 @@ class SessionHandleHostTestCase(unittest.TestCase):
 
         terminal.display.move_cursor.assert_called_with(row=0, column=3)
 
+        self.assertFalse(session.vt100_screen.dirty)
+
 class SessionHandleKeyTestCase(unittest.TestCase):
     def setUp(self):
         self.terminal = Mock()