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 __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 __init__(self): self._task_runner = TaskRunner() self._server = UPnPServer(self._task_runner, SERVER_PRODUCT_NAME, SERVER_ROOT_DEVICE_CONFIG) self._client = UPnPClient(self._task_runner) # Add a couple of services self._server.add_service(SwitchPower("SwitchPower")) self._server.add_service(URLService("URLService")) self._server.add_service(BookmarkService()) # Console Namespace namespace = {} namespace['S'] = self._server namespace['C'] = self._client self._console = ObjectConsole(self, namespace, run="_run", stop="_stop", name="UPnP")
"""Mock Event Dispatcher.""" def __init__(self): pass def dispatch(self, sid, event_key, callback_url, variables): """Mock method.""" print "Notify", sid, event_key, callback_url, variables class MockServiceManager: """Mock Service Manager.""" def __init__(self): self._ed = MockEventDispatcher() def get_event_dispatcher(self): """Mock method.""" return self._ed def get_base_url(self): """Mock method.""" return "http://myhost:44444" def get_logger(self): """Mock method.""" return None SM = MockServiceManager() from Tribler.UPnP.services import SwitchPower SERVICE = SwitchPower('SwitchPower') SERVICE.set_service_manager(SM) print SERVICE.get_xml_description()