Create a sub menu for all elements of the schema that are placed at t…

…he top level
opsmill · Oct 30, 2024 · fdbf0fa · fdbf0fa
1 parent 9f7300a
commit fdbf0fa
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/backend/infrahub/menu/generator.py b/backend/infrahub/menu/generator.py
@@ -109,6 +109,10 @@ async def generate_menu(db: InfrahubDatabase, branch: Branch, menu_items: list[C
                 continue
 
             if not schema.menu_placement:
+                first_element = MenuItemDict.from_schema(model=schema)
+                first_element.identifier = f"{first_element.identifier}Sub"
+                first_element.order_weight = 1
+                menu_item.children[first_element.identifier] = first_element
                 structure.data[menu_item.identifier] = menu_item
                 items_to_add[item_name] = True
             elif menu_placement := structure.find_item(name=schema.menu_placement):

diff --git a/backend/tests/unit/menu/test_generator.py b/backend/tests/unit/menu/test_generator.py
@@ -86,6 +86,7 @@ async def test_generate_menu_top_level(
     assert menu
     assert "TestMenu0" in menu.data.keys()
     assert "TestCar" in menu.data.keys()
+    assert "TestCarSub" in menu.data["TestCar"].children.keys()
 
 
 async def test_generate_menu_default(