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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
import os
import signal
import sys
import threading
import time
import unittest
from unittest.mock import Mock
from unittest import skip, skipUnless
from curtsies import events
from curtsies.input import Input
class CustomEvent(events.Event):
pass
class CustomScheduledEvent(events.ScheduledEvent):
pass
@skipUnless(sys.stdin.isatty(), "stdin must be a tty")
class TestInput(unittest.TestCase):
def test_create(self):
Input()
def test_iter(self):
inp = Input()
inp.send = Mock()
inp.send.return_value = None
for i, e in zip(range(3), inp):
self.assertEqual(e, None)
self.assertEqual(inp.send.call_count, 3)
def test_send(self):
inp = Input()
inp.unprocessed_bytes = [b"a"]
self.assertEqual(inp.send("nonsensical value"), "a")
def test_send_nonblocking_no_event(self):
inp = Input()
inp.unprocessed_bytes = []
self.assertEqual(inp.send(0), None)
def test_nonblocking_read(self):
inp = Input()
self.assertEqual(inp._nonblocking_read(), 0)
def test_send_paste(self):
inp = Input()
inp.unprocessed_bytes = []
inp._wait_for_read_ready_or_timeout = Mock()
inp._wait_for_read_ready_or_timeout.return_value = (True, None)
inp._nonblocking_read = Mock()
n = inp.paste_threshold + 1
first_time = [True]
def side_effect():
if first_time:
inp.unprocessed_bytes.extend([b"a"] * n)
first_time.pop()
return n
else:
return None
inp._nonblocking_read.side_effect = side_effect
r = inp.send(0)
self.assertEqual(type(r), events.PasteEvent)
self.assertEqual(r.events, ["a"] * n)
def test_event_trigger(self):
inp = Input()
f = inp.event_trigger(CustomEvent)
self.assertEqual(inp.send(0), None)
f()
self.assertEqual(type(inp.send(0)), CustomEvent)
self.assertEqual(inp.send(0), None)
def test_schedule_event_trigger(self):
inp = Input()
f = inp.scheduled_event_trigger(CustomScheduledEvent)
self.assertEqual(inp.send(0), None)
f(when=time.time())
self.assertEqual(type(inp.send(0)), CustomScheduledEvent)
self.assertEqual(inp.send(0), None)
f(when=time.time() + 0.01)
self.assertEqual(inp.send(0), None)
time.sleep(0.01)
self.assertEqual(type(inp.send(0)), CustomScheduledEvent)
self.assertEqual(inp.send(0), None)
def test_schedule_event_trigger_blocking(self):
inp = Input()
f = inp.scheduled_event_trigger(CustomScheduledEvent)
f(when=time.time() + 0.05)
self.assertEqual(type(next(inp)), CustomScheduledEvent)
def test_threadsafe_event_trigger(self):
inp = Input()
f = inp.threadsafe_event_trigger(CustomEvent)
def check_event():
self.assertEqual(type(inp.send(1)), CustomEvent)
self.assertEqual(inp.send(0), None)
t = threading.Thread(target=check_event)
t.start()
f()
t.join()
def test_interrupting_sigint(self):
inp = Input(sigint_event=True)
def send_sigint():
os.kill(os.getpid(), signal.SIGINT)
with inp:
t = threading.Thread(target=send_sigint)
t.start()
self.assertEqual(type(inp.send(1)), events.SigIntEvent)
self.assertEqual(inp.send(0), None)
t.join()
def test_create_in_thread_with_sigint_event(self):
def create():
inp = Input(sigint_event=True)
t = threading.Thread(target=create)
t.start()
t.join()
def test_use_in_thread_with_sigint_event(self):
inp = Input(sigint_event=True)
def use():
with inp:
pass
t = threading.Thread(target=use)
t.start()
t.join()
def test_cleanup(self):
input_generator = Input()
for i in range(1000):
with input_generator:
pass
|
