From c58f5149d9412012d3d50a9b6499322f9b60d37f Mon Sep 17 00:00:00 2001 From: Ryan Timmons Date: Wed, 11 May 2022 17:56:21 -0400 Subject: [PATCH] Add use_presto parameter to test_stats_by_project --- CHANGELOG.md | 4 +++- pyproject.toml | 2 +- src/evergreen/api.py | 4 ++++ tests/evergreen/test_api.py | 4 +++- 4 files changed, 11 insertions(+), 3 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 19b4db88..df2d2f7e 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,7 +1,9 @@ # Changelog -## 3.4.5 - 2022-05-11 +## 3.4.6 - 2022-05-11 +- Add `use_presto` parameter to the `test_stats_by_project` operation. +## 3.4.5 - 2022-05-11 - Add `project_identifier` field to `Version` objects. ## 3.4.4 - 2022-03-31 diff --git a/pyproject.toml b/pyproject.toml index 7141635d..f34a2d08 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "evergreen.py" -version = "3.4.5" +version = "3.4.6" description = "Python client for the Evergreen API" authors = [ "Dev Prod DAG ", diff --git a/src/evergreen/api.py b/src/evergreen/api.py index 79d49dde..435e610e 100644 --- a/src/evergreen/api.py +++ b/src/evergreen/api.py @@ -565,6 +565,7 @@ def test_stats_by_project( distros: Optional[List[str]] = None, group_by: Optional[str] = None, sort: Optional[str] = None, + use_presto: Optional[bool] = None, ) -> List[TestStats]: """ Get a patch by patch id. @@ -580,6 +581,7 @@ def test_stats_by_project( :param distros: Only include specified distros. :param group_by: How to group results (test_task_variant, test_task, or test) :param sort: How to sort results (earliest or latest). + :param use_presto: Use the presto data backend. :return: Patch queried for. """ params: Dict[str, Any] = { @@ -602,6 +604,8 @@ def test_stats_by_project( params["group_by"] = group_by if sort is not None: params["sort"] = sort + if use_presto is not None: + params["use_presto"] = use_presto url = self._create_url(f"/projects/{project_id}/test_stats") test_stats_list = self._paginate(url, params) return [TestStats(test_stat, self) for test_stat in test_stats_list] # type: ignore[arg-type] diff --git a/tests/evergreen/test_api.py b/tests/evergreen/test_api.py index 04739b02..8bec5693 100644 --- a/tests/evergreen/test_api.py +++ b/tests/evergreen/test_api.py @@ -343,7 +343,7 @@ def test_test_stats_by_project(self, mocked_api): url=expected_url, params=expected_params, timeout=None, data=None, method="GET" ) - def test_test_stats_by_project_with_requester(self, mocked_api): + def test_test_stats_by_project_with_requester_and_presto(self, mocked_api): after_date = "2019-01-01" before_date = "2019-02-01" expected_url = mocked_api._create_url("/projects/project_id/test_stats") @@ -351,6 +351,7 @@ def test_test_stats_by_project_with_requester(self, mocked_api): "after_date": after_date, "before_date": before_date, "requesters": "patch", + "use_presto": True, } mocked_api.test_stats_by_project( @@ -358,6 +359,7 @@ def test_test_stats_by_project_with_requester(self, mocked_api): from_iso_format(after_date), from_iso_format(before_date), requesters=Requester.PATCH_REQUEST, + use_presto=True, ) mocked_api.session.request.assert_called_with(