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