1 files changed, 72 insertions, 0 deletions
diff --git a/tests/test_socket_expect.py b/tests/test_socket_expect.py
new file mode 100644
index 0000000..8b22f3f
--- /dev/null
+++ b/tests/test_socket_expect.py
@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+'''
+PEXPECT LICENSE
+
+    This license is approved by the OSI and FSF as GPL-compatible.
+        http://opensource.org/licenses/isc-license.txt
+
+    Copyright (c) 2012, Noah Spurrier <noah@noah.org>
+    PERMISSION TO USE, COPY, MODIFY, AND/OR DISTRIBUTE THIS SOFTWARE FOR ANY
+    PURPOSE WITH OR WITHOUT FEE IS HEREBY GRANTED, PROVIDED THAT THE ABOVE
+    COPYRIGHT NOTICE AND THIS PERMISSION NOTICE APPEAR IN ALL COPIES.
+    THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+    WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+    MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+    ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+    WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+    ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+    OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+
+'''
+import pexpect
+from pexpect import socket_expect
+import unittest
+from . import PexpectTestCase
+import socket
+
+def open_file_socket(filename):
+    read_socket, write_socket = socket.socketpair()
+    with open(filename, "rb") as file:
+        write_socket.sendall(file.read())
+    write_socket.close()
+    return read_socket
+
+class ExpectTestCase(PexpectTestCase.PexpectTestCase):
+    def setUp(self):
+        print(self.id())
+        PexpectTestCase.PexpectTestCase.setUp(self)
+
+    def test_socket (self):
+        socket = open_file_socket('TESTDATA.txt')
+        s = socket_expect.SocketSpawn(socket)
+        s.expect(b'This is the end of test data:')
+        s.expect(pexpect.EOF)
+        self.assertEqual(s.before, b' END\n')
+
+    def test_maxread (self):
+        socket = open_file_socket('TESTDATA.txt')
+        s = socket_expect.SocketSpawn(socket)
+        s.maxread = 100
+        s.expect('2')
+        s.expect ('This is the end of test data:')
+        s.expect (pexpect.EOF)
+        self.assertEqual(s.before, b' END\n')
+
+    def test_socket_isalive (self):
+        socket = open_file_socket('TESTDATA.txt')
+        s = socket_expect.SocketSpawn(socket)
+        assert s.isalive()
+        s.close()
+        assert not s.isalive(), "Should not be alive after close()"
+
+    def test_socket_isatty (self):
+        socket = open_file_socket('TESTDATA.txt')
+        s = socket_expect.SocketSpawn(socket)
+        assert not s.isatty()
+        s.close()
+
+
+if __name__ == '__main__':
+    unittest.main()
+
+suite = unittest.TestLoader().loadTestsFromTestCase(ExpectTestCase)