Example #1
0
 def test_open_port(self):
     port_config = get_port_config([])
     upnpc = MagicMock()
     upnpc.port_open_on_router = MagicMock(side_effect=[True, True, False, False, False])
     mapper = PortMapper(port_config, upnpc)
     existing_ports = []
     self.assertEquals([2001, 2002, 2003], mapper.find_available_ports_to_try(existing_ports, 1999, 3))
Example #2
0
 def test_local_port(self):
     port_config = get_port_config([])
     upnpc = MagicMock()
     upnpc.port_open_on_router = MagicMock(return_value=False)
     mapper = PortMapper(port_config, upnpc)
     existing_ports = [2000, 2001, 2003]
     self.assertEquals([80, 2002, 2004], mapper.find_available_ports_to_try(existing_ports, 80, 3))
Example #3
0
 def test_first_gap(self):
     port_config = get_port_config([])
     upnpc = MagicMock()
     upnpc.port_open_on_router = MagicMock(return_value=False)
     mapper = PortMapper(port_config, upnpc)
     existing_ports = [2000, 2001, 2003]
     ports_to_try = mapper.find_available_ports_to_try(existing_ports, 2000, 3)
     self.assertEquals([2002, 2004, 2005], ports_to_try)