lucius-mcp 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

src/tools/search.py

@@ -122,12 +122,12 @@ async def search_test_cases(
     Examples:
         search_test_cases(query="login")
         → "Found 5 test cases matching 'login':
-           - [TC-1] User Login Flow (tags: smoke, auth)
-           - [TC-2] Admin Login Test (tags: admin)"
+           - [#1] User Login Flow (tags: smoke, auth)
+           - [#2] Admin Login Test (tags: admin)"
 
         search_test_cases(aql='status="failed" and tag="regression"')
         → "Found 12 test cases matching 'status=\"failed\" and tag=\"regression\"':
-           - [TC-5] Critical Path Test ..."
+           - [#5] Critical Path Test ..."
     """
     # Validate that at least one search parameter is provided
     if not aql and not query:

src/tools/shared_steps.py

@@ -1,4 +1,4 @@
-from typing import Annotated, Any
+from typing import Annotated
 
 from pydantic import Field
 
@@ -9,7 +9,7 @@ from src.services.shared_step_service import SharedStepService
 async def create_shared_step(
     name: Annotated[str, Field(description='The name of the shared step (e.g., "Login as Admin").')],
     steps: Annotated[
-        list[dict[str, Any]] | None,
+        list[dict[str, object]] | None,
         Field(
             description="Optional list of steps. Each step is a dictionary with:"
             ' - action (str): The step description (e.g., "Enter username").'
@@ -86,9 +86,11 @@ async def update_shared_step(
     step_id: Annotated[int, Field(description="The shared step ID to update (required).")],
     name: Annotated[str | None, Field(description="New name for the shared step (optional).")] = None,
     description: Annotated[str | None, Field(description="New description (optional).")] = None,
+    confirm: Annotated[bool, Field(description="Must be set to True to proceed with update. Safety measure.")] = False,
     project_id: Annotated[int | None, Field(description="Optional override for the default Project ID.")] = None,
 ) -> str:
     """Update an existing shared step.
+    ⚠️ CAUTION: Destructive.
 
     ⚠️ IMPORTANT: Changes propagate to ALL test cases using this shared step.
 
@@ -100,6 +102,8 @@ async def update_shared_step(
             Found via list_shared_steps or in the Allure URL.
         name: New name for the shared step (optional).
         description: New description (optional).
+        confirm: Must be set to True to proceed with update.
+            This is a safety measure to prevent accidental updates.
         project_id: Optional override for the default Project ID.
 
     Returns:
@@ -108,9 +112,16 @@ async def update_shared_step(
     Example:
         update_shared_step(
             step_id=789,
-            name="Login as Admin (Updated)"
+            name="Login as Admin (Updated)",
+            confirm=True
         )
     """
+    if not confirm:
+        return (
+            "⚠️ Update requires confirmation.\n\n"
+            "Changes propagate to ALL test cases using this shared step. "
+            f"Please call again with confirm=True to proceed with updating shared step {step_id}."
+        )
 
     async with AllureClient.from_env(project=project_id) as client:
         service = SharedStepService(client=client)
@@ -136,6 +147,7 @@ async def delete_shared_step(
     project_id: Annotated[int | None, Field(description="Optional override for the default Project ID.")] = None,
 ) -> str:
     """Delete a shared step from the library.
+    ⚠️ CAUTION: Destructive.
 
     ⚠️ CAUTION: If this shared step is used by test cases, deleting it
     will break those references.
@@ -157,13 +169,16 @@ async def delete_shared_step(
     """
     if not confirm:
         return (
-            "⚠️ Delete confirmation required\n\n"
-            "To delete this shared step, set confirm=True.\n\n"
-            "WARNING: Deleting a shared step used by test cases will break those references."
+            "⚠️ Deletion requires confirmation.\n\n"
+            "Deleting a shared step used by test cases will be breaking those references. "
+            f"Please call again with confirm=True to proceed with archiving shared step {step_id}."
         )
 
     async with AllureClient.from_env(project=project_id) as client:
         service = SharedStepService(client=client)
-        await service.delete_shared_step(step_id=step_id)
+        deleted = await service.delete_shared_step(step_id=step_id)
 
-        return f"🗑️  Archived Shared Step {step_id}\n\nThe shared step has been successfully archived."
+        if deleted:
+            return f"✅ Archived Shared Step {step_id}\n\nThe shared step has been successfully archived."
+        else:
+            return f"ℹ️ Shared Step {step_id} was already archived or doesn't exist."  # noqa: RUF001

src/tools/unlink_shared_step.py

@@ -1,6 +1,6 @@
 """Tool for unlinking Shared Steps from Test Cases."""
 
-from typing import Annotated, Any
+from typing import Annotated
 
 from pydantic import Field
 
@@ -9,13 +9,14 @@ from src.client.generated.models.shared_step_step_dto import SharedStepStepDto
 from src.services.test_case_service import TestCaseService
 
 
-def _format_steps(scenario: Any) -> str:
+def _format_steps(scenario: object) -> str:
     """Format steps for display."""
-    if not scenario or not scenario.steps:
+    steps = getattr(scenario, "steps", None)
+    if not isinstance(steps, list) or not steps:
         return "No steps."
 
     output = []
-    for i, step in enumerate(scenario.steps):
+    for i, step in enumerate(steps):
         if step.actual_instance and isinstance(step.actual_instance, SharedStepStepDto):
             output.append(f"{i + 1}. [Shared Step] ID: {step.actual_instance.shared_step_id}")
         else:
@@ -30,8 +31,12 @@ async def unlink_shared_step(
     test_case_id: Annotated[int, Field(description="The test case to modify.")],
     shared_step_id: Annotated[int, Field(description="The shared step to unlink.")],
     project_id: Annotated[int | None, Field(description="Optional override for the default Project ID.")] = None,
+    confirm: Annotated[
+        bool, Field(description="Must be set to True to proceed with unlinking. Safety measure.")
+    ] = False,
 ) -> str:
     """Remove a shared step reference from a test case.
+    ⚠️ CAUTION: Destructive.
 
     Removes the link to the shared step. The test case will no longer
     include those steps at execution time.
@@ -40,6 +45,8 @@ async def unlink_shared_step(
         test_case_id: The test case to modify.
         shared_step_id: The shared step to unlink.
         project_id: Optional override for the default Project ID.
+        confirm: Must be set to True to proceed with unlinking.
+            This is a safety measure to prevent accidental unlinking.
 
     Returns:
         Confirmation with updated step list.
@@ -48,8 +55,15 @@ async def unlink_shared_step(
     remains in the library and other test cases are unaffected.
 
     Example:
-        unlink_shared_step(test_case_id=12345, shared_step_id=789)
+        unlink_shared_step(test_case_id=12345, shared_step_id=789, confirm=True)
     """
+    if not confirm:
+        return (
+            "⚠️ Unlinking requires confirmation.\n\n"
+            "This will remove a shared step from the test case scenario. "
+            f"Please call again with confirm=True to proceed with unlinking "
+            f"shared step {shared_step_id} from test case {test_case_id}."
+        )
 
     async with AllureClient.from_env(project=project_id) as client:
         service = TestCaseService(client=client)

src/tools/update_custom_field_value.py

@@ -0,0 +1,55 @@
+from typing import Annotated
+
+from pydantic import Field
+
+from src.client import AllureClient
+from src.services.custom_field_value_service import CustomFieldValueService
+
+
+async def update_custom_field_value(
+    cfv_id: Annotated[int, Field(description="Custom field value ID to update.")],
+    name: Annotated[str, Field(description="New name for the custom field value.")],
+    custom_field_id: Annotated[
+        int | None,
+        Field(description="Project-scoped custom field ID (optional, resolves by name if missing)."),
+    ] = None,
+    custom_field_name: Annotated[
+        str | None,
+        Field(description="Custom field name to resolve when custom_field_id is not provided."),
+    ] = None,
+    project_id: Annotated[int | None, Field(description="Optional override for the default Project ID.")] = None,
+    confirm: Annotated[bool, Field(description="Must be set to True to proceed with update. Safety measure.")] = False,
+) -> str:
+    """Update an existing custom field value.
+    ⚠️ CAUTION: Destructive.
+
+    Args:
+        cfv_id: Custom field value ID to update.
+        name: New name for the custom field value.
+        custom_field_id: Project-scoped custom field ID (optional, resolves by name if missing).
+        custom_field_name: Custom field name to resolve when custom_field_id is not provided.
+        project_id: Optional Allure TestOps project ID override.
+        confirm: Must be set to True to proceed with update.
+            This is a safety measure to prevent accidental updates.
+
+    Returns:
+        Confirmation message indicating whether changes were made.
+    """
+    if not confirm:
+        return (
+            "⚠️ Update requires confirmation.\n\n"
+            "Updating a custom field value affects all test cases using it. "
+            f"Please call again with confirm=True to proceed with updating custom field value {cfv_id}."
+        )
+
+    async with AllureClient.from_env(project=project_id) as client:
+        service = CustomFieldValueService(client=client)
+        await service.update_custom_field_value(
+            project_id=project_id,
+            cfv_id=cfv_id,
+            custom_field_id=custom_field_id,
+            custom_field_name=custom_field_name,
+            name=name,
+        )
+
+        return f"✅ Custom field value {cfv_id} updated successfully!\nNew name: {name}"

src/tools/update_test_case.py

@@ -33,14 +33,48 @@ async def update_test_case(  # noqa: C901
     expected_result: Annotated[str | None, Field(description="Global expected result for the test case")] = None,
     status_id: Annotated[int | None, Field(description="ID of the test case status")] = None,
     test_layer_id: Annotated[int | None, Field(description="ID of the test layer")] = None,
+    test_layer_name: Annotated[str | None, Field(description="Name of the test layer")] = None,
     workflow_id: Annotated[int | None, Field(description="ID of the workflow")] = None,
     links: Annotated[
         list[dict[str, str]] | None,
         Field(description="New list of external links. Each dict has 'name', 'url', and optional 'type'."),
     ] = None,
+    # Issue Linking
+    issues: Annotated[
+        list[str] | None,
+        Field(description="List of issue keys to ADD (e.g. ['PROJ-123'])."),
+    ] = None,
+    remove_issues: Annotated[
+        list[str] | None,
+        Field(description="List of issue keys to REMOVE."),
+    ] = None,
+    clear_issues: Annotated[
+        bool | None,
+        Field(description="If True, remove ALL issues from the test case."),
+    ] = None,
+    integration_id: Annotated[
+        int | None,
+        Field(
+            description=(
+                "Optional integration ID for issue linking (use list_integrations to find IDs). "
+                "Required when multiple integrations exist and adding issues. Mutually exclusive with integration_name."
+            )
+        ),
+    ] = None,
+    integration_name: Annotated[
+        str | None,
+        Field(
+            description=(
+                "Optional integration name for issue linking (exact case-sensitive match). "
+                "Required when multiple integrations exist and adding issues. Mutually exclusive with integration_id."
+            )
+        ),
+    ] = None,
     project_id: Annotated[int | None, Field(description="Optional override for the default Project ID.")] = None,
+    confirm: Annotated[bool, Field(description="Must be set to True to proceed with update. Safety measure.")] = False,
 ) -> str:
     """Update an existing test case in Allure TestOps.
+    ⚠️ CAUTION: Destructive.
 
     Performs a partial update: only supplied fields are sent to the API. When
     provided, ``steps`` replace all existing steps, and ``attachments`` replace
@@ -61,10 +95,18 @@ async def update_test_case(  # noqa: C901
         expected_result: Global expected result for the test case.
         status_id: ID of the test case status.
         test_layer_id: ID of the test layer.
+        test_layer_name: Name of the test layer.
         workflow_id: ID of the workflow.
         links: New list of external links. Each dict has ``name``, ``url``,
             and optional ``type``.
+        issues: List of issue keys to ADD (e.g. ['PROJ-123']).
+        remove_issues: List of issue keys to REMOVE.
+        clear_issues: If True, remove ALL issues from the test case.
+        integration_id: Optional integration ID for issue linking.
+        integration_name: Optional integration name for issue linking.
         project_id: Optional override for the default Project ID.
+        confirm: Must be set to True to proceed with update.
+            This is a safety measure to prevent accidental updates.
 
     Returns:
         A confirmation message summarizing the update.
@@ -73,6 +115,12 @@ async def update_test_case(  # noqa: C901
         AuthenticationError: If no API token available from environment or
             arguments.
     """
+    if not confirm:
+        return (
+            "⚠️ Update requires confirmation.\n\n"
+            "This will modify test case properties and may overwrite existing data. "
+            f"Please call again with confirm=True to proceed with updating test case {test_case_id}."
+        )
 
     async with AllureClient.from_env(project=project_id) as client:
         service = TestCaseService(client=client)
@@ -89,8 +137,14 @@ async def update_test_case(  # noqa: C901
             expected_result=expected_result,
             status_id=status_id,
             test_layer_id=test_layer_id,
+            test_layer_name=test_layer_name,
             workflow_id=workflow_id,
             links=links,
+            issues=issues,
+            remove_issues=remove_issues,
+            clear_issues=clear_issues,
+            integration_id=integration_id,
+            integration_name=integration_name,
         )
 
         updated_case = await service.update_test_case(test_case_id, update_data)
@@ -130,10 +184,14 @@ async def update_test_case(  # noqa: C901
             current_status_id = getattr(current_status, "id", None) if current_status else None
             if status_id != current_status_id:
                 changes.append("status updated")
-        if test_layer_id is not None:
+        if test_layer_id is not None or test_layer_name is not None:
             current_test_layer = getattr(current_case, "test_layer", None)
             current_test_layer_id = getattr(current_test_layer, "id", None) if current_test_layer else None
-            if test_layer_id != current_test_layer_id:
+
+            updated_test_layer = getattr(updated_case, "test_layer", None)
+            updated_test_layer_id = getattr(updated_test_layer, "id", None) if updated_test_layer else None
+
+            if updated_test_layer_id != current_test_layer_id:
                 changes.append("test layer updated")
         if workflow_id is not None:
             current_workflow = getattr(current_case, "workflow", None)
@@ -145,6 +203,13 @@ async def update_test_case(  # noqa: C901
             if normalize_links(current_links) != normalize_links(links):
                 changes.append("links updated")
 
+        if issues:
+            changes.append(f"added {len(issues)} issues ({', '.join(issues)})")
+        if remove_issues:
+            changes.append(f"removed {len(remove_issues)} issues ({', '.join(remove_issues)})")
+        if clear_issues:
+            changes.append("cleared all issues")
+
         summary = ", ".join(changes) if changes else "No changes made (idempotent)"
 
         return f"Test Case {updated_case.id} updated successfully. Changes: {summary}"

src/tools/update_test_layer.py

@@ -12,17 +12,28 @@ async def update_test_layer(
     layer_id: Annotated[int, Field(description="ID of the test layer to update.")],
     name: Annotated[str, Field(description="New name for the test layer.")],
     project_id: Annotated[int | None, Field(description="Allure TestOps project ID.")] = None,
+    confirm: Annotated[bool, Field(description="Must be set to True to proceed with update. Safety measure.")] = False,
 ) -> str:
     """Update an existing test layer's name.
+    ⚠️ CAUTION: Destructive.
 
     Args:
-        layer_id: ID of the test layer to update
-        name: New name for the test layer
-        project_id: Optional project ID override
+        layer_id: ID of the test layer to update.
+        name: New name for the test layer.
+        project_id: Optional Allure TestOps project ID override.
+        confirm: Must be set to True to proceed with update.
+            This is a safety measure to prevent accidental updates.
 
     Returns:
-        Confirmation message indicating whether changes were made
+        Confirmation message indicating whether changes were made.
     """
+    if not confirm:
+        return (
+            "⚠️ Update requires confirmation.\n\n"
+            "Renaming a test layer affects all associated test cases. "
+            f"Please call again with confirm=True to proceed with updating test layer {layer_id}."
+        )
+
     async with AllureClient.from_env(project=project_id) as client:
         service = TestLayerService(client)
         updated, changed = await service.update_test_layer(layer_id=layer_id, name=name)

src/tools/update_test_layer_schema.py

@@ -13,18 +13,29 @@ async def update_test_layer_schema(
     test_layer_id: Annotated[int | None, Field(description="New test layer ID to link to.")] = None,
     key: Annotated[str | None, Field(description="New schema key.")] = None,
     project_id: Annotated[int | None, Field(description="Allure TestOps project ID.")] = None,
+    confirm: Annotated[bool, Field(description="Must be set to True to proceed with update. Safety measure.")] = False,
 ) -> str:
     """Update an existing test layer schema.
+    ⚠️ CAUTION: Destructive.
 
     Args:
-        schema_id: ID of the schema to update
-        test_layer_id: Optional new test layer ID
-        key: Optional new schema key
-        project_id: Optional project ID override
+        schema_id: ID of the schema to update.
+        test_layer_id: Optional new test layer ID to link to.
+        key: Optional new schema key.
+        project_id: Optional Allure TestOps project ID override.
+        confirm: Must be set to True to proceed with update.
+            This is a safety measure to prevent accidental updates.
 
     Returns:
-        Confirmation message indicating whether changes were made
+        Confirmation message indicating whether changes were made.
     """
+    if not confirm:
+        return (
+            "⚠️ Update requires confirmation.\n\n"
+            "Modifying a schema affects how test layers are automatically assigned. "
+            f"Please call again with confirm=True to proceed with updating test layer schema {schema_id}."
+        )
+
     async with AllureClient.from_env(project=project_id) as client:
         service = TestLayerService(client)
         updated, changed = await service.update_test_layer_schema(