arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-15 20:19:23 +00:00
Code Issues Releases Wiki Activity

Improve tests

Browse Source
This commit is contained in:
M1nd3r 2024-03-13 11:18:37 +01:00 committed by M1nd3r
parent bbaac5ec1f
commit 22418a2fe0
4 changed files with 11 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
core/src/storage/cache.py
View File

@ -102,8 +102,7 @@ def delete(key: int) -> None:
if TYPE_CHECKING:
@overload
def get(key: int) -> bytes | None:
...
def get(key: int) -> bytes | None: ...
@overload
def get(key: int, default: T) -> bytes | T: # noqa: F811

15
core/src/trezor/utils.py
View File

@ -148,24 +148,19 @@ def chunks(items: Chunkable, size: int) -> Iterator[Chunkable]:
if TYPE_CHECKING:
class HashContext(Protocol):
def update(self, __buf: bytes) -> None:
...
def update(self, __buf: bytes) -> None: ...
def digest(self) -> bytes:
...
def digest(self) -> bytes: ...
class HashContextInitable(HashContext, Protocol):
def __init__( # pylint: disable=super-init-not-called
self, __data: bytes | None = None
) -> None:
...
) -> None: ...
class Writer(Protocol):
def append(self, __b: int) -> None:
...
def append(self, __b: int) -> None: ...
def extend(self, __buf: bytes) -> None:
...
def extend(self, __buf: bytes) -> None: ...
if False: # noqa

8
core/src/trezor/wire/context.py
View File

@ -78,14 +78,14 @@ class Context:
if TYPE_CHECKING:
@overload
async def read(self, expected_types: Container[int]) -> protobuf.MessageType:
...
async def read(
self, expected_types: Container[int]
) -> protobuf.MessageType: ...
@overload
async def read(
self, expected_types: Container[int], expected_type: type[LoadedMessageType]
) -> LoadedMessageType:
...
) -> LoadedMessageType: ...
async def read(
self,

6
core/tests/test_storage.cache.py
View File

@ -230,13 +230,9 @@ class TestStorageCache(unittest.TestCase):
self.assertEqual(cache.get(KEY), b"hello")
def test_EndSession(self):
<<<<<<< HEAD
self.assertRaises(cache.InvalidSessionError, cache.get, KEY)
cache.start_session()
=======
self.assertRaises(InvalidSessionError, cache.get, KEY)
session_id = cache.start_session()
>>>>>>> 8681ba167 (Basic THP functinality - not-polished prototype)
self.assertTrue(is_session_started())
self.assertIsNone(cache.get(KEY))
await_result(handle_EndSession(EndSession()))