fail branch report command on main branch

Author: wvandeun

infrahub_sdk/ctl/branch.py

@@ -1,4 +1,5 @@
 import logging
+import sys
 from datetime import datetime, timezone
 from typing import TYPE_CHECKING
 
@@ -195,6 +196,10 @@ async def report(  # noqa: PLR0915
 ) -> None:
     """Generate branch cleanup status report."""
 
+    if branch_name == "main":
+        console.print("[red]Cannot create a report for the main branch!")
+        sys.exit(1)
+
     client = initialize_client()
 
     # Fetch branch metadata first (needed for diff creation)

tests/unit/ctl/test_branch_report.py

@@ -199,6 +199,15 @@ def test_branch_report_command_without_proposed_change(
     assert "No proposed changes for this branch" in result.stdout
 
 
+def test_branch_report_command_main_branch() -> None:
+    """Test branch report CLI command on main branch."""
+    runner = CliRunner()
+    result = runner.invoke(app, ["report", "main"])
+
+    assert result.exit_code == 1
+    assert "Cannot create a report for the main branch!" in result.stdout
+
+
 @pytest.fixture
 async def schema_with_proposed_change() -> dict:
     """Schema fixture that includes CoreProposedChange with is_draft."""

tests/unit/sdk/test_node.py

@@ -1736,10 +1736,20 @@ async def test_update_input_data__with_relationships_01(
     expected_result_with_property = {
         "data": {
             "id": "llllllll-llll-llll-llll-llllllllllll",
-            "name": {"is_protected": True, "is_visible": True, "value": "DFW", "updated_at": "2024-01-15T10:30:00.000000Z"},
+            "name": {
+                "is_protected": True,
+                "is_visible": True,
+                "value": "DFW",
+                "updated_at": "2024-01-15T10:30:00.000000Z",
+            },
             "primary_tag": {"id": "gggggggg-gggg-gggg-gggg-gggggggggggg"},
             "tags": [{"id": "gggggggg-gggg-gggg-gggg-gggggggggggg"}, {"id": "rrrrrrrr-rrrr-rrrr-rrrr-rrrrrrrrrrrr"}],
-            "type": {"is_protected": True, "is_visible": True, "value": "SITE","updated_at": "2024-01-15T10:30:00.000000Z"},
+            "type": {
+                "is_protected": True,
+                "is_visible": True,
+                "value": "SITE",
+                "updated_at": "2024-01-15T10:30:00.000000Z",
+            },
         },
     }
 
@@ -1900,10 +1910,20 @@ async def test_update_input_data_empty_relationship(
     expected_result_with_property = {
         "data": {
             "id": "llllllll-llll-llll-llll-llllllllllll",
-            "name": {"is_protected": True, "is_visible": True, "value": "DFW", "updated_at": "2024-01-15T10:30:00.000000Z"},
+            "name": {
+                "is_protected": True,
+                "is_visible": True,
+                "value": "DFW",
+                "updated_at": "2024-01-15T10:30:00.000000Z",
+            },
             "primary_tag": None,
             "tags": [],
-            "type": {"is_protected": True, "is_visible": True, "value": "SITE", "updated_at": "2024-01-15T10:30:00.000000Z"},
+            "type": {
+                "is_protected": True,
+                "is_visible": True,
+                "value": "SITE",
+                "updated_at": "2024-01-15T10:30:00.000000Z",
+            },
         },
     }