class TxRxTest(unittest.TestCase): # Setup logging logger = logging.getLogger("TxRxTest") 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" server_port = 8080 simple_msg = Message(1, {"test_data": 12345}) def setUp(self): # Setup the transmitter self.client = UdpClient() # Need data and algorithm to transmit it # self.data = TxRxTest.simple_msg self.data = None self.tx_alg = TransmissionAlgorithm() self.source = Source(self.data, self.tx_alg) self.remote_hosts = [(TxRxTest.server_address, TxRxTest.server_port)] self.transmitter = Transmitter(self.client, self.source, self.remote_hosts) # Setup the receiver self.server = UdpServer((TxRxTest.server_address, TxRxTest.server_port), UdpServerHandler) # Need reconstruction algorithm # Need a callback for the sink as well self.rx_alg = ReconstructionAlgorithm() self.sink = Sink(self.rx_alg) self.receiver = Receiver(self.server, self.sink) def tearDown(self): pass @log_test(logger, globals.log_seperator) def testSimpleConnectivity(self): self.receiver.start() self.transmitter.start() def verifySimpleConnectivity(self): pass
def setUp(self): # Setup the transmitter self.client = UdpClient() # Need data and algorithm to transmit it # self.data = TxRxTest.simple_msg self.data = None self.tx_alg = TransmissionAlgorithm() self.source = Source(self.data, self.tx_alg) self.remote_hosts = [(TxRxTest.server_address, TxRxTest.server_port)] self.transmitter = Transmitter(self.client, self.source, self.remote_hosts) # Setup the receiver self.server = UdpServer((TxRxTest.server_address, TxRxTest.server_port), UdpServerHandler) # Need reconstruction algorithm # Need a callback for the sink as well self.rx_alg = ReconstructionAlgorithm() self.sink = Sink(self.rx_alg) self.receiver = Receiver(self.server, self.sink)