def setUp(self): # Initialize OneToManyService. self.service = OneToManyService('testhost', '_testService._tcp', 1600) self.service.start()
class OneToManyServiceTest(unittest.TestCase): def setUp(self): # Initialize OneToManyService. self.service = OneToManyService('testhost', '_testService._tcp', 1600) self.service.start() def tearDown(self): self.service.kill() def testSimpleMessage(self): # Generate a session UUID and register it as a destination. sessionUUID = str(uuid.uuid1()) self.service.registerDestination(sessionUUID) # Send message. self.service.sendMessage(sessionUUID, 'test') # Alow messages to be processed. time.sleep(1) # Validate. self.assertEquals(self.service.countReceivedMessages(sessionUUID), 1) self.assertEquals(self.service.receiveMessage(sessionUUID), 'test') def testSimpleServiceMessage(self): # Send service message. self.service.sendServiceMessage('test') # Alow messages to be processed. time.sleep(1) # Validate. self.assertEquals(self.service.countReceivedServiceMessages(), 1) self.assertEquals(self.service.receiveServiceMessage(), 'test') def testAdvancedMessaging(self): # Generate two session UUIDs and register them as a destinations. sessionUUID_a = str(uuid.uuid1()) sessionUUID_b = str(uuid.uuid1()) self.service.registerDestination(sessionUUID_a) self.service.registerDestination(sessionUUID_b) # Send messages. self.service.sendMessage(sessionUUID_a, 'test A1') self.service.sendMessage(sessionUUID_b, 'test B1') self.service.sendMessage(sessionUUID_b, 'test B2') self.service.sendServiceMessage('test service') # Alow messages to be processed. time.sleep(1) # Validate. self.assertEquals(self.service.countReceivedServiceMessages(), 1) self.assertEquals(self.service.receiveServiceMessage(), 'test service') self.assertEquals(self.service.countReceivedMessages(sessionUUID_a), 1) self.assertEquals(self.service.receiveMessage(sessionUUID_a), 'test A1') self.assertEquals(self.service.countReceivedMessages(sessionUUID_b), 2) self.assertEquals(self.service.receiveMessage(sessionUUID_b), 'test B1')