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'