def receiver(user, port):
s = udpSocket()
try:
s.bind(('', port))
while True:
msgBytes, address = s.recvfrom(4096)
m = helper.parse_message(str(msgBytes.decode()))
handleMessageReceived(s, port, user, address, m[0], m[1], m[2],
m[3])
finally:
s.close()
def get_message(c_socket, state, new_chat_pipe):
""" Receive a message from connection and emit a signal to GUI thread
"""
logging.debug("Client thread started")
addr = c_socket.getpeername()
message_part = c_socket.recv(1024)
message = b""
while message_part:
message += message_part
message_part = c_socket.recv(1024)
message = message.decode('latin1')
logging.debug("Got message {0} from {1}".format(message, addr))
message_dict = parse_message(message)
if (message_dict.get('type') == 'msg'):
pipes = state.pipes
handle_msg(message_dict, pipes, state, new_chat_pipe)
elif (message_dict.get('type') == 'ping'):
handle_ping(c_socket)
c_socket.close()
def get_message(c_socket, state, new_chat_pipe):
""" Receive a message from connection and emit a signal to GUI thread
"""
logging.debug("Client thread started")
addr = c_socket.getpeername()
message_part = c_socket.recv(1024)
message = b""
while message_part:
message += message_part
message_part = c_socket.recv(1024)
message = message.decode('latin1')
logging.debug("Got message {0} from {1}".format(message, addr))
message_dict = parse_message(message)
if(message_dict.get('type') == 'msg'):
pipes = state.pipes
handle_msg(message_dict, pipes, state, new_chat_pipe)
elif(message_dict.get('type') == 'ping'):
handle_ping(c_socket)
c_socket.close()
def test_parse_mismatch(self):
""" Test parse mismatch """
message = "<null><null>"
message_dict = parse_message(message)
assert message_dict == {}
def test_parse_body(self):
""" Test body parse part """
message = "<null><null><null>hejsan"
message_dict = parse_message(message)
assert message_dict['body'] == 'hejsan'
def test_parse_reciever(self):
""" Test reciever parse part """
message = "<null><null><test_recv>null"
message_dict = parse_message(message)
assert message_dict['receiver'] == 'test_recv'
def test_parse_sender(self):
""" Test sender parse part """
message = "<null><test_sender><null>"
message_dict = parse_message(message)
assert message_dict['sender'] == 'test_sender'
def test_parse_type(self):
""" Test type parse part """
message = "<ping><null><null>"
message_dict = parse_message(message)
assert message_dict['type'] == 'ping'
