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)