update execution actions

Author: Aravindhan Palanisamy

nisystemlink/clients/test_plan/models/_execution_definition.py

@@ -69,12 +69,34 @@ class NoneExecution(JsonModel):
     """Type of execution, default is 'NONE'."""
 
 
+class ScheduleExecution(JsonModel):
+    """Represents a scheduled execution definition."""
+
+    action: str
+    """User defined action to perform in workflow (user defined)."""
+
+    type: Literal["SCHEDULE"] = Field(default="SCHEDULE")
+    """Type of execution, default is 'SCHEDULE'."""
+
+
+class UnscheduleExecution(JsonModel):
+    """Represents an unscheduled execution definition."""
+
+    action: str
+    """User defined action to perform in workflow (user defined)."""
+
+    type: Literal["UNSCHEDULE"] = Field(default="UNSCHEDULE")
+    """Type of execution, default is 'UNSCHEDULE'."""
+
+
 ExecutionDefinition = Annotated[
     Union[
         NotebookExecution,
         ManualExecution,
         JobExecution,
         NoneExecution,
+        ScheduleExecution,
+        UnscheduleExecution,
     ],
     Field(discriminator="type"),
 ]

tests/integration/test_plan/test_test_plan_client.py

@@ -96,7 +96,7 @@ def _create_test_plan_templates(
 @pytest.mark.enterprise
 class TestTestPlanClient:
 
-    _workspace = "33eba2fe-fe42-48a1-a47f-a6669479a8aa"
+    _workspace = "2300760d-38c4-48a1-9acb-800260812337"
 
     _dashboard = Dashboard(
         id="DashBoardId", variables={"product": "PXIe-4080", "location": "Lab1"}