def send_side_effect(data):
try:
next_data = next(sent_data)
except StopIteration:
self.fail("unexpected sent data: %s" % data)
if type(next_data) is not dict:
self.assertEqual(next_data, data)
elif next_data["assert"]:
self.assertEqual(next_data["value"], data)
asynctest.set_read_ready(self.socket_mock, self.loop)
return len(data)
def recv(self, max_bytes):
if not self.recv_buffer:
try:
self.recv_buffer.extend(next(self.recv_data))
asynctest.set_read_ready(self, self.loop)
except StopIteration:
# nothing left
pass
data = self.recv_buffer[:max_bytes]
self.recv_buffer = self.recv_buffer[max_bytes:]
if self.recv_buffer:
# Some more data to read
asynctest.set_read_ready(self, self.loop)
return data
def recv_side_effect(max_bytes):
nonlocal recv_buffer
if not recv_buffer:
try:
recv_buffer.extend(next(recv_data))
asynctest.set_read_ready(socket_mock, self.loop)
except StopIteration:
# nothing left
pass
data = recv_buffer[:max_bytes]
recv_buffer = recv_buffer[max_bytes:]
if recv_buffer:
# Some more data to read
asynctest.set_read_ready(socket_mock, self.loop)
return data
def send_side_effect(data):
asynctest.set_read_ready(socket_mock, self.loop)
return len(data)
def send(self, data):
asynctest.set_read_ready(self, self.loop)
return len(data)
