def setUp(self):
self.client = UdpClient()
self.server = UdpServer( (UdpClientServerTest.server_address, 0), UdpServerHandler)
self.ip_address, self.port = self.server.server_address
UdpClientServerTest.logger.debug( 'UDP server created with address: %s:%s' % \
(self.ip_address, self.port) )
self.server_thread = threading.Thread(name='udp server', target=self.server.serve_forever)
self.server_thread.start()
class UdpClientServerTest(unittest.TestCase):
# Setup logging
logger = logging.getLogger('UdpClientServerTest')
logger.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(levelname)s %(asctime)s %(name)s Line: %(lineno)d | %(message)s')
handler = logging.StreamHandler()
handler.setFormatter(formatter)
logger.addHandler(handler)
# Class constants
server_address = 'localhost'
simple_msg = Message( 1, {'test_data': 12345} )
def setUp(self):
self.client = UdpClient()
self.server = UdpServer( (UdpClientServerTest.server_address, 0), UdpServerHandler)
self.ip_address, self.port = self.server.server_address
UdpClientServerTest.logger.debug( 'UDP server created with address: %s:%s' % \
(self.ip_address, self.port) )
self.server_thread = threading.Thread(name='udp server', target=self.server.serve_forever)
self.server_thread.start()
def tearDown(self):
self.server.shutdown()
self.client.close()
@log_test(logger, globals.log_seperator)
def testSimpleConnectivity(self):
wait_time_for_msg = 0.5
self.client.sendMsg(self.ip_address, int(self.port), UdpClientServerTest.simple_msg)
time.sleep(wait_time_for_msg)
if self.server.last_rcvd_msg == None:
assert False, 'No message received.'
rcvd_msg = self.server.last_rcvd_msg
assert UdpClientServerTest.simple_msg == rcvd_msg, 'Incorrect message value: %s' % \
(rcvd_msg.convertToJson())