fix tests

codecov · Oct 3, 2024 · 6fe5eef · 6fe5eef
1 parent b922568
commit 6fe5eef
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 16 deletions.
diff --git a/src/services/repo/useRepoBackfilled.tsx b/src/services/repo/useRepoBackfilled.tsx
@@ -39,9 +39,13 @@ query BackfillFlagMemberships($name: String!, $repo: String!) {
 
 const RepositorySchema = z.object({
   __typename: z.literal('Repository'),
-  flagsMeasurementsActive: z.boolean().nullish(),
-  flagsMeasurementsBackfilled: z.boolean().nullish(),
-  flagsCount: z.number().nullish(),
+  coverageAnalytics: z
+    .object({
+      flagsMeasurementsActive: z.boolean().nullish(),
+      flagsMeasurementsBackfilled: z.boolean().nullish(),
+      flagsCount: z.number().nullish(),
+    })
+    .nullable(),
 })
 
 const RepoBackfilledSchema = z.object({

diff --git a/src/services/repo/useRepoFlagsSelect.tsx b/src/services/repo/useRepoFlagsSelect.tsx
@@ -34,19 +34,23 @@ interface FetchRepoFlagsArgs {
 
 const RepositorySchema = z.object({
   __typename: z.literal('Repository'),
-  flags: z.object({
-    pageInfo: z.object({
-      hasNextPage: z.boolean(),
-      endCursor: z.string().nullable(),
-    }),
-    edges: z.array(
-      z.object({
-        node: z.object({
-          name: z.string(),
+  coverageAnalytics: z
+    .object({
+      flags: z.object({
+        pageInfo: z.object({
+          hasNextPage: z.boolean(),
+          endCursor: z.string().nullable(),
         }),
-      })
-    ),
-  }),
+        edges: z.array(
+          z.object({
+            node: z.object({
+              name: z.string(),
+            }),
+          })
+        ),
+      }),
+    })
+    .nullable(),
 })
 
 const FetchRepoFlagsSchema = z.object({
@@ -157,7 +161,7 @@ function fetchRepoFlags({
       } satisfies NetworkErrorObject)
     }
 
-    const flags = data?.owner?.repository?.flags
+    const flags = data?.owner?.repository?.coverageAnalytics?.flags
     const pageInfo = flags?.pageInfo ?? null
 
     return {