feat: add total_{fail, pass, skip}_count field to TestResult GQL model

Author: joseph-sentry

graphql_api/tests/test_test_result.py

@@ -227,3 +227,96 @@ def test_fetch_test_result_avg_duration(self) -> None:
         assert result["owner"]["repository"]["testAnalytics"]["testResults"]["edges"][
             0
         ]["node"]["avgDuration"] == (5.6 / 3)
+
+    def test_fetch_test_result_total_fail_count(self) -> None:
+        query = """
+            query {
+               owner(username: "%s") {
+                    repository(name: "%s") {
+                        ... on Repository {
+                            testAnalytics {
+                                testResults {
+                                    edges {
+                                        node {
+                                            totalFailCount
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                 }
+            }
+        """ % (self.owner.username, self.repository.name)
+
+        result = self.gql_request(query, owner=self.owner)
+
+        assert "errors" not in result
+        assert (
+            result["owner"]["repository"]["testAnalytics"]["testResults"]["edges"][0][
+                "node"
+            ]["totalFailCount"]
+            == 9
+        )
+
+    def test_fetch_test_result_total_skip_count(self) -> None:
+        query = """
+            query {
+               owner(username: "%s") {
+                    repository(name: "%s") {
+                        ... on Repository {
+                            testAnalytics {
+                                testResults {
+                                    edges {
+                                        node {
+                                            totalSkipCount
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                 }
+            }
+        """ % (self.owner.username, self.repository.name)
+
+        result = self.gql_request(query, owner=self.owner)
+
+        assert "errors" not in result
+        assert (
+            result["owner"]["repository"]["testAnalytics"]["testResults"]["edges"][0][
+                "node"
+            ]["totalSkipCount"]
+            == 6
+        )
+
+    def test_fetch_test_result_total_pass_count(self) -> None:
+        query = """
+            query {
+               owner(username: "%s") {
+                    repository(name: "%s") {
+                        ... on Repository {
+                            testAnalytics {
+                                testResults {
+                                    edges {
+                                        node {
+                                            totalPassCount
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                 }
+            }
+        """ % (self.owner.username, self.repository.name)
+
+        result = self.gql_request(query, owner=self.owner)
+
+        assert "errors" not in result
+        assert (
+            result["owner"]["repository"]["testAnalytics"]["testResults"]["edges"][0][
+                "node"
+            ]["totalPassCount"]
+            == 3
+        )

graphql_api/types/test_results/test_results.graphql

@@ -6,4 +6,7 @@ type TestResult {
   flakeRate: Float!
   avgDuration: Float!
   lastDuration: Float!
+  totalFailCount: Int!
+  totalSkipCount: Int!
+  totalPassCount: Int!
 }

graphql_api/types/test_results/test_results.py

@@ -13,6 +13,9 @@ class TestDict(TypedDict):
     avg_duration: float
     last_duration: float
     flake_rate: float
+    total_fail_count: int
+    total_skip_count: int
+    total_pass_count: int
 
 
 test_result_bindable = ObjectType("TestResult")
@@ -51,3 +54,18 @@ def resolve_avg_duration(test: TestDict, _: GraphQLResolveInfo) -> float:
 @test_result_bindable.field("lastDuration")
 def resolve_last_duration(test: TestDict, _: GraphQLResolveInfo) -> float:
     return test["last_duration"]
+
+
+@test_result_bindable.field("totalFailCount")
+def resolve_total_fail_count(test: TestDict, _: GraphQLResolveInfo) -> int:
+    return test["total_fail_count"]
+
+
+@test_result_bindable.field("totalSkipCount")
+def resolve_total_skip_count(test: TestDict, _: GraphQLResolveInfo) -> int:
+    return test["total_skip_count"]
+
+
+@test_result_bindable.field("totalPassCount")
+def resolve_total_pass_count(test: TestDict, _: GraphQLResolveInfo) -> int:
+    return test["total_pass_count"]

utils/test_results.py

@@ -172,6 +172,8 @@ def generate_test_results(
         total_flaky_fail_count=Cast(
             Sum(F("flaky_fail_count")), output_field=FloatField()
         ),
+        total_skip_count=Cast(Sum(F("skip_count")), output_field=FloatField()),
+        total_pass_count=Cast(Sum(F("pass_count")), output_field=FloatField()),
         failure_rate=Case(
             When(
                 total_test_count=0,