def test_run_with_expect_and_closed_socket(self):
node = ExpectClose()
runner = Runner(node)
runner.state.msg_sock = mock.MagicMock()
runner.state.msg_sock.recvMessageBlocking = \
mock.MagicMock(side_effect=TLSAbruptCloseError())
runner.run()
def test_run_with_expect_and_unexpected_closed_socket(self):
node = mock.MagicMock()
node.is_command = mock.Mock(return_value=False)
node.is_expect = mock.Mock(return_value=True)
node.is_generator = mock.Mock(return_value=False)
node.child = None
runner = Runner(node)
runner.state.msg_sock = mock.MagicMock()
runner.state.msg_sock.recvMessageBlocking = \
mock.MagicMock(side_effect=TLSAbruptCloseError())
with self.assertRaises(AssertionError):
runner.run()