25 lines
1018 B
Python
25 lines
1018 B
Python
import unittest
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
class Discovery044RegressionTests(unittest.TestCase):
|
|
def test_standard_discovers_before_port_scan(self):
|
|
scanner = (ROOT / 'src/librenet_scanner/scanner.py').read_text()
|
|
block = scanner[scanner.index('def _standard_baseline'):scanner.index('def _standard_scan')]
|
|
discovery = block.index('self._standard_discovery(')
|
|
ports = block.index('self._naabu_ports(')
|
|
self.assertLess(discovery, ports)
|
|
self.assertIn('ips = sorted(known_ips', block)
|
|
|
|
def test_standard_discovery_disables_dns_and_bounds_retries(self):
|
|
scanner = (ROOT / 'src/librenet_scanner/scanner.py').read_text()
|
|
block = scanner[scanner.index('def _standard_discovery'):scanner.index('def _standard_baseline')]
|
|
self.assertIn('"-sn", "-n", "-T4", "--max-retries", "1"', block)
|
|
self.assertIn('timeout_seconds=discovery_timeout', block)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|