From c837a9d91083b3ecaa55624e8288c06400cbe080 Mon Sep 17 00:00:00 2001 From: Ximo Date: Fri, 3 Jan 2025 15:16:08 +0100 Subject: [PATCH] added extract all parameter to test_plan_client method --- .../azure/devops/released/test_plan/test_plan_client.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/azure-devops/azure/devops/released/test_plan/test_plan_client.py b/azure-devops/azure/devops/released/test_plan/test_plan_client.py index d562675f..f0f0ac98 100644 --- a/azure-devops/azure/devops/released/test_plan/test_plan_client.py +++ b/azure-devops/azure/devops/released/test_plan/test_plan_client.py @@ -334,7 +334,7 @@ def get_test_suite_by_id(self, project, plan_id, suite_id, expand=None): query_parameters=query_parameters) return self._deserialize('TestSuite', response) - def get_test_suites_for_plan(self, project, plan_id, expand=None, continuation_token=None, as_tree_view=None): + def get_test_suites_for_plan(self, project, plan_id, expand=None, continuation_token=None, as_tree_view=None, extract_all=False): """GetTestSuitesForPlan. Get test suites for plan. :param str project: Project ID or project name @@ -361,6 +361,9 @@ def get_test_suites_for_plan(self, project, plan_id, expand=None, continuation_t version='7.0', route_values=route_values, query_parameters=query_parameters) + if extract_all and 'x-ms-continuationtoken' in response.headers: + continuation_token = response.headers['x-ms-continuationtoken'] + return self._deserialize('[TestPlan]', self._unwrap_collection(response)) + self.get_test_plans(project, plan_id, expand=expand, continuation_token=continuation_token, as_tree_view=as_tree_view, extract_all=True) return self._deserialize('[TestSuite]', self._unwrap_collection(response)) def update_test_suite(self, test_suite_update_params, project, plan_id, suite_id):