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()