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
|
#!/usr/bin/python
# Copyright 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import unittest
import mock
import hid_constants
import keyboard_gadget
import usb_constants
class KeyboardGadgetTest(unittest.TestCase):
def test_key_press(self):
g = keyboard_gadget.KeyboardGadget()
chip = mock.Mock()
g.Connected(chip, usb_constants.Speed.FULL)
g.KeyDown(0x04)
self.assertEqual(g.ControlRead(0xA1, 1, 0x0100, 0, 8),
'\x00\x00\x04\x00\x00\x00\x00\x00')
g.KeyUp(0x04)
self.assertEqual(g.ControlRead(0xA1, 1, 0x0100, 0, 8),
'\x00\x00\x00\x00\x00\x00\x00\x00')
chip.SendPacket.assert_has_calls([
mock.call(0x81, '\x00\x00\x04\x00\x00\x00\x00\x00'),
mock.call(0x81, '\x00\x00\x00\x00\x00\x00\x00\x00'),
])
def test_key_press_with_modifier(self):
g = keyboard_gadget.KeyboardGadget()
chip = mock.Mock()
g.Connected(chip, usb_constants.Speed.FULL)
g.ModifierDown(hid_constants.ModifierKey.L_SHIFT)
g.KeyDown(0x04)
g.KeyDown(0x05)
g.KeyUp(0x04)
g.KeyUp(0x05)
g.ModifierUp(hid_constants.ModifierKey.L_SHIFT)
chip.SendPacket.assert_has_calls([
mock.call(0x81, '\x02\x00\x00\x00\x00\x00\x00\x00'),
mock.call(0x81, '\x02\x00\x04\x00\x00\x00\x00\x00'),
mock.call(0x81, '\x02\x00\x04\x05\x00\x00\x00\x00'),
mock.call(0x81, '\x02\x00\x00\x05\x00\x00\x00\x00'),
mock.call(0x81, '\x02\x00\x00\x00\x00\x00\x00\x00'),
mock.call(0x81, '\x00\x00\x00\x00\x00\x00\x00\x00'),
])
def test_set_leds(self):
g = keyboard_gadget.KeyboardGadget()
chip = mock.Mock()
g.Connected(chip, usb_constants.Speed.FULL)
g.SetConfiguration(1)
self.assertEqual(g.ControlRead(0xA1, 1, 0x0200, 0, 8), '\x00')
self.assertTrue(g.ControlWrite(0x21, 9, 0x0200, 0, '\x01'))
self.assertEqual(g.ControlRead(0xA1, 1, 0x0200, 0, 8), '\x01')
g.ReceivePacket(0x01, '\x03')
self.assertFalse(chip.HaltEndpoint.called)
self.assertEqual(g.ControlRead(0xA1, 1, 0x0200, 0, 8), '\x03')
if __name__ == '__main__':
unittest.main()
|
