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
|
from __future__ import unicode_literals
from prompt_toolkit.key_binding import InputProcessor, Registry
from prompt_toolkit.keys import Keys
from prompt_toolkit.inputstream import KeyPress
import unittest
class KeyBindingTest(unittest.TestCase):
def setUp(self):
class Handlers(object):
def __init__(self):
self.called = []
def __getattr__(self, name):
def func(event):
self.called.append(name)
return func
self.handlers = Handlers()
self.registry = Registry()
self.registry.add_binding(Keys.ControlX, Keys.ControlC)(self.handlers.controlx_controlc)
self.registry.add_binding(Keys.ControlX)(self.handlers.control_x)
self.registry.add_binding(Keys.ControlD)(self.handlers.control_d)
self.registry.add_binding(Keys.ControlSquareClose, Keys.Any)(self.handlers.control_square_close_any)
self.processor = InputProcessor(self.registry)
def test_feed_simple(self):
self.processor.feed_key(KeyPress(Keys.ControlX, '\x18'))
self.processor.feed_key(KeyPress(Keys.ControlC, '\x03'))
self.assertEqual(self.handlers.called, ['controlx_controlc'])
def test_feed_several(self):
# First an unknown key first.
self.processor.feed_key(KeyPress(Keys.ControlQ, ''))
self.assertEqual(self.handlers.called, [])
# Followed by a know key sequence.
self.processor.feed_key(KeyPress(Keys.ControlX, ''))
self.processor.feed_key(KeyPress(Keys.ControlC, ''))
self.assertEqual(self.handlers.called, ['controlx_controlc'])
# Followed by another unknown sequence.
self.processor.feed_key(KeyPress(Keys.ControlR, ''))
self.processor.feed_key(KeyPress(Keys.ControlS, ''))
# Followed again by a know key sequence.
self.processor.feed_key(KeyPress(Keys.ControlD, ''))
self.assertEqual(self.handlers.called, ['controlx_controlc', 'control_d'])
def test_control_square_closed_any(self):
self.processor.feed_key(KeyPress(Keys.ControlSquareClose, ''))
self.processor.feed_key(KeyPress('C', 'C'))
self.assertEqual(self.handlers.called, ['control_square_close_any'])
def test_common_prefix(self):
# Sending Control_X should not yet do anything, because there is
# another sequence starting with that as well.
self.processor.feed_key(KeyPress(Keys.ControlX, ''))
self.assertEqual(self.handlers.called, [])
# When another key is pressed, we know that we did not meant the longer
# "ControlX ControlC" sequence and the callbacks are called.
self.processor.feed_key(KeyPress(Keys.ControlD, ''))
self.assertEqual(self.handlers.called, ['control_x', 'control_d'])
|