Esempio n. 1
0
 def test_ListenRecvExit(self):
     server = ListenRecvExit()
     serverProcess, port = genericServer.StartServer(server)
     # Create the udp socket
     sock = comms.udp()
     # Send the message to the server
     comms.sendMessage(sock, 'test', comms.OK, (genericServer.ADDR, port))
     # Receive a message
     response, serverAddr = sock.recvfrom(comms.RECV_SIZE)
     # Parse the response and return it
     reply, status = comms.parseMessage(response)
     # Stop the server
     serverProcess.terminate()
     # If the test errored
     self.assertEqual(EXIT_MESSAGE, reply, \
         'Server reply is not exit message {0}'.format(reply))
     self.assertEqual(comms.EXIT, status, \
         'Server did not exit, status code {0}'.format(status))
Esempio n. 2
0
 def __init__(self):
     """
     Creates a udp socket that will be used to send data.
     """
     super(Client, self).__init__()
     self.sock = comms.udp()