github.com/lowobservable.oec
1
0
Fork 0
mirror of https://github.com/lowobservable/oec.git synced 2026-04-16 08:26:25 +00:00
Code Issues Projects Releases Wiki Activity

Disconnect session on ConnectionResetError

Browse Source
This commit is contained in:
Andrew Kay
2020-03-26 07:17:43 -05:00
parent cc11fe2ed6
commit 52a05ad1df
2 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_tn3270.py
View File

@@ -75,6 +75,16 @@ class SessionHandleHostTestCase(unittest.TestCase):
self.telnet.close.assert_called()
def test_connection_reset(self):
# Arrange
self.session.emulator.update = Mock(side_effect=ConnectionResetError)
# Act and assert
with self.assertRaises(SessionDisconnectedError):
self.session.handle_host()
self.telnet.close.assert_called()
def test_keyboard_locked(self):
# Arrange
self.session.emulator.update = Mock(return_value=True)