Ejemplo n.º 1
0
class UPnPServerConsole:

    """This class wraps ObjectConsole to implement a 
    custom console for UPnP server."""

    def __init__(self):
        
        # Task Runner
        self._task_runner = TaskRunner()

        # UPnP Server        
        self._server = UPnPServer(self._task_runner, "Product 1.0")
        
        # Switch Power Service
        service_id = "SwitchPower"
        service = SwitchPower(service_id)
        self._server.add_service(service)

        # URL Service
        service_id_2 = "URLService"
        service_2 = URLService(service_id_2)
        self._server.add_service(service_2)

        # Bookmark Service
        service_3 = BookmarkService()
        self._server.add_service(service_3)

        # Console Namespace
        name_space = {}
        root_device = self._server.get_root_device()
        name_space[root_device.name] = root_device
        name_space[service_id] = service
        name_space[service_id_2] = service_2
        name_space[service_3.get_short_service_id()] = service_3
        name_space['add_service'] = self._server.add_service
        name_space['get_service'] = self._server.get_service
        name_space['get_services'] = self._server.get_service_ids
        name_space['announce'] = self._server.announce

        self._console = ObjectConsole(self, name_space, 
                                      run="_run", 
                                      stop="_stop", name="UPnP Server")

    def _run(self):
        """Start TaskRunner."""
        self._task_runner.run_forever()

    def _stop(self):
        """Stop UPnPServer and TaskRunner"""
        self._server.close()
        self._task_runner.stop()


    def run(self):
        """Runs the UPnP Console."""
        self._console.run()
Ejemplo n.º 2
0
    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
Ejemplo n.º 3
0
    def __init__(self):
        
        # Task Runner
        self._task_runner = TaskRunner()

        # UPnP Server        
        self._server = UPnPServer(self._task_runner, "Product 1.0")
        
        # Switch Power Service
        service_id = "SwitchPower"
        service = SwitchPower(service_id)
        self._server.add_service(service)

        # URL Service
        service_id_2 = "URLService"
        service_2 = URLService(service_id_2)
        self._server.add_service(service_2)

        # Bookmark Service
        service_3 = BookmarkService()
        self._server.add_service(service_3)

        # Console Namespace
        name_space = {}
        root_device = self._server.get_root_device()
        name_space[root_device.name] = root_device
        name_space[service_id] = service
        name_space[service_id_2] = service_2
        name_space[service_3.get_short_service_id()] = service_3
        name_space['add_service'] = self._server.add_service
        name_space['get_service'] = self._server.get_service
        name_space['get_services'] = self._server.get_service_ids
        name_space['announce'] = self._server.announce

        self._console = ObjectConsole(self, name_space, 
                                      run="_run", 
                                      stop="_stop", name="UPnP Server")
Ejemplo n.º 4
0
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)