Name	Size	Mode	Actions
__pycache__/	-	0755	rm
conftest.py	2715	0644	editdlrm
helper.py	4755	0644	editdlrm
test_cli.py	2664	0644	editdlrm
test_conn.py	53393	0644	editdlrm
test_core.py	14699	0644	editdlrm
test_dispatch.py	1210	0644	editdlrm
test_errors.py	924	0644	editdlrm
test_makefile.py	1191	0644	editdlrm
test_server.py	16697	0644	editdlrm
test_ssl.py	22268	0644	editdlrm
test_wsgi.py	2823	0644	editdlrm
test__compat.py	1661	0644	editdlrm
webtest.py	18398	0644	editdlrm
_pytest_plugin.py	1768	0644	editdlrm
__init__.py	26	0644	editdlrm

"""Tests for :py:mod:`cheroot.makefile`.""" from cheroot import makefile class MockSocket: """A mock socket.""" def __init__(self): """Initialize :py:class:`MockSocket`.""" self.messages = [] def recv_into(self, buf): """Simulate ``recv_into`` for Python 3.""" if not self.messages: return 0 msg = self.messages.pop(0) for index, byte in enumerate(msg): buf[index] = byte return len(msg) def recv(self, size): """Simulate ``recv`` for Python 2.""" try: return self.messages.pop(0) except IndexError: return '' def send(self, val): """Simulate a send.""" return len(val) def test_bytes_read(): """Reader should capture bytes read.""" sock = MockSocket() sock.messages.append(b'foo') rfile = makefile.MakeFile(sock, 'r') rfile.read() assert rfile.bytes_read == 3 def test_bytes_written(): """Writer should capture bytes written.""" sock = MockSocket() sock.messages.append(b'foo') wfile = makefile.MakeFile(sock, 'w') wfile.write(b'bar') assert wfile.bytes_written == 3