コード例 #1
0
ファイル: test_upnp.py プロジェクト: Anaconda84/Anaconda
    def setUp(self):
        """Set up test environment."""
        self._tr = TaskRunner()
        self.server = UPnPServer(self._tr, 
                            SERVER_PRODUCT_NAME,
                            SERVER_ROOT_DEVICE_CONFIG)

        self.service1 = SwitchPower("SwitchPower")
        self.service2 = URLService("URLService")
        self.server.add_service(self.service1)
        self.server.add_service(self.service2)

        self.client = UPnPClient(self._tr)
        self.thread = threading.Thread(target=self._tr.run_forever)
        self.thread.start()
        time.sleep(0.1) # wait for client to discover server
コード例 #2
0
ファイル: test_upnp.py プロジェクト: Anaconda84/Anaconda
class UPnPTestCase(unittest.TestCase):

    """TestCase for UPnPServer and UPnPClient."""

    def setUp(self):
        """Set up test environment."""
        self._tr = TaskRunner()
        self.server = UPnPServer(self._tr, 
                            SERVER_PRODUCT_NAME,
                            SERVER_ROOT_DEVICE_CONFIG)

        self.service1 = SwitchPower("SwitchPower")
        self.service2 = URLService("URLService")
        self.server.add_service(self.service1)
        self.server.add_service(self.service2)

        self.client = UPnPClient(self._tr)
        self.thread = threading.Thread(target=self._tr.run_forever)
        self.thread.start()
        time.sleep(0.1) # wait for client to discover server

    def tearDown(self):
        """Clean up test environment."""
        self.client.close()
        self.server.close()
        self._tr.stop()
        self.thread.join()

    def test_device_discovery(self):
        """
        Test if UPnPClient is able to discover the UPnPDevice
        hosted by UPnPServer.
        """
        root = self.server.get_root_device()
        self.assertTrue( root.uuid in self.client.get_device_uuids())

    def test_switchpower(self):
        """Test discovery and use of SwitchPowerService."""

        stub = self.client.get_services_by_short_id("SwitchPower")[0]

        # Test service_id
        self.assertEqual(self.service1.get_service_id(), 
                         stub.get_service_id())
        # Test service_type
        self.assertEqual(self.service1.get_service_type(), 
                         stub.get_service_type())

        # Subscribe
        evr = _EventReceiver()
        stub.subscribe(evr.handler)

        # Update SwitchPowerService by using Stub.
        value = stub.GetStatus()
        stub.SetTarget(not value)
        new_value = stub.GetStatus()
        self.assertEqual(not value, new_value, 
                         evr.get_last_event()[2])

        # Update SwitchPowerService directly.
        self.service1.set_target(not new_value)
        time.sleep(0.1) # wait for notification
        self.assertEqual(not new_value, evr.get_last_event()[2])

        # Unsubscribe
        stub.unsubscribe(evr.handler)

    def test_urlservice(self):
        """Test discovery and use of URLService."""
        stub = self.client.get_services_by_short_id("URLService")[0]
        
        # Subscribe
        evr = _EventReceiver()
        stub.subscribe(evr.handler)
   
        # Update URLService by using Stub.
        stub.GetURL()
        new_url = "http://p2p-next.org"
        stub.SetURL(new_url)
        service_url = self.service2.get_url()
        url = stub.GetURL()
        self.assertEqual(new_url, service_url, url)
        self.assertEqual(url, evr.get_last_event()[2])

        # Update URLService directly
        url2 = "http://itek.norut.no"
        self.service2.set_url(url2)
        time.sleep(0.1) # wait for notification
        self.assertEqual(url2, evr.get_last_event()[2])

        # Unsubscribe
        stub.unsubscribe(evr.handler)