Skip to content

Commit cc0dd4c

Browse files
hongquanmindflayer
authored andcommitted
Fix compliant with original recv_into() (#98)
1 parent 6dbfdec commit cc0dd4c

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

mocket/mocket.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ def sendall(self, data, entry=None, *args, **kwargs):
261261
def read(self, buffersize):
262262
return self.fd.read(buffersize)
263263

264-
def recv_into(self, buffer, buffersize, flags=None):
264+
def recv_into(self, buffer, buffersize=None, flags=None):
265265
return buffer.write(self.fd.read(buffersize))
266266

267267
def recv(self, buffersize, flags=None):

tests/main/test_mocket.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def test_getresponse(self):
8282
def test_empty_getresponse(self):
8383
entry = MocketEntry(('localhost', 8080), [])
8484
self.assertEqual(entry.get_response(), encode_to_bytes(''))
85-
85+
8686
def test_subsequent_recv_requests_have_correct_length(self):
8787
Mocket.register(
8888
MocketEntry(
@@ -119,7 +119,7 @@ def test_recv_into(self):
119119
_so.sendall(b'first\r\n')
120120
assert _so.recv_into(buffer, 4096) == 12
121121
_so.sendall(b'second\r\n')
122-
assert _so.recv_into(buffer, 4096) == 5
122+
assert _so.recv_into(buffer) == 5
123123
_so.close()
124124
buffer.seek(0)
125125
assert buffer.read() == b'Long payloadShort'

0 commit comments

Comments
 (0)