mirror of
https://github.com/romanz/amodem.git
synced 2026-02-07 01:18:02 +08:00
48 lines
841 B
Python
48 lines
841 B
Python
from amodem import calib
|
|
|
|
from io import BytesIO
|
|
|
|
|
|
class ProcessMock(object):
|
|
def __init__(self):
|
|
self.buf = BytesIO()
|
|
self.stdin = self
|
|
self.stdout = self
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
return self
|
|
|
|
def kill(self):
|
|
pass
|
|
|
|
def write(self, data):
|
|
self.buf.write(data)
|
|
if self.buf.tell() > 1e6:
|
|
raise KeyboardInterrupt
|
|
|
|
def read(self, n):
|
|
return self.buf.read(n)
|
|
|
|
|
|
def test_success():
|
|
p = ProcessMock()
|
|
calib.send(p)
|
|
p.buf.seek(0)
|
|
calib.recv(p)
|
|
|
|
|
|
def test_errors():
|
|
p = ProcessMock()
|
|
|
|
def _write(data):
|
|
raise IOError()
|
|
p.write = _write
|
|
calib.send(p)
|
|
assert p.buf.tell() == 0
|
|
|
|
def _read(data):
|
|
raise KeyboardInterrupt()
|
|
p.read = _read
|
|
calib.recv(p)
|
|
assert p.buf.tell() == 0
|