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"])
