first commit

tests/test_v041.py

@@ -0,0 +1,26 @@
+import unittest
+
+from librenet_scanner.ui_layout import balanced_column_widths, compact_warning
+
+
+class UI041LayoutTests(unittest.TestCase):
+    def test_balanced_columns_fill_viewport(self):
+        widths = balanced_column_widths(1200, show_evolution=True)
+        self.assertEqual(len(widths), 4)
+        self.assertGreaterEqual(widths[0], 300)
+        self.assertGreater(widths[2], widths[1])
+        self.assertLessEqual(sum(widths), 1200)
+        self.assertGreaterEqual(sum(widths), 1180)
+
+    def test_evolution_can_be_hidden(self):
+        widths = balanced_column_widths(1000, show_evolution=False)
+        self.assertEqual(widths[3], 0)
+        self.assertGreater(widths[0], 300)
+
+    def test_arp_warning_is_shortened(self):
+        message = "arp-scan n'a pas les privilèges nécessaires pour le scan ARP. Les MAC seront récupérées via ip neigh."
+        self.assertEqual(compact_warning(message), "ARP limité : certaines adresses MAC peuvent être incomplètes.")
+
+
+if __name__ == "__main__":
+    unittest.main()