def test_add_host_with_invalid_port_options(self):
firstPort = randomPort(max=90)
lastPort = randomPort(91, 100)
self.mock_scan_manager.addHost.side_effect = PortRangeBackwardException()
result = self.runner.invoke(NmapCommands.add_host, [self.address, '-fp', lastPort,
'-lp', firstPort])
self.assertEqual(result.exit_code, 0)
self.assertEqual(result.output, 'First port must be lower than last port\n')
def test_list_hosts_with_one_host(self):
hosts = [Host(i, generateIP(), '{}-{}'.format(randomPort(max=90), randomPort(91, 100)))
for i in range(2)]
listResult = '\n'.join([str(host) for host in hosts])
self.mock_scan_manager.getHosts.return_value = hosts
result = self.runner.invoke(NmapCommands.list_hosts)
self.assertEqual(result.output, listResult + '\n')
def test_add_host_with_valid_ports_range(self):
firstPort = randomPort(max=90)
lastPort = randomPort(91, 100)
host = self.scanManager.addHost(self.address, firstPort, lastPort)
hosts = self.scanManager.hosts
self.assertEqual(len(hosts), 1)
self.assertEqual(host.ip, self.address)
self.assertEqual(host.ports, '{}-{}'.format(firstPort, lastPort))
def test_add_host_with_port_options(self):
hostId = random.randint(0, 3)
firstPort = randomPort(max=90)
lastPort = randomPort(91, 100)
ports = '{}-{}'.format(firstPort, lastPort)
stub_host = Host(hostId, self.address, ports)
self.mock_scan_manager.addHost.return_value = stub_host
result = self.runner.invoke(NmapCommands.add_host, [self.address, '-fp', firstPort,
'-lp', lastPort])
self.assertEqual(result.output, 'Added: Id: {}\n{}:{}\n'.format(hostId, self.address, ports))
def test_list_hosts_results_with_host_ids(self):
self.__addHost(self.address, randomPort())
results = self.scanManager.getHostResults(False, 0)
self.assertEqual(len(results), 1)
self.assertEqual(results, self.results)
def test_add_host_with_first_port(self):
firstPort = randomPort()
host = self.scanManager.addHost(self.address, firstPort)
self.assertIsInstance(host, Host)
self.assertEqual(host.ip, self.address)
self.assertEqual(host.ports, str(firstPort))
def test_add_host_with_last_port_option(self):
hostId = random.randint(0, 3)
lastPort = randomPort()
stub_host = Host(hostId, self.address, None)
self.mock_scan_manager.addHost.return_value = stub_host
result = self.runner.invoke(NmapCommands.add_host, [self.address, '-lp', lastPort])
self.assertEqual(result.output, 'Added: Id: {}\n{}\n'.format(hostId, self.address))
def setUp(self):
super(NmapScanManagerScanHostTestCase, self).setUp()
self.port = randomPort()
self.stub_result = {
'nmap': {
'scaninfo': {
'tcp': {
'services': str(self.port)
}
}
}
}
def test_add_host_with_last_port(self):
host = self.scanManager.addHost(self.address, lastPort=randomPort())
self.assertIsInstance(host, Host)
self.assertEqual(host.ip, self.address)
self.assertEqual(host.ports, None)
def test_add_host_with_ports_range_inverted(self):
firstPort = randomPort(90, 100)
lastPort = randomPort(80, 89)
with self.assertRaises(PortRangeBackwardException):
self.scanManager.addHost(self.address, firstPort, lastPort)
def test_list_hosts_results_all_host_set_to_true(self):
self.__addHost(generateIP(), randomPort())
self.__addHost(generateIP(), randomPort())
results = self.scanManager.getHostResults(True)
self.assertEqual(len(results), 2)
self.assertEqual(results, self.results)
def setUp(self):
super(NmapCommandsListHostResultsTestCase, self).setUp()
self.address = generateIP()
self.hosts = [Host(i, generateIP(), '{}-{}'.format(randomPort(max=90), randomPort(91, 100)))
for i in range(2)]
