def test_send_text_correct_call(self, recv_message_function_mock): message = 'La' for i in range(253): message += 'b' recv_message_function_mock.return_value = message text = Network.receive_text(self.socket_mock) recv_message_function_mock.assert_called_once_with(self.socket_mock) assert text == message[1:] assert len(text) == 254
def test_send_text_correct_call_long_message(self, recv_message_function_mock): message = 'Ma' for i in range(253): message += 'b' message2 = 'Lc' recv_message_function_mock.side_effect = [message, message2] text = Network.receive_text(self.socket_mock) assert recv_message_function_mock.call_count == 2 assert text == message[1:] + message2[1:] assert len(text) == 255
def connect_and_get_result_from_command(command, args=None): n = Network.connect_ipv4(SERVER_ADDRESS, SERVER_PORT) json_object = {} json_object["command"] = command if args: json_object["args"] = args json_string = json.dumps(json_object) Network.send_text(n, json_string) response = Network.receive_text(n) json_response = json.loads(response) if json_response["status"] == "ok": if "result" in json_response: return json_response["result"] else: raise CommandException(json_response["message"])