|
26 | 26 | # POSSIBILITY OF SUCH DAMAGE. |
27 | 27 |
|
28 | 28 | from u2flib_host.hid_transport import list_devices |
29 | | -from nose import SkipTest |
| 29 | +import unittest |
30 | 30 |
|
31 | 31 |
|
32 | | -def get_device(): |
33 | | - devs = list_devices() |
34 | | - if len(devs) != 1: |
35 | | - raise SkipTest("Tests require a single U2FHID device") |
36 | | - return devs[0] |
| 32 | +class HidTest(unittest.TestCase): |
| 33 | + def get_device(self): |
| 34 | + devs = list_devices() |
| 35 | + if len(devs) != 1: |
| 36 | + raise Exception("Tests require a single U2F HID device") |
| 37 | + return devs[0] |
37 | 38 |
|
| 39 | + def test_open_close(self): |
| 40 | + dev = self.get_device() |
| 41 | + for i in xrange(0, 10): |
| 42 | + dev.open() |
| 43 | + dev.close() |
38 | 44 |
|
39 | | -def test_open_close(): |
40 | | - dev = get_device() |
41 | | - for i in xrange(0, 10): |
42 | | - dev.open() |
43 | | - dev.close() |
44 | | - |
45 | | - |
46 | | -def test_echo(): |
47 | | - msg1 = 'hello world!' |
48 | | - msg2 = ' ' |
49 | | - msg3 = '' |
50 | | - with get_device() as dev: |
51 | | - resp1 = dev.send_apdu(0x40, 0, 0, msg1) |
52 | | - resp2 = dev.send_apdu(0x40, 0, 0, msg2) |
53 | | - resp3 = dev.send_apdu(0x40, 0, 0, msg3) |
54 | | - assert resp1 == msg1 |
55 | | - assert resp2 == msg2 |
56 | | - assert resp3 == msg3 |
| 45 | + def test_echo(self): |
| 46 | + msg1 = 'hello world!' |
| 47 | + msg2 = ' ' |
| 48 | + msg3 = '' |
| 49 | + with self.get_device() as dev: |
| 50 | + resp1 = dev.send_apdu(0x40, 0, 0, msg1) |
| 51 | + resp2 = dev.send_apdu(0x40, 0, 0, msg2) |
| 52 | + resp3 = dev.send_apdu(0x40, 0, 0, msg3) |
| 53 | + assert resp1 == msg1 |
| 54 | + assert resp2 == msg2 |
| 55 | + assert resp3 == msg3 |
0 commit comments