feat: add totalFlakyFailCount to testResult (#887)

Author: joseph-sentry

graphql_api/tests/test_test_result.py

@@ -31,13 +31,15 @@ def setUp(self):
             date=date.today() - timedelta(days=2),
             avg_duration_seconds=0.6,
             latest_run=datetime.now() - timedelta(days=2),
+            flaky_fail_count=0,
         )
         _ = DailyTestRollupFactory(
             test=self.test,
             commits_where_fail=["123", "456"],
             date=datetime.now() - timedelta(days=1),
             avg_duration_seconds=2,
             latest_run=datetime.now() - timedelta(days=1),
+            flaky_fail_count=1,
         )
         _ = DailyTestRollupFactory(
             test=self.test,
@@ -46,6 +48,7 @@ def setUp(self):
             last_duration_seconds=5.0,
             avg_duration_seconds=3,
             latest_run=datetime.now(),
+            flaky_fail_count=1,
         )
 
     def test_fetch_test_result_name(self) -> None:
@@ -354,3 +357,34 @@ def test_fetch_test_result_total_pass_count(self) -> None:
             ]["totalPassCount"]
             == 3
         )
+
+    def test_fetch_test_result_total_flaky_fail_count(self) -> None:
+        query = """
+            query {
+               owner(username: "%s") {
+                    repository(name: "%s") {
+                        ... on Repository {
+                            testAnalytics {
+                                testResults {
+                                    edges {
+                                        node {
+                                            totalFlakyFailCount
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                 }
+            }
+        """ % (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"
+            ]["totalFlakyFailCount"]
+            == 2
+        )

graphql_api/types/test_results/test_results.graphql

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

graphql_api/types/test_results/test_results.py

@@ -14,6 +14,7 @@ class TestDict(TypedDict):
     last_duration: float
     flake_rate: float
     total_fail_count: int
+    total_flaky_fail_count: int
     total_skip_count: int
     total_pass_count: int
     computed_name: str | None
@@ -62,6 +63,11 @@ def resolve_total_fail_count(test: TestDict, _: GraphQLResolveInfo) -> int:
     return test["total_fail_count"]
 
 
+@test_result_bindable.field("totalFlakyFailCount")
+def resolve_total_flaky_fail_count(test: TestDict, _: GraphQLResolveInfo) -> int:
+    return test["total_flaky_fail_count"]
+
+
 @test_result_bindable.field("totalSkipCount")
 def resolve_total_skip_count(test: TestDict, _: GraphQLResolveInfo) -> int:
     return test["total_skip_count"]