def test_private_ip():
responses.add(responses.GET,
"http://api.domain.com/probe/port_v2",
status=200,
body='{"message": "OK", "device_ip": "0"}',
content_type="application/json")
prober = PortProber('http://api.domain.com', '')
result, message = prober.probe_port(80, 'http', '0')
assert len(responses.calls) == 0
assert result is False
def test_public_ip():
responses.add(responses.GET,
"http://api.domain.com/probe/port_v2",
status=200,
body='{"message": "OK", "device_ip": "0"}',
content_type="application/json")
prober = PortProber('http://api.domain.com', '')
result, message = prober.probe_port(80, 'http', '8.8.8.8')
assert result is True
assert 'ip=' in responses.calls[0].request.url
def _get_port_prober(self):
if self.user_platform_config.is_redirect_enabled():
return PortProber(
self.user_platform_config.get_redirect_api_url(),
self.user_platform_config.get_domain_update_token())
else:
return NoneProber()
def get_drill(self, external_access):
if not external_access or not self.user_platform_config.get_port_drilling_enabled(
):
return NonePortDrill()
drill = None
mapper = provide_mapper(self.nat_pmp_port_mapper,
self.upnp_port_mapper)
if mapper:
prober = PortProber(
self.user_platform_config.get_redirect_api_url(),
self.user_platform_config.get_domain_update_token())
drill = PortDrill(self.port_config, mapper, prober)
return drill
def get_drill(self, upnp_enabled, external_access, manual_public_ip, manual_public_port):
if not external_access:
return NonePortDrill()
drill = None
if upnp_enabled:
mapper = self.port_mapper_factory.provide_mapper()
else:
mapper = ManualPortMapper(manual_public_ip, manual_public_port)
if mapper:
prober = PortProber(
self.user_platform_config.get_redirect_api_url(),
self.user_platform_config.get_domain_update_token())
drill = PortDrill(self.port_config, mapper, prober)
return drill