arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-02 10:51:06 +00:00
Code Issues Releases Wiki Activity

core/tests: extract common await_result() method

Browse Source
This commit is contained in:
matejcik 2020-01-30 15:57:19 +01:00 committed by Pavol Rusnak
parent ee07b32f52
commit d4171aaedc
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
2 changed files with 21 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
core/tests/common.py
View File

@ -1,9 +1,23 @@
import sys import sys
sys.path.append('../src') sys.path.append("../src")
from ubinascii import hexlify, unhexlify # noqa: F401 from ubinascii import hexlify, unhexlify # noqa: F401
import unittest # noqa: F401 import unittest # noqa: F401
from trezor import utils # noqa: F401 from trezor import utils # noqa: F401
def await_result(task: Awaitable) -> Any:
value = None
while True:
try:
result = task.send(value)
except StopIteration as e:
return e.value
if result:
value = await_result(result)
else:
value = None

26
core/tests/test_protobuf.py
View File

@ -44,28 +44,14 @@ class ByteArrayWriter:
return len(buf) return len(buf)
def run_until_complete(task: Awaitable) -> Any:
value = None
while True:
try:
result = task.send(value)
except StopIteration as e:
return e.value
if result:
value = run_until_complete(result)
else:
value = None
def load_uvarint(data: bytes) -> int: def load_uvarint(data: bytes) -> int:
reader = ByteReader(data) reader = ByteReader(data)
return run_until_complete(protobuf.load_uvarint(reader)) return await_result(protobuf.load_uvarint(reader))
def dump_uvarint(value: int) -> bytearray: def dump_uvarint(value: int) -> bytearray:
writer = ByteArrayWriter() writer = ByteArrayWriter()
run_until_complete(protobuf.dump_uvarint(writer, value)) await_result(protobuf.dump_uvarint(writer, value))
return writer.buf return writer.buf
@ -106,9 +92,9 @@ class TestProtobuf(unittest.TestCase):
# ok message: # ok message:
msg = Message(-42, 5) msg = Message(-42, 5)
writer = ByteArrayWriter() writer = ByteArrayWriter()
run_until_complete(protobuf.dump_message(writer, msg)) await_result(protobuf.dump_message(writer, msg))
reader = ByteReader(bytes(writer.buf)) reader = ByteReader(bytes(writer.buf))
nmsg = run_until_complete(protobuf.load_message(reader, Message)) nmsg = await_result(protobuf.load_message(reader, Message))
self.assertEqual(msg.sint_field, nmsg.sint_field) self.assertEqual(msg.sint_field, nmsg.sint_field)
self.assertEqual(msg.enum_field, nmsg.enum_field) self.assertEqual(msg.enum_field, nmsg.enum_field)
@ -116,10 +102,10 @@ class TestProtobuf(unittest.TestCase):
# bad enum value: # bad enum value:
msg = Message(-42, 42) msg = Message(-42, 42)
writer = ByteArrayWriter() writer = ByteArrayWriter()
run_until_complete(protobuf.dump_message(writer, msg)) await_result(protobuf.dump_message(writer, msg))
reader = ByteReader(bytes(writer.buf)) reader = ByteReader(bytes(writer.buf))
with self.assertRaises(TypeError): with self.assertRaises(TypeError):
run_until_complete(protobuf.load_message(reader, Message)) await_result(protobuf.load_message(reader, Message))
if __name__ == "__main__": if __name__ == "__main__":