make fetchLinkableGitRepositories get all linkable git repositories

firebase · Mar 19, 2024 · 3da72f5 · 3da72f5
1 parent c76e863
commit 3da72f5
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 10 deletions.
diff --git a/src/gcp/devConnect.ts b/src/gcp/devConnect.ts
@@ -185,18 +185,29 @@ export async function fetchLinkableGitRepositories(
   projectId: string,
   location: string,
   connectionId: string,
-  pageToken = "",
-  pageSize = 1000,
-): Promise<LinkableGitRepositories> {
+): Promise<LinkableGitRepository[]> {
   const name = `projects/${projectId}/locations/${location}/connections/${connectionId}:fetchLinkableRepositories`;
-  const res = await client.get<LinkableGitRepositories>(name, {
-    queryParams: {
-      pageSize,
-      pageToken,
-    },
-  });
+  const repos: LinkableGitRepository[] = [];
 
-  return res.body;
+  const getNextPage = async (pageToken = ""): Promise<void> => {
+    const res = await client.get<LinkableGitRepositories>(name, {
+      queryParams: {
+        PAGE_SIZE_MAX,
+        pageToken,
+      },
+    });
+
+    if (Array.isArray(res.body.repositories)) {
+      repos.push(...res.body.repositories);
+    }
+
+    if (res.body.nextPageToken) {
+      await getNextPage(res.body.nextPageToken);
+    }
+  };
+
+  await getNextPage();
+  return repos;
 }
 
 /**

diff --git a/src/test/gcp/devconnect.spec.ts b/src/test/gcp/devconnect.spec.ts
@@ -67,4 +67,41 @@ describe("developer connect", () => {
       expect(conns).to.deep.equal([firstConnection, secondConnection, thirdConnection]);
     });
   });
+  describe("fetchLinkableGitRepositories", () => {
+    it("interates through all pages and returns a single list", async () => {
+      const firstRepo = { cloneUri: "repo1" };
+      const secondRepo = { cloneUri: "repo2" };
+      const thirdRepo = { cloneUri: "repo3" };
+
+      get
+        .onFirstCall()
+        .returns({
+          body: {
+            repositories: [firstRepo],
+            nextPageToken: "someToken",
+          },
+        })
+        .onSecondCall()
+        .returns({
+          body: {
+            repositories: [secondRepo],
+            nextPageToken: "someToken2",
+          },
+        })
+        .onThirdCall()
+        .returns({
+          body: {
+            repositories: [thirdRepo],
+          },
+        });
+
+      const conns = await devconnect.fetchLinkableGitRepositories(
+        projectId,
+        location,
+        connectionId,
+      );
+      expect(get).callCount(3);
+      expect(conns).to.deep.equal([firstRepo, secondRepo, thirdRepo]);
+    });
+  });
 });