Beispiel #1
0
 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()
Beispiel #2
0
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())