update msg tests
This commit is contained in:
parent
8f49a104bf
commit
b41c94dee4
|
@ -5,7 +5,7 @@ import unittest
|
||||||
|
|
||||||
from trezor import loop
|
from trezor import loop
|
||||||
from trezor import msg
|
from trezor import msg
|
||||||
from trezor.msg import read_report, read_message, write_message
|
from trezor.msg import read_wire_msg, write_wire_msg
|
||||||
|
|
||||||
|
|
||||||
def chunks(l, n):
|
def chunks(l, n):
|
||||||
|
@ -15,47 +15,32 @@ def chunks(l, n):
|
||||||
|
|
||||||
class TestMsg(unittest.TestCase):
|
class TestMsg(unittest.TestCase):
|
||||||
|
|
||||||
def test_read_report(self):
|
def test_read_wire_msg(self):
|
||||||
|
|
||||||
reader = read_report()
|
reader = read_wire_msg()
|
||||||
syscall = reader.send(None)
|
|
||||||
|
|
||||||
self.assertIsInstance(syscall, loop.Select)
|
|
||||||
self.assertEqual(syscall.events, (loop.HID_READ,))
|
|
||||||
|
|
||||||
empty_report = b'\x3f' + b'\x00' * 63
|
|
||||||
try:
|
|
||||||
reader.send(empty_report)
|
|
||||||
except StopIteration as e:
|
|
||||||
result = e.value
|
|
||||||
self.assertEqual(result, empty_report)
|
|
||||||
|
|
||||||
def test_read_message(self):
|
|
||||||
|
|
||||||
reader = read_message()
|
|
||||||
reader.send(None)
|
reader.send(None)
|
||||||
|
|
||||||
empty_message = b'\x3f##\xab\xcd\x00\x00\x00\x00' + b'\x00' * 55
|
empty_message = b'\x3f##\xab\xcd\x00\x00\x00\x00' + b'\x00' * 55
|
||||||
try:
|
try:
|
||||||
reader.send(empty_message)
|
reader.send((loop.HID_READ, empty_message))
|
||||||
except StopIteration as e:
|
except StopIteration as e:
|
||||||
restype, resmsg = e.value
|
restype, resmsg = e.value
|
||||||
self.assertEqual(restype, int('0xabcd', 16))
|
self.assertEqual(restype, int('0xabcd', 16))
|
||||||
self.assertEqual(resmsg, b'')
|
self.assertEqual(resmsg, b'')
|
||||||
|
|
||||||
reader = read_message()
|
reader = read_wire_msg()
|
||||||
reader.send(None)
|
reader.send(None)
|
||||||
|
|
||||||
content = bytes([x for x in range(0, 55)])
|
content = bytes([x for x in range(0, 55)])
|
||||||
message = b'\x3f##\xab\xcd\x00\x00\x00\x37' + content
|
message = b'\x3f##\xab\xcd\x00\x00\x00\x37' + content
|
||||||
try:
|
try:
|
||||||
reader.send(message)
|
reader.send((loop.HID_READ, message))
|
||||||
except StopIteration as e:
|
except StopIteration as e:
|
||||||
restype, resmsg = e.value
|
restype, resmsg = e.value
|
||||||
self.assertEqual(restype, int('0xabcd', 16))
|
self.assertEqual(restype, int('0xabcd', 16))
|
||||||
self.assertEqual(resmsg, content)
|
self.assertEqual(resmsg, content)
|
||||||
|
|
||||||
reader = read_message()
|
reader = read_wire_msg()
|
||||||
reader.send(None)
|
reader.send(None)
|
||||||
|
|
||||||
content = bytes([x for x in range(0, 256)])
|
content = bytes([x for x in range(0, 256)])
|
||||||
|
@ -63,22 +48,22 @@ class TestMsg(unittest.TestCase):
|
||||||
reports = [b'\x3f' + ch + '\x00' * (63 - len(ch)) for ch in chunks(message, 63)]
|
reports = [b'\x3f' + ch + '\x00' * (63 - len(ch)) for ch in chunks(message, 63)]
|
||||||
try:
|
try:
|
||||||
for report in reports:
|
for report in reports:
|
||||||
reader.send(report)
|
reader.send((loop.HID_READ, report))
|
||||||
except StopIteration as e:
|
except StopIteration as e:
|
||||||
restype, resmsg = e.value
|
restype, resmsg = e.value
|
||||||
self.assertEqual(restype, int('0xabcd', 16))
|
self.assertEqual(restype, int('0xabcd', 16))
|
||||||
self.assertEqual(resmsg, content)
|
self.assertEqual(resmsg, content)
|
||||||
|
|
||||||
def test_write_message(self):
|
def test_write_wire_msg(self):
|
||||||
|
|
||||||
written_reports = []
|
sent_reps = []
|
||||||
msg.write_report = lambda report: written_reports.append(bytes(report))
|
msg.send = lambda rep: sent_reps.append(bytes(rep))
|
||||||
|
|
||||||
content = bytes([x for x in range(0, 256)])
|
content = bytes([x for x in range(0, 256)])
|
||||||
message = b'##\xab\xcd\x00\x00\x01\00' + content
|
message = b'##\xab\xcd\x00\x00\x01\00' + content
|
||||||
reports = [b'\x3f' + ch + '\x00' * (63 - len(ch)) for ch in chunks(message, 63)]
|
reports = [b'\x3f' + ch + '\x00' * (63 - len(ch)) for ch in chunks(message, 63)]
|
||||||
write_message(int('0xabcd'), content)
|
write_wire_msg(int('0xabcd'), content)
|
||||||
self.assertEqual(written_reports, reports)
|
self.assertEqual(sent_reps, reports)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
Loading…
Reference in New Issue