1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
#!/usr/bin/python3
# Validate Terminal handling
#
# Copyright (C) 2018 Canonical, Ltd.
# Author: Mathieu Trudel-Lapierre <mathieu.trudel-lapierre@canonical.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 3.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import fcntl
import sys
import os
import termios
import unittest
import netplan.terminal
@unittest.skipUnless(sys.__stdin__.isatty(), "not supported when run from a script")
class TestTerminal(unittest.TestCase):
def setUp(self):
self.terminal = netplan.terminal.Terminal(sys.stdin.fileno())
def test_echo(self):
self.terminal.disable_echo()
attrs = termios.tcgetattr(self.terminal.fd)
self.assertFalse(attrs[3] & termios.ECHO)
self.terminal.enable_echo()
attrs = termios.tcgetattr(self.terminal.fd)
self.assertTrue(attrs[3] & termios.ECHO)
def test_nonblocking_io(self):
orig_flags = flags = fcntl.fcntl(self.terminal.fd, fcntl.F_GETFL)
self.assertFalse(flags & os.O_NONBLOCK)
self.terminal.enable_nonblocking_io()
flags = fcntl.fcntl(self.terminal.fd, fcntl.F_GETFL)
self.assertTrue(flags & os.O_NONBLOCK)
self.assertNotEqual(flags, orig_flags)
self.terminal.disable_nonblocking_io()
flags = fcntl.fcntl(self.terminal.fd, fcntl.F_GETFL)
self.assertFalse(flags & os.O_NONBLOCK)
self.assertEqual(flags, orig_flags)
def test_save(self):
self.terminal.enable_nonblocking_io()
flags = self.terminal.orig_flags
self.terminal.save()
self.terminal.disable_nonblocking_io()
self.assertNotEqual(flags, self.terminal.orig_flags)
self.terminal.reset()
flags = fcntl.fcntl(self.terminal.fd, fcntl.F_GETFL)
self.assertEqual(flags, self.terminal.orig_flags)
self.terminal.disable_nonblocking_io()
self.terminal.save()
def test_save_and_restore_with_dict(self):
self.terminal.enable_nonblocking_io()
orig_settings = dict()
self.terminal.save(orig_settings)
self.terminal.disable_nonblocking_io()
flags = fcntl.fcntl(self.terminal.fd, fcntl.F_GETFL)
self.assertNotEqual(flags, orig_settings.get('flags'))
self.terminal.reset(orig_settings)
flags = fcntl.fcntl(self.terminal.fd, fcntl.F_GETFL)
self.assertEqual(flags, orig_settings.get('flags'))
self.terminal.disable_nonblocking_io()
def test_reset(self):
self.terminal.enable_nonblocking_io()
flags = fcntl.fcntl(self.terminal.fd, fcntl.F_GETFL)
self.assertTrue(flags & os.O_NONBLOCK)
self.terminal.reset()
flags = fcntl.fcntl(self.terminal.fd, fcntl.F_GETFL)
self.assertFalse(flags & os.O_NONBLOCK)
|
