def test_get_service(self): service_config = get_service_config([]) port_config = get_port_config([]) upnpc = get_upnpc(external_ip='192.167.44.52') port_mapper = get_port_mapper(port_config, upnpc) domain_config = get_domain_config(None) insider_config = get_insider_config('domain.com', 'http://api.domain.com') dns = Dns(insider_config, domain_config, service_config, port_mapper, '127.0.0.1') dns.add_service("ownCloud", "http", "_http._tcp", 80, url="owncloud") service = dns.get_service("ownCloud") self.assertIsNotNone(service) self.assertEquals("ownCloud", service.name) self.assertEquals("_http._tcp", service.type) self.assertEquals(80, service.port) self.assertEquals("owncloud", service.url)
def test_add_service(self): service_config = get_service_config([]) port_config = get_port_config([]) upnpc = get_upnpc(external_ip='192.167.44.52') port_mapper = get_port_mapper(port_config, upnpc) domain_config = get_domain_config(None) insider_config = get_insider_config('domain.com', 'http://api.domain.com') dns = Dns(insider_config, domain_config, service_config, port_mapper, '127.0.0.1') dns.add_service("ownCloud", "http", "_http._tcp", 80, url="owncloud") services = service_config.load() self.assertEquals(1, len(services)) service = services[0] self.assertEquals("ownCloud", service.name) self.assertEquals("_http._tcp", service.type) self.assertEquals(80, service.port) self.assertEquals("owncloud", service.url) mappings = port_config.load() self.assertEquals(1, len(mappings)) mapping = mappings[0] self.assertEquals(80, mapping.local_port)