From a91dd7534d08dac84fa03ee154f129a427083c7c Mon Sep 17 00:00:00 2001
From: Rula Abu Hasna <rula@codecov.io>
Date: Tue, 23 Jul 2024 12:51:51 +0200
Subject: [PATCH] feat: Add testResultsCount to useTestResults

---
 .../FailedTestsTab/hooks/useInfiniteTestResults.spec.tsx        | 1 +
 .../RepoPage/FailedTestsTab/hooks/useInfiniteTestResults.tsx    | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/src/pages/RepoPage/FailedTestsTab/hooks/useInfiniteTestResults.spec.tsx b/src/pages/RepoPage/FailedTestsTab/hooks/useInfiniteTestResults.spec.tsx
index 5589d77531..a0b79f2017 100644
--- a/src/pages/RepoPage/FailedTestsTab/hooks/useInfiniteTestResults.spec.tsx
+++ b/src/pages/RepoPage/FailedTestsTab/hooks/useInfiniteTestResults.spec.tsx
@@ -35,6 +35,7 @@ const mockTestResults = {
           hasNextPage: true,
         },
       },
+      testResultsCount: 2,
     },
   },
 }
diff --git a/src/pages/RepoPage/FailedTestsTab/hooks/useInfiniteTestResults.tsx b/src/pages/RepoPage/FailedTestsTab/hooks/useInfiniteTestResults.tsx
index 94c0cf294d..e2cb9ccba9 100644
--- a/src/pages/RepoPage/FailedTestsTab/hooks/useInfiniteTestResults.tsx
+++ b/src/pages/RepoPage/FailedTestsTab/hooks/useInfiniteTestResults.tsx
@@ -40,6 +40,7 @@ const GetTestResultsSchema = z.object({
               hasNextPage: z.boolean(),
             }),
           }),
+          testResultsCount: z.number(),
         }),
         RepoNotFoundErrorSchema,
         RepoOwnerNotActivatedErrorSchema,
@@ -83,6 +84,7 @@ query GetTestResults(
             hasNextPage
           }
         }
+        testResultsCount
       }
       ... on NotFoundError {
         message