first commit

tests/test_v0411.py

@@ -0,0 +1,32 @@
+import unittest
+
+from librenet_scanner.models import Host
+from librenet_scanner.visual_identity import device_icon_key_for_host, os_icon_key_for_host
+
+
+class VisualIdentitySplitRegressionTests(unittest.TestCase):
+    """0.4.12 deliberately replaces the product-specific 0.4.11 icon model."""
+
+    def test_openwrt_is_linux_os_but_network_equipment_type_is_independent(self):
+        host = Host("192.0.2.1", os_name="OpenWrt 23.05", device_type="Équipement réseau")
+        self.assertEqual(os_icon_key_for_host(host), "linux")
+        self.assertEqual(device_icon_key_for_host(host), "network-device")
+
+    def test_opnsense_is_bsd_os_and_firewall_equipment(self):
+        host = Host("192.0.2.3", hostname="opns01.local", os_name="FreeBSD 13", device_type="Pare-feu / routeur")
+        self.assertEqual(os_icon_key_for_host(host), "bsd")
+        self.assertEqual(device_icon_key_for_host(host), "firewall")
+
+    def test_proxmox_is_linux_os_and_hypervisor_equipment(self):
+        host = Host("192.0.2.4", os_name="Linux 6.x", device_type="Hyperviseur Proxmox")
+        self.assertEqual(os_icon_key_for_host(host), "linux")
+        self.assertEqual(device_icon_key_for_host(host), "hypervisor")
+
+    def test_windows_workstation_is_windows_os_and_workstation_equipment(self):
+        host = Host("192.0.2.2", os_name="Microsoft Windows 11", device_type="Poste / serveur Windows")
+        self.assertEqual(os_icon_key_for_host(host), "windows")
+        self.assertEqual(device_icon_key_for_host(host), "workstation")
+
+
+if __name__ == "__main__":
+    unittest.main()