arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-24 21:32:03 +00:00
Code Issues Releases Wiki Activity

Improve test slightly

Browse Source
This commit is contained in:
M1nd3r 2024-03-14 16:26:31 +01:00 committed by M1nd3r
parent fa82bb1132
commit 7200af7547
1 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
core/tests/test_trezor.wire.thp_v1.py
View File

@ -101,12 +101,10 @@ class TestWireTrezorHostProtocolV1(unittest.TestCase):
gen = thp_v1.read_message(self.interface, buffer) gen = thp_v1.read_message(self.interface, buffer)
query = gen.send(None) query = gen.send(None)
self.assertObjectEqual(query, self.interface.wait_object(io.POLL_READ)) self.assertObjectEqual(query, self.interface.wait_object(io.POLL_READ))
gen.send(cid_req_message)
gen.send(None)
gen.send(message)
with self.assertRaises(StopIteration) as e: with self.assertRaises(StopIteration) as e:
gen.send(cid_req_message)
gen.send(None)
gen.send(message)
gen.send(None)
gen.send(None) gen.send(None)
# e.value is StopIteration. e.value.value is the return value of the call # e.value is StopIteration. e.value.value is the return value of the call
@ -133,10 +131,8 @@ class TestWireTrezorHostProtocolV1(unittest.TestCase):
query = gen.send(None) query = gen.send(None)
self.assertObjectEqual(query, self.interface.wait_object(io.POLL_READ)) self.assertObjectEqual(query, self.interface.wait_object(io.POLL_READ))
gen.send(message)
with self.assertRaises(StopIteration) as e: with self.assertRaises(StopIteration) as e:
gen.send(message)
gen.send(None)
gen.send(None) gen.send(None)
# e.value is StopIteration. e.value.value is the return value of the call # e.value is StopIteration. e.value.value is the return value of the call
@ -211,9 +207,8 @@ class TestWireTrezorHostProtocolV1(unittest.TestCase):
gen = thp_v1.read_message(self.interface, buffer) gen = thp_v1.read_message(self.interface, buffer)
query = gen.send(None) query = gen.send(None)
self.assertObjectEqual(query, self.interface.wait_object(io.POLL_READ)) self.assertObjectEqual(query, self.interface.wait_object(io.POLL_READ))
gen.send(packet)
with self.assertRaises(StopIteration) as e: with self.assertRaises(StopIteration) as e:
gen.send(packet)
gen.send(None)
gen.send(None) gen.send(None)
# e.value is StopIteration. e.value.value is the return value of the call # e.value is StopIteration. e.value.value is the return value of the call