import errno import os import sys import tempfile import unittest from test.support.import_helper import import_module termios = import_module('termios') @unittest.skipUnless(hasattr(os, 'openpty'), "need os.openpty()") class TestFunctions(unittest.TestCase): def setUp(self): master_fd, self.fd = os.openpty() self.addCleanup(os.close, master_fd) self.stream = self.enterContext(open(self.fd, 'wb', buffering=0)) tmp = self.enterContext(tempfile.TemporaryFile(mode='wb', buffering=0)) self.bad_fd = tmp.fileno() def assertRaisesTermiosError(self, errno, callable, *args): with self.assertRaises(termios.error) as cm: callable(*args) self.assertEqual(cm.exception.args[0], errno) def test_tcgetattr(self): attrs = termios.tcgetattr(self.fd) self.assertIsInstance(attrs, list) self.assertEqual(len(attrs), 7) for i in range(6): self.assertIsInstance(attrs[i], int) iflag, oflag, cflag, lflag, ispeed, ospeed, cc = attrs self.assertIsInstance(cc, list) self.assertEqual(len(cc), termios.NCCS) for i, x in enumerate(cc): if ((lflag & termios.ICANON) == 0 and (i == termios.VMIN or i == termios.VTIME)): self.assertIsInstance(x, int) else: self.assertIsInstance(x, bytes) self.assertEqual(len(x), 1) self.assertEqual(termios.tcgetattr(self.stream), attrs) def test_tcgetattr_errors(self): self.assertRaisesTermiosError(errno.ENOTTY, termios.tcgetattr, self.bad_fd) self.assertRaises(ValueError, termios.tcgetattr, -1) self.assertRaises(OverflowError, termios.tcgetattr, 2**1000) self.assertRaises(TypeError, termios.tcgetattr, object()) self.assertRaises(TypeError, termios.tcgetattr) def test_tcsetattr(self): attrs = termios.tcgetattr(self.fd) termios.tcsetattr(self.fd, termios.TCSANOW, attrs) termios.tcsetattr(self.fd, termios.TCSADRAIN, attrs) termios.tcsetattr(self.fd, termios.TCSAFLUSH, attrs) termios.tcsetattr(self.stream, termios.TCSANOW, attrs) def test_tcsetattr_errors(self): attrs = termios.tcgetattr(self.fd) self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW, tuple(attrs)) self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW, attrs[:-1]) self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW, attrs + [0]) for i in range(6): attrs2 = attrs[:] attrs2[i] = 2**1000 self.assertRaises(OverflowError, termios.tcsetattr, self.fd, termios.TCSANOW, attrs2) attrs2[i] = object() self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW, attrs2) self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW, attrs[:-1] + [attrs[-1][:-1]]) self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW, attrs[:-1] + [attrs[-1] + [b'\0']]) for i in range(len(attrs[-1])): attrs2 = attrs[:] attrs2[-1] = attrs2[-1][:] attrs2[-1][i] = 2**1000 self.assertRaises(OverflowError, termios.tcsetattr, self.fd, termios.TCSANOW, attrs2) attrs2[-1][i] = object() self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW, attrs2) attrs2[-1][i] = b'' self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW, attrs2) attrs2[-1][i] = b'\0\0' self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW, attrs2) self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW, object()) self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW) self.assertRaisesTermiosError(errno.EINVAL, termios.tcsetattr, self.fd, -1, attrs) self.assertRaises(OverflowError, termios.tcsetattr, self.fd, 2**1000, attrs) self.assertRaises(TypeError, termios.tcsetattr, self.fd, object(), attrs) self.assertRaisesTermiosError(errno.ENOTTY, termios.tcsetattr, self.bad_fd, termios.TCSANOW, attrs) self.assertRaises(ValueError, termios.tcsetattr, -1, termios.TCSANOW, attrs) self.assertRaises(OverflowError, termios.tcsetattr, 2**1000, termios.TCSANOW, attrs) self.assertRaises(TypeError, termios.tcsetattr, object(), termios.TCSANOW, attrs) self.assertRaises(TypeError, termios.tcsetattr, self.fd, termios.TCSANOW) def test_tcsendbreak(self): try: termios.tcsendbreak(self.fd, 1) except termios.error as exc: if exc.args[0] == errno.ENOTTY and sys.platform.startswith('freebsd'): self.skipTest('termios.tcsendbreak() is not supported ' 'with pseudo-terminals (?) on this platform') raise termios.tcsendbreak(self.stream, 1) def test_tcsendbreak_errors(self): self.assertRaises(OverflowError, termios.tcsendbreak, self.fd, 2**1000) self.assertRaises(TypeError, termios.tcsendbreak, self.fd, 0.0) self.assertRaises(TypeError, termios.tcsendbreak, self.fd, object()) self.assertRaisesTermiosError(errno.ENOTTY, termios.tcsendbreak, self.bad_fd, 0) self.assertRaises(ValueError, termios.tcsendbreak, -1, 0) self.assertRaises(OverflowError, termios.tcsendbreak, 2**1000, 0) self.assertRaises(TypeError, termios.tcsendbreak, object(), 0) self.assertRaises(TypeError, termios.tcsendbreak, self.fd) def test_tcdrain(self): termios.tcdrain(self.fd) termios.tcdrain(self.stream) def test_tcdrain_errors(self): self.assertRaisesTermiosError(errno.ENOTTY, termios.tcdrain, self.bad_fd) self.assertRaises(ValueError, termios.tcdrain, -1) self.assertRaises(OverflowError, termios.tcdrain, 2**1000) self.assertRaises(TypeError, termios.tcdrain, object()) self.assertRaises(TypeError, termios.tcdrain) def test_tcflush(self): termios.tcflush(self.fd, termios.TCIFLUSH) termios.tcflush(self.fd, termios.TCOFLUSH) termios.tcflush(self.fd, termios.TCIOFLUSH) def test_tcflush_errors(self): self.assertRaisesTermiosError(errno.EINVAL, termios.tcflush, self.fd, -1) self.assertRaises(OverflowError, termios.tcflush, self.fd, 2**1000) self.assertRaises(TypeError, termios.tcflush, self.fd, object()) self.assertRaisesTermiosError(errno.ENOTTY, termios.tcflush, self.bad_fd, termios.TCIFLUSH) self.assertRaises(ValueError, termios.tcflush, -1, termios.TCIFLUSH) self.assertRaises(OverflowError, termios.tcflush, 2**1000, termios.TCIFLUSH) self.assertRaises(TypeError, termios.tcflush, object(), termios.TCIFLUSH) self.assertRaises(TypeError, termios.tcflush, self.fd) def test_tcflow(self): termios.tcflow(self.fd, termios.TCOOFF) termios.tcflow(self.fd, termios.TCOON) termios.tcflow(self.fd, termios.TCIOFF) termios.tcflow(self.fd, termios.TCION) def test_tcflow_errors(self): self.assertRaisesTermiosError(errno.EINVAL, termios.tcflow, self.fd, -1) self.assertRaises(OverflowError, termios.tcflow, self.fd, 2**1000) self.assertRaises(TypeError, termios.tcflow, self.fd, object()) self.assertRaisesTermiosError(errno.ENOTTY, termios.tcflow, self.bad_fd, termios.TCOON) self.assertRaises(ValueError, termios.tcflow, -1, termios.TCOON) self.assertRaises(OverflowError, termios.tcflow, 2**1000, termios.TCOON) self.assertRaises(TypeError, termios.tcflow, object(), termios.TCOON) self.assertRaises(TypeError, termios.tcflow, self.fd) def test_tcgetwinsize(self): size = termios.tcgetwinsize(self.fd) self.assertIsInstance(size, tuple) self.assertEqual(len(size), 2) self.assertIsInstance(size[0], int) self.assertIsInstance(size[1], int) self.assertEqual(termios.tcgetwinsize(self.stream), size) def test_tcgetwinsize_errors(self): self.assertRaisesTermiosError(errno.ENOTTY, termios.tcgetwinsize, self.bad_fd) self.assertRaises(ValueError, termios.tcgetwinsize, -1) self.assertRaises(OverflowError, termios.tcgetwinsize, 2**1000) self.assertRaises(TypeError, termios.tcgetwinsize, object()) self.assertRaises(TypeError, termios.tcgetwinsize) def test_tcsetwinsize(self): size = termios.tcgetwinsize(self.fd) termios.tcsetwinsize(self.fd, size) termios.tcsetwinsize(self.fd, list(size)) termios.tcsetwinsize(self.stream, size) def test_tcsetwinsize_errors(self): size = termios.tcgetwinsize(self.fd) self.assertRaises(TypeError, termios.tcsetwinsize, self.fd, size[:-1]) self.assertRaises(TypeError, termios.tcsetwinsize, self.fd, size + (0,)) self.assertRaises(TypeError, termios.tcsetwinsize, self.fd, object()) self.assertRaises(OverflowError, termios.tcsetwinsize, self.fd, (size[0], 2**1000)) self.assertRaises(TypeError, termios.tcsetwinsize, self.fd, (size[0], float(size[1]))) self.assertRaises(TypeError, termios.tcsetwinsize, self.fd, (size[0], object())) self.assertRaises(OverflowError, termios.tcsetwinsize, self.fd, (2**1000, size[1])) self.assertRaises(TypeError, termios.tcsetwinsize, self.fd, (float(size[0]), size[1])) self.assertRaises(TypeError, termios.tcsetwinsize, self.fd, (object(), size[1])) self.assertRaisesTermiosError(errno.ENOTTY, termios.tcsetwinsize, self.bad_fd, size) self.assertRaises(ValueError, termios.tcsetwinsize, -1, size) self.assertRaises(OverflowError, termios.tcsetwinsize, 2**1000, size) self.assertRaises(TypeError, termios.tcsetwinsize, object(), size) self.assertRaises(TypeError, termios.tcsetwinsize, self.fd) class TestModule(unittest.TestCase): def test_constants(self): self.assertIsInstance(termios.B0, int) self.assertIsInstance(termios.B38400, int) self.assertIsInstance(termios.TCSANOW, int) self.assertIsInstance(termios.TCSADRAIN, int) self.assertIsInstance(termios.TCSAFLUSH, int) self.assertIsInstance(termios.TCIFLUSH, int) self.assertIsInstance(termios.TCOFLUSH, int) self.assertIsInstance(termios.TCIOFLUSH, int) self.assertIsInstance(termios.TCOOFF, int) self.assertIsInstance(termios.TCOON, int) self.assertIsInstance(termios.TCIOFF, int) self.assertIsInstance(termios.TCION, int) self.assertIsInstance(termios.VTIME, int) self.assertIsInstance(termios.VMIN, int) self.assertIsInstance(termios.NCCS, int) self.assertLess(termios.VTIME, termios.NCCS) self.assertLess(termios.VMIN, termios.NCCS) def test_exception(self): self.assertTrue(issubclass(termios.error, Exception)) self.assertFalse(issubclass(termios.error, OSError)) if __name__ == '__main__': unittest.main()