From eabf2e62e3931069d70bfda69b0c5cfcf3cfb490 Mon Sep 17 00:00:00 2001 From: M1nd3r Date: Sat, 27 Apr 2024 01:54:58 +0200 Subject: [PATCH] feat(core): improve thp unittests --- core/tests/test_trezor.wire.thp_v1.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/core/tests/test_trezor.wire.thp_v1.py b/core/tests/test_trezor.wire.thp_v1.py index 4bea7dd8a..a3120d43e 100644 --- a/core/tests/test_trezor.wire.thp_v1.py +++ b/core/tests/test_trezor.wire.thp_v1.py @@ -101,13 +101,11 @@ class TestWireTrezorHostProtocolV1(unittest.TestCase): gen = thp_v1.read_message(self.interface, buffer) query = gen.send(None) 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: - gen.send(cid_req_message) - gen.send(None) # TODO fix this weird behaviour - gen.send(message) - gen.send(None) # TODO fix this weird behaviour - gen.send(None) # TODO fix this weird behaviour + gen.send(None) # e.value is StopIteration. e.value.value is the return value of the call result = e.value.value @@ -133,10 +131,8 @@ class TestWireTrezorHostProtocolV1(unittest.TestCase): query = gen.send(None) self.assertObjectEqual(query, self.interface.wait_object(io.POLL_READ)) - + gen.send(message) with self.assertRaises(StopIteration) as e: - gen.send(message) - gen.send(None) gen.send(None) # 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) query = gen.send(None) self.assertObjectEqual(query, self.interface.wait_object(io.POLL_READ)) + gen.send(packet) with self.assertRaises(StopIteration) as e: - gen.send(packet) - gen.send(None) gen.send(None) # e.value is StopIteration. e.value.value is the return value of the call