from types import SimpleNamespace
import unittest

from meshtastic.protobuf import config_pb2, module_config_pb2

from contact.ui.menus import generate_menu_from_protobuf


class MenusTests(unittest.TestCase):
    def test_main_menu_includes_factory_reset_config_after_factory_reset(self) -> None:
        local_node = SimpleNamespace(
            localConfig=config_pb2.Config(),
            moduleConfig=module_config_pb2.ModuleConfig(),
            getChannelByChannelIndex=lambda _: None,
        )
        interface = SimpleNamespace(
            localNode=local_node,
            getMyNodeInfo=lambda: {
                "user": {"longName": "Test User", "shortName": "TU", "isLicensed": False},
                "position": {"latitude": 0.0, "longitude": 0.0, "altitude": 0},
            },
        )

        menu = generate_menu_from_protobuf(interface)
        keys = list(menu["Main Menu"].keys())

        self.assertLess(keys.index("Factory Reset"), keys.index("factory_reset_config"))
        self.assertEqual(keys[keys.index("Factory Reset") + 1], "factory_reset_config")