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()
