def read_session_buff_gen(raw_response): buffers = [] if raw_response: decoded_response, dummy = frame_decode_all(raw_response) for data in decoded_response: payload = seto.Payload() payload.ParseFromString(bytes(data)) buffers.extend([Frame(bytes=data, protobuf=payload)]) class indexer(object): index = 0 def ret_buff(): if indexer.index < len(buffers): val = buffers[indexer.index] indexer.index += 1 return val return None return ret_buff
def check_frame_decode_all(byte_array, output): eq_(frame_decode_all(byte_array), output)
def read(self): self.read_buffer += self.socket.recv() messages, remaining_buffer = frame_decode_all(self.read_buffer) self.buffered_incoming_payloads.extend(messages) self.read_buffer = remaining_buffer
def check_frame_decode_all(input_, output): eq_(frame_decode_all(input_), output)