mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-07-24 15:38:22 +00:00
chore(core, tests): add MockAsync and await_result_patched
[no changelog]
This commit is contained in:
parent
b081f67786
commit
be287648d0
@ -24,6 +24,19 @@ class Mock:
|
||||
return self.return_value
|
||||
|
||||
|
||||
class MockAsync(Mock):
|
||||
def __call__(self, *args, **kwargs) -> Any:
|
||||
self.calls.append((args, kwargs))
|
||||
|
||||
if self.raises is not None:
|
||||
raise self.raises
|
||||
|
||||
async def _async_return() -> Any:
|
||||
return self.return_value
|
||||
|
||||
return _async_return()
|
||||
|
||||
|
||||
class patch:
|
||||
MOCK_OBJECT = object()
|
||||
NO_VALUE = object()
|
||||
|
Loading…
Reference in New Issue
Block a user