lili/amodem
1
0
Fork 0
mirror of https://github.com/romanz/amodem.git synced 2026-04-20 04:56:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

tests: refactor mocks and fakes

Browse Source
This commit is contained in:
Roman Zeyde
2016-01-22 12:04:24 +02:00
parent a4b905cd6f
commit 7f36097c15
3 changed files with 21 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
trezor_agent/tests/test_utils.py
View File

@@ -24,7 +24,7 @@ def test_frames():
assert util.read_frame(io.BytesIO(f)) == b''.join(msgs)
class SocketMock(object):
class FakeSocket(object):
def __init__(self):
self.buf = io.BytesIO()
@@ -36,7 +36,7 @@ class SocketMock(object):
def test_send_recv():
s = SocketMock()
s = FakeSocket()
util.send(s, b'123')
util.send(s, data=[42], fmt='B')
assert s.buf.getvalue() == b'123*'