first commit

2026-08-23 09:52:32 +02:00
commit 1a7ed8ef94
94 changed files with 10489 additions and 0 deletions
@@ -0,0 +1,62 @@
+import unittest
+from pathlib import Path
+from unittest.mock import patch
+
+from librenet_scanner.intelligence import enrich_host
+from librenet_scanner.models import Host
+from librenet_scanner.network import (
+    NetworkInterface,
+    interface_mac_address,
+    normalize_interface_mac,
+    target_contains_ip,
+)
+
+
+ROOT = Path(__file__).resolve().parents[1]
+
+
+class LocalHost046Tests(unittest.TestCase):
+    def test_mac_normalization(self):
+        self.assertEqual(normalize_interface_mac("aa:bb:cc:dd:ee:ff\n"), "AA:BB:CC:DD:EE:FF")
+        self.assertEqual(normalize_interface_mac("00:00:00:00:00:00"), "")
+        self.assertEqual(normalize_interface_mac("invalid"), "")
+
+    def test_target_contains_local_ip_for_cidr_and_range(self):
+        self.assertTrue(target_contains_ip("192.168.10.0/24", "192.168.10.1"))
+        self.assertTrue(target_contains_ip("192.168.10.1-254", "192.168.10.1"))
+        self.assertFalse(target_contains_ip("192.168.10.100-254", "192.168.10.1"))
+        self.assertFalse(target_contains_ip("192.168.5.0/24", "192.168.10.1"))
+
+    def test_sysfs_mac_is_preferred(self):
+        with patch("pathlib.Path.read_text", return_value="a6:2b:b0:a5:49:a7\n"):
+            self.assertEqual(interface_mac_address("enp42s0"), "A6:2B:B0:A5:49:A7")
+
+    def test_local_host_classification_has_priority(self):
+        host = Host("192.168.10.1", is_local=True)
+        enrich_host(host)
+        self.assertEqual(host.device_type, "Ce poste")
+
+    def test_local_flag_survives_merge(self):
+        current = Host("192.168.10.1", is_local=True, mac="AA:BB:CC:DD:EE:FF")
+        current.merge(Host("192.168.10.1", device_type="Serveur Linux"))
+        enrich_host(current)
+        self.assertTrue(current.is_local)
+        self.assertEqual(current.device_type, "Ce poste")
+
+    def test_interface_keeps_mac_without_breaking_old_positional_signature(self):
+        old_style = NetworkInterface("enp42s0", "192.168.10.1", 24, "192.168.10.0/24", False)
+        self.assertFalse(old_style.is_virtual)
+        self.assertEqual(old_style.mac, "")
+        new_style = NetworkInterface("enp42s0", "192.168.10.1", 24, "192.168.10.0/24", False, "AA:BB:CC:DD:EE:FF")
+        self.assertEqual(new_style.mac, "AA:BB:CC:DD:EE:FF")
+
+    def test_scanner_injects_local_host_additively(self):
+        scanner = (ROOT / "src/librenet_scanner/scanner.py").read_text()
+        block = scanner[scanner.index("def _discover_hosts"):scanner.index("def run(self)")]
+        self.assertIn("local_hosts = self._emit_local_host()", block)
+        self.assertIn("known_ips.update(union_host_ips(local_hosts))", block)
+        self.assertIn("interface_mac_address(iface.name)", scanner)
+
+
+if __name__ == "__main__":
+    unittest.main()