fix: add search term

Author: joseph-sentry

graphql_api/tests/test_test_analytics.py

@@ -996,8 +996,8 @@ def test_flake_aggregates_7_days(self) -> None:
 
     def test_test_suites(self) -> None:
         repo = RepositoryFactory(author=self.owner, active=True, private=True)
-        test = TestFactory(repository=repo, testsuite="test_suite_1")
-        test2 = TestFactory(repository=repo, testsuite="test_suite_2")
+        test = TestFactory(repository=repo, testsuite="hello_world")
+        test2 = TestFactory(repository=repo, testsuite="goodbye_world")
 
         repo_flag = RepositoryFlagFactory(repository=repo, flag_name="hello_world")
 
@@ -1018,34 +1018,23 @@ def test_test_suites(self) -> None:
         )
         res = self.fetch_test_analytics(
             repo.name,
-            """testSuites""",
+            """testSuites(term: "hello")""",
         )
-        assert res["testSuites"] == ["test_suite_1", "test_suite_2"]
+        assert res["testSuites"] == ["hello_world"]
 
     def test_flags(self) -> None:
         repo = RepositoryFactory(author=self.owner, active=True, private=True)
         test = TestFactory(repository=repo)
         test2 = TestFactory(repository=repo)
 
         repo_flag = RepositoryFlagFactory(repository=repo, flag_name="hello_world")
+        repo_flag2 = RepositoryFlagFactory(repository=repo, flag_name="goodbye_world")
 
         _ = TestFlagBridgeFactory(flag=repo_flag, test=test)
-        _ = DailyTestRollupFactory(
-            test=test,
-            created_at=datetime.datetime.now(),
-            repoid=repo.repoid,
-            branch="main",
-            avg_duration_seconds=0.1,
-        )
-        _ = DailyTestRollupFactory(
-            test=test2,
-            created_at=datetime.datetime.now(),
-            repoid=repo.repoid,
-            branch="main",
-            avg_duration_seconds=20.0,
-        )
+        _ = TestFlagBridgeFactory(flag=repo_flag2, test=test2)
+
         res = self.fetch_test_analytics(
             repo.name,
-            """flags""",
+            """flags(term: "hello")""",
         )
         assert res["flags"] == ["hello_world"]

graphql_api/types/test_analytics/test_analytics.graphql

@@ -18,10 +18,10 @@ type TestAnalytics {
   "Flake aggregates are flake totals across all tests"
   flakeAggregates(interval: MeasurementInterval): FlakeAggregates
 
-  testSuites: [String!]!
+  testSuites(term: String): [String!]!
 
   "Only flag names relevant to Test Analytics"
-  flags: [String!]!
+  flags(term: String): [String!]!
 }
 
 type TestResultConnection {

graphql_api/types/test_analytics/test_analytics.py

@@ -95,13 +95,17 @@ async def resolve_flake_aggregates(
 
 
 @test_analytics_bindable.field("testSuites")
-async def resolve_test_suites(repository: Repository, info: GraphQLResolveInfo, **_):
-    return await sync_to_async(get_test_suites)(repository.repoid)
+async def resolve_test_suites(
+    repository: Repository, info: GraphQLResolveInfo, term: str | None = None, **_
+):
+    return await sync_to_async(get_test_suites)(repository.repoid, term)
 
 
 @test_analytics_bindable.field("flags")
-async def resolve_flags(repository: Repository, info: GraphQLResolveInfo, **_):
-    return await sync_to_async(get_flags)(repository.repoid)
+async def resolve_flags(
+    repository: Repository, info: GraphQLResolveInfo, term: str | None = None, **_
+):
+    return await sync_to_async(get_flags)(repository.repoid, term)
 
 
 def convert_interval_to_timedelta(interval: MeasurementInterval | None) -> timedelta:

utils/test_results.py

@@ -563,18 +563,35 @@ def generate_flake_aggregates(
     )
 
 
-def get_test_suites(repoid: int) -> list[str]:
-    return list(
-        Test.objects.filter(repository_id=repoid)
-        .values_list("testsuite", flat=True)
-        .distinct()
-    )
+def get_test_suites(repoid: int, term: str | None = None) -> list[str]:
+    if term:
+        return list(
+            Test.objects.filter(repository_id=repoid, testsuite__icontains=term)
+            .values_list("testsuite", flat=True)
+            .distinct()
+        )
+    else:
+        return list(
+            Test.objects.filter(repository_id=repoid)
+            .values_list("testsuite", flat=True)
+            .distinct()
+        )
 
 
-def get_flags(repoid: int) -> list[str]:
-    return list(
-        TestFlagBridge.objects.filter(test__repository_id=repoid)
-        .select_related("flag")
-        .values_list("flag__flag_name", flat=True)
-        .distinct()
-    )
+def get_flags(repoid: int, term: str | None = None) -> list[str]:
+    if term:
+        return list(
+            TestFlagBridge.objects.filter(
+                test__repository_id=repoid, flag__flag_name__icontains=term
+            )
+            .select_related("flag")
+            .values_list("flag__flag_name", flat=True)
+            .distinct()
+        )
+    else:
+        return list(
+            TestFlagBridge.objects.filter(test__repository_id=repoid)
+            .select_related("flag")
+            .values_list("flag__flag_name", flat=True)
+            .distinct()
+        )