added test plan module

Author: Ammar Husain Mian Fazulul

nisystemlink/clients/test_plan/models/_execution_definition.py

@@ -82,8 +82,8 @@ class NoneExecution(JsonModel):
     action: str
     """User defined action to perform in workflow (user defined)."""
 
-    type: Literal["None"] = Field(default="None")
-    """Type of execution, default is 'None'."""
+    type: Literal["NONE"] = Field(default="NONE")
+    """Type of execution, default is 'NONE'."""
 
 
 ExecutionDefinition = Annotated[

nisystemlink/clients/test_plan/test_plan/_test_plan_client.py

@@ -4,15 +4,18 @@
 from nisystemlink.clients.core._http_configuration import HttpConfiguration
 from nisystemlink.clients.core._uplink._base_client import BaseClient
 from nisystemlink.clients.core._uplink._methods import get, post
+from nisystemlink.clients.test_plan.test_plan.models._paged_test_plans import (
+    PagedTestPlans,
+)
 from uplink import retry
 
+
 from .models import (
     _QueryTestPlansRequest,
     CreateTestPlansRequest,
     CreateTestPlansResponse,
     DeleteTestPlansRequest,
     QueryTestPlansRequest,
-    QueryTestPlansResponse,
     ScheduleTestPlansRequest,
     ScheduleTestPlansResponse,
     TestPlan,
@@ -85,27 +88,25 @@ def delete_test_plans(self, ids: DeleteTestPlansRequest) -> None:
     @post("query-testplans")
     def __query_test_plans(
         self, query_request: _QueryTestPlansRequest
-    ) -> QueryTestPlansResponse:
+    ) -> PagedTestPlans:
         """Query test plans.
 
         Args:
             query: The query to execute.
 
         Returns:
-            A QueryTestPlansResponse object containing test plans that match the query.
+            A PagedTestPlans object containing test plans that match the query.
         """
         ...
 
-    def query_test_plans(
-        self, query_request: QueryTestPlansRequest
-    ) -> QueryTestPlansResponse:
+    def query_test_plans(self, query_request: QueryTestPlansRequest) -> PagedTestPlans:
         """Query test plans.
 
         Args:
             query: The query to execute.
 
         Returns:
-            A QueryTestPlansResponse object containing test plans that match the query.
+            A PagedTestPlans object containing test plans that match the query.
         """
         projection_str = (
             [projection.name for projection in query_request.projection]

nisystemlink/clients/test_plan/test_plan/models/__init__.py

@@ -10,7 +10,6 @@
     TestPlanField,
     _QueryTestPlansRequest,
 )
-from ._query_test_plans_response import QueryTestPlansResponse
 from ._schedule_test_plans_request import ScheduleTestPlansRequest
 from ._state import State
 from ._test_plan import TestPlan
@@ -19,5 +18,6 @@
 from ._update_test_plans_response import UpdateTestPlansResponse
 from ._workflow_definition import WorkflowDefinition
 from ._order_by import OrderBy
+from ._paged_test_plans import PagedTestPlans
 
 # flake8: noqa

nisystemlink/clients/test_plan/test_plan/models/_paged_test_plans.py

@@ -0,0 +1,16 @@
+from typing import List, Optional
+
+from nisystemlink.clients.core._uplink._with_paging import WithPaging
+from nisystemlink.clients.test_plan.test_plan.models._test_plan import TestPlan
+
+
+class PagedTestPlans(WithPaging):
+    """The response containing the list of products, total count of products and the continuation
+    token if applicable.
+    """
+
+    test_plans: List[TestPlan]
+    """A list of all the products in this page."""
+
+    total_count: Optional[int] = None
+    """The total number of products that match the query."""

nisystemlink/clients/test_plan/test_plan/models/_query_test_plans_response.py

@@ -18,4 +18,3 @@ class ScheduleTestPlansResponse(JsonModel):
 
     error: Optional[ApiError] = None
     """The error that occurred when scheduling the test plans."""
-