1 files changed, 134 insertions, 0 deletions

diff --git a/packages/create-turbo/__tests__/examples.test.ts b/packages/create-turbo/__tests__/examples.test.ts
new file mode 100644
index 0000000..20d4464
--- /dev/null
+++ b/packages/create-turbo/__tests__/examples.test.ts
@@ -0,0 +1,134 @@
+import got from "got";
+import * as Got from "got";
+import { isUrlOk, getRepoInfo, hasRepo } from "../src/utils/examples";
+
+jest.mock("got", () => ({
+  __esModule: true,
+  ...jest.requireActual("got"),
+}));
+
+describe("examples", () => {
+  describe("isUrlOk", () => {
+    it("returns true if url returns 200", async () => {
+      const mockGot = jest
+        .spyOn(got, "head")
+        .mockReturnValue({ statusCode: 200 } as any);
+
+      const url = "https://github.com/vercel/turbo/";
+      const result = await isUrlOk(url);
+      expect(result).toBe(true);
+
+      expect(mockGot).toHaveBeenCalledWith(url);
+      mockGot.mockRestore();
+    });
+
+    it("returns false if url returns status != 200", async () => {
+      const mockGot = jest
+        .spyOn(got, "head")
+        .mockReturnValue({ statusCode: 401 } as any);
+
+      const url = "https://not-github.com/vercel/turbo/";
+      const result = await isUrlOk(url);
+      expect(result).toBe(false);
+
+      expect(mockGot).toHaveBeenCalledWith(url);
+      mockGot.mockRestore();
+    });
+  });
+
+  describe("getRepoInfo", () => {
+    test.each([
+      {
+        repoUrl: "https://github.com/vercel/turbo/",
+        examplePath: undefined,
+        defaultBranch: "main",
+        expectBranchLookup: true,
+        expected: {
+          username: "vercel",
+          name: "turbo",
+          branch: "main",
+          filePath: "",
+        },
+      },
+      {
+        repoUrl:
+          "https://github.com/vercel/turbo/tree/canary/examples/kitchen-sink",
+        examplePath: undefined,
+        defaultBranch: "canary",
+        expectBranchLookup: false,
+        expected: {
+          username: "vercel",
+          name: "turbo",
+          branch: "canary",
+          filePath: "examples/kitchen-sink",
+        },
+      },
+      {
+        repoUrl: "https://github.com/vercel/turbo/tree/tek/test-branch/",
+        examplePath: "examples/basic",
+        defaultBranch: "canary",
+        expectBranchLookup: false,
+        expected: {
+          username: "vercel",
+          name: "turbo",
+          branch: "tek/test-branch",
+          filePath: "examples/basic",
+        },
+      },
+    ])(
+      "retrieves repo info for $repoUrl and $examplePath",
+      async ({
+        repoUrl,
+        examplePath,
+        defaultBranch,
+        expectBranchLookup,
+        expected,
+      }) => {
+        const mockGot = jest.spyOn(Got, "default").mockReturnValue({
+          body: JSON.stringify({ default_branch: defaultBranch }),
+        } as any);
+
+        const url = new URL(repoUrl);
+        const result = await getRepoInfo(url, examplePath);
+        expect(result).toMatchObject(expected);
+
+        if (result && expectBranchLookup) {
+          expect(mockGot).toHaveBeenCalledWith(
+            `https://api.github.com/repos/${result.username}/${result.name}`
+          );
+        }
+
+        mockGot.mockRestore();
+      }
+    );
+  });
+
+  describe("hasRepo", () => {
+    test.each([
+      {
+        repoInfo: {
+          username: "vercel",
+          name: "turbo",
+          branch: "main",
+          filePath: "",
+        },
+        expected: true,
+        expectedUrl:
+          "https://api.github.com/repos/vercel/turbo/contents/package.json?ref=main",
+      },
+    ])(
+      "checks repo at $expectedUrl",
+      async ({ expected, repoInfo, expectedUrl }) => {
+        const mockGot = jest
+          .spyOn(got, "head")
+          .mockReturnValue({ statusCode: 200 } as any);
+
+        const result = await hasRepo(repoInfo);
+        expect(result).toBe(expected);
+
+        expect(mockGot).toHaveBeenCalledWith(expectedUrl);
+        mockGot.mockRestore();
+      }
+    );
+  });
+});