From 0176f167296e3c96ecf05b8d9b09fe72eb07fae3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nguy=E1=BB=85n=20H=E1=BB=93ng=20Qu=C3=A2n?= Date: Thu, 8 Aug 2019 13:52:30 +0700 Subject: [PATCH] Fix compliant with original recv_into() --- mocket/mocket.py | 2 +- tests/main/test_mocket.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mocket/mocket.py b/mocket/mocket.py index 380064cb..d90ba625 100644 --- a/mocket/mocket.py +++ b/mocket/mocket.py @@ -261,7 +261,7 @@ def sendall(self, data, entry=None, *args, **kwargs): def read(self, buffersize): return self.fd.read(buffersize) - def recv_into(self, buffer, buffersize, flags=None): + def recv_into(self, buffer, buffersize=None, flags=None): return buffer.write(self.fd.read(buffersize)) def recv(self, buffersize, flags=None): diff --git a/tests/main/test_mocket.py b/tests/main/test_mocket.py index 3ebe56a5..d54a5c77 100644 --- a/tests/main/test_mocket.py +++ b/tests/main/test_mocket.py @@ -82,7 +82,7 @@ def test_getresponse(self): def test_empty_getresponse(self): entry = MocketEntry(('localhost', 8080), []) self.assertEqual(entry.get_response(), encode_to_bytes('')) - + def test_subsequent_recv_requests_have_correct_length(self): Mocket.register( MocketEntry( @@ -119,7 +119,7 @@ def test_recv_into(self): _so.sendall(b'first\r\n') assert _so.recv_into(buffer, 4096) == 12 _so.sendall(b'second\r\n') - assert _so.recv_into(buffer, 4096) == 5 + assert _so.recv_into(buffer) == 5 _so.close() buffer.seek(0) assert buffer.read() == b'Long payloadShort'