Skip to content

Commit bb70322

Browse files
pythongh-110388: Add tests for tty
1 parent d8c00d2 commit bb70322

File tree

2 files changed

+79
-0
lines changed

2 files changed

+79
-0
lines changed

Lib/test/test_tty.py

Lines changed: 78 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,78 @@
1+
import io
2+
import unittest
3+
from test.support.import_helper import import_module
4+
5+
termios = import_module('termios')
6+
tty = import_module('tty')
7+
8+
9+
class TestTty(unittest.TestCase):
10+
11+
def setUp(self):
12+
self.stream = open('/dev/tty', 'wb', buffering=0)
13+
self.addCleanup(self.stream.close)
14+
self.fd = self.stream.fileno()
15+
self.mode = termios.tcgetattr(self.fd)
16+
self.addCleanup(termios.tcsetattr, self.fd, termios.TCSANOW, self.mode)
17+
self.addCleanup(termios.tcsetattr, self.fd, termios.TCSAFLUSH, self.mode)
18+
19+
def check_cbreak(self, mode):
20+
self.assertEqual(mode[0] & termios.ICRNL, 0)
21+
self.assertEqual(mode[3] & termios.ECHO, 0)
22+
self.assertEqual(mode[3] & termios.ICANON, 0)
23+
self.assertEqual(mode[6][termios.VMIN], 1)
24+
self.assertEqual(mode[6][termios.VTIME], 0)
25+
26+
def check_raw(self, mode):
27+
self.check_cbreak(mode)
28+
self.assertEqual(mode[0] & termios.ISTRIP, 0)
29+
self.assertEqual(mode[0] & termios.ICRNL, 0)
30+
self.assertEqual(mode[1] & termios.OPOST, 0)
31+
self.assertEqual(mode[2] & termios.PARENB, termios.CS8 & termios.PARENB)
32+
self.assertEqual(mode[2] & termios.CSIZE, termios.CS8 & termios.CSIZE)
33+
self.assertEqual(mode[2] & termios.CS8, termios.CS8)
34+
self.assertEqual(mode[3] & termios.ECHO, 0)
35+
self.assertEqual(mode[3] & termios.ICANON, 0)
36+
self.assertEqual(mode[3] & termios.ISIG, 0)
37+
self.assertEqual(mode[6][termios.VMIN], 1)
38+
self.assertEqual(mode[6][termios.VTIME], 0)
39+
40+
def test_cfmakeraw(self):
41+
mode = termios.tcgetattr(self.fd)
42+
self.assertEqual(mode, self.mode)
43+
tty.cfmakeraw(mode)
44+
self.check_raw(mode)
45+
self.assertEqual(mode[4], self.mode[4])
46+
self.assertEqual(mode[5], self.mode[5])
47+
48+
def test_cfmakecbreak(self):
49+
mode = termios.tcgetattr(self.fd)
50+
self.assertEqual(mode, self.mode)
51+
tty.cfmakecbreak(mode)
52+
self.check_cbreak(mode)
53+
self.assertEqual(mode[1], self.mode[1])
54+
self.assertEqual(mode[2], self.mode[2])
55+
self.assertEqual(mode[4], self.mode[4])
56+
self.assertEqual(mode[5], self.mode[5])
57+
58+
def test_setraw(self):
59+
mode = tty.setraw(self.fd)
60+
mode2 = termios.tcgetattr(self.fd)
61+
self.check_raw(mode2)
62+
mode3 = tty.setraw(self.fd, termios.TCSANOW)
63+
self.assertEqual(mode3, mode2)
64+
tty.setraw(self.stream)
65+
tty.setraw(fd=self.fd, when=termios.TCSANOW)
66+
67+
def test_setcbreak(self):
68+
mode = tty.setcbreak(self.fd)
69+
mode2 = termios.tcgetattr(self.fd)
70+
self.check_cbreak(mode2)
71+
mode3 = tty.setcbreak(self.fd, termios.TCSANOW)
72+
self.assertEqual(mode3, mode2)
73+
tty.setcbreak(self.stream)
74+
tty.setcbreak(fd=self.fd, when=termios.TCSANOW)
75+
76+
77+
if __name__ == '__main__':
78+
unittest.main()
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Add tests for :mod:`tty`.

0 commit comments

Comments
 (0)