feat(gql): add testSuites, flags to TestAnalytics

Author: joseph-sentry

graphql_api/tests/test_test_analytics.py

@@ -1134,3 +1134,59 @@ def test_flake_aggregates_7_days(self) -> None:
             "flakeCountPercentChange": -50.0,
             "flakeRatePercentChange": -43.75,
         }
+
+    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")
+
+        repo_flag = RepositoryFlagFactory(repository=repo, flag_name="hello_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,
+        )
+        res = self.fetch_test_analytics(
+            repo.name,
+            """testSuites""",
+        )
+        assert res["testSuites"] == ["test_suite_1", "test_suite_2"]
+
+    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")
+
+        _ = 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,
+        )
+        res = self.fetch_test_analytics(
+            repo.name,
+            """flags""",
+        )
+        assert res["flags"] == ["hello_world"]

graphql_api/types/test_analytics/test_analytics.graphql

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

graphql_api/types/test_analytics/test_analytics.py

@@ -13,6 +13,8 @@
     generate_flake_aggregates,
     generate_test_results,
     generate_test_results_aggregates,
+    get_flags,
+    get_test_suites,
 )
 
 log = logging.getLogger(__name__)
@@ -88,6 +90,16 @@ 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)
+
+
+@test_analytics_bindable.field("flags")
+async def resolve_flags(repository: Repository, info: GraphQLResolveInfo, **_):
+    return await sync_to_async(get_flags)(repository.repoid)
+
+
 def convert_interval_to_timedelta(interval: MeasurementInterval | None) -> timedelta:
     if interval is None:
         return timedelta(days=30)

utils/test_results.py

@@ -16,6 +16,7 @@
 from shared.django_apps.reports.models import (
     DailyTestRollup,
     Flake,
+    Test,
     TestFlagBridge,
 )
 
@@ -623,3 +624,20 @@ def generate_flake_aggregates(
         curr_numbers,
         past_numbers,
     )
+
+
+def get_test_suites(repoid: int) -> list[str]:
+    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()
+    )