diff options
-rw-r--r-- | tests/test_socket.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tests/test_socket.py b/tests/test_socket.py index 4f100ed..56d69c7 100644 --- a/tests/test_socket.py +++ b/tests/test_socket.py @@ -153,6 +153,20 @@ class ExpectTestCase(PexpectTestCase.PexpectTestCase): session.expect(pexpect.EOF) self.assertEqual(session.before, b'') + def test_socket_with_write(self): + sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + sock.connect((self.host, self.port)) + session = fdpexpect.fdspawn(sock.fileno(), timeout=10) + session.expect(self.prompt1) + self.assertEqual(session.before, self.motd) + session.write(self.enter) + session.expect(self.prompt2) + session.write(self.enter) + session.expect(self.prompt3) + session.write(self.exit) + session.expect(pexpect.EOF) + self.assertEqual(session.before, b'') + def test_not_int(self): with self.assertRaises(pexpect.ExceptionPexpect): session = fdpexpect.fdspawn('bogus', timeout=10) |