setup-uv/__tests__/download/versions-client.test.ts

import { beforeEach, describe, expect, it, jest } from "@jest/globals";

// biome-ignore lint/suspicious/noExplicitAny: Mock requires flexible typing in tests.
const mockFetch = jest.fn<any>();

jest.unstable_mockModule("../../src/utils/fetch", () => ({
  fetch: mockFetch,
}));

const {
  clearCache,
  fetchVersionData,
  getAllVersions,
  getArtifact,
  getHighestSatisfyingVersion,
  getLatestVersion,
  parseVersionData,
} = await import("../../src/download/versions-client");

const sampleNdjsonResponse = `{"version":"0.9.26","artifacts":[{"platform":"aarch64-apple-darwin","variant":"default","url":"https://github.com/astral-sh/uv/releases/download/0.9.26/uv-aarch64-apple-darwin.tar.gz","archive_format":"tar.gz","sha256":"fcf0a9ea6599c6ae28a4c854ac6da76f2c889354d7c36ce136ef071f7ab9721f"},{"platform":"x86_64-pc-windows-msvc","variant":"default","url":"https://github.com/astral-sh/uv/releases/download/0.9.26/uv-x86_64-pc-windows-msvc.zip","archive_format":"zip","sha256":"eb02fd95d8e0eed462b4a67ecdd320d865b38c560bffcda9a0b87ec944bdf036"}]}
{"version":"0.9.25","artifacts":[{"platform":"aarch64-apple-darwin","variant":"default","url":"https://github.com/astral-sh/uv/releases/download/0.9.25/uv-aarch64-apple-darwin.tar.gz","archive_format":"tar.gz","sha256":"606b3c6949d971709f2526fa0d9f0fd23ccf60e09f117999b406b424af18a6a6"}]}`;

const multiVariantNdjsonResponse = `{"version":"0.9.26","artifacts":[{"platform":"aarch64-apple-darwin","variant":"python-managed","url":"https://github.com/astral-sh/uv/releases/download/0.9.26/uv-aarch64-apple-darwin-managed.tar.gz","archive_format":"tar.gz","sha256":"managed-checksum"},{"platform":"aarch64-apple-darwin","variant":"default","url":"https://github.com/astral-sh/uv/releases/download/0.9.26/uv-aarch64-apple-darwin.zip","archive_format":"zip","sha256":"default-checksum"}]}`;

function createMockStream(chunks: string[]): ReadableStream<Uint8Array> {
  const encoder = new TextEncoder();

  return new ReadableStream<Uint8Array>({
    start(controller) {
      for (const chunk of chunks) {
        controller.enqueue(encoder.encode(chunk));
      }
      controller.close();
    },
  });
}

function createMockResponse(
  ok: boolean,
  status: number,
  statusText: string,
  data: string,
  chunks: string[] = [data],
) {
  return {
    body: createMockStream(chunks),
    ok,
    status,
    statusText,
    text: async () => data,
  };
}

describe("versions-client", () => {
  beforeEach(() => {
    clearCache();
    mockFetch.mockReset();
  });

  describe("fetchVersionData", () => {
    it("should fetch and parse NDJSON data", async () => {
      mockFetch.mockResolvedValue(
        createMockResponse(true, 200, "OK", sampleNdjsonResponse),
      );

      const versions = await fetchVersionData();

      expect(versions).toHaveLength(2);
      expect(versions[0].version).toBe("0.9.26");
      expect(versions[1].version).toBe("0.9.25");
    });

    it("should throw error on failed fetch", async () => {
      mockFetch.mockResolvedValue(
        createMockResponse(false, 500, "Internal Server Error", ""),
      );

      await expect(fetchVersionData()).rejects.toThrow(
        "Failed to fetch version data: 500 Internal Server Error",
      );
    });

    it("should cache results", async () => {
      mockFetch.mockResolvedValue(
        createMockResponse(true, 200, "OK", sampleNdjsonResponse),
      );

      await fetchVersionData();
      await fetchVersionData();

      expect(mockFetch).toHaveBeenCalledTimes(1);
    });
  });

  describe("getLatestVersion", () => {
    it("should return the first version (newest)", async () => {
      mockFetch.mockResolvedValue(
        createMockResponse(true, 200, "OK", sampleNdjsonResponse),
      );

      const latest = await getLatestVersion();

      expect(latest).toBe("0.9.26");
    });

    it("should stop after the first record when resolving latest", async () => {
      mockFetch.mockResolvedValue(
        createMockResponse(
          true,
          200,
          "OK",
          `${sampleNdjsonResponse}\n{"version":`,
          [`${sampleNdjsonResponse.split("\n")[0]}\n`, '{"version":'],
        ),
      );

      const latest = await getLatestVersion();

      expect(latest).toBe("0.9.26");
    });
  });

  describe("getAllVersions", () => {
    it("should return all version strings", async () => {
      mockFetch.mockResolvedValue(
        createMockResponse(true, 200, "OK", sampleNdjsonResponse),
      );

      const versions = await getAllVersions();

      expect(versions).toEqual(["0.9.26", "0.9.25"]);
    });
  });

  describe("getHighestSatisfyingVersion", () => {
    it("should return the first matching version from the stream", async () => {
      mockFetch.mockResolvedValue(
        createMockResponse(
          true,
          200,
          "OK",
          `${sampleNdjsonResponse}\n{"version":`,
          [`${sampleNdjsonResponse.split("\n")[0]}\n`, '{"version":'],
        ),
      );

      const version = await getHighestSatisfyingVersion("^0.9.0");

      expect(version).toBe("0.9.26");
    });
  });

  describe("getArtifact", () => {
    beforeEach(() => {
      mockFetch.mockResolvedValue(
        createMockResponse(true, 200, "OK", sampleNdjsonResponse),
      );
    });

    it("should find artifact by version and platform", async () => {
      const artifact = await getArtifact("0.9.26", "aarch64", "apple-darwin");

      expect(artifact).toEqual({
        archiveFormat: "tar.gz",
        sha256:
          "fcf0a9ea6599c6ae28a4c854ac6da76f2c889354d7c36ce136ef071f7ab9721f",
        url: "https://github.com/astral-sh/uv/releases/download/0.9.26/uv-aarch64-apple-darwin.tar.gz",
      });
    });

    it("should stop once the requested version is found", async () => {
      mockFetch.mockResolvedValue(
        createMockResponse(
          true,
          200,
          "OK",
          `${sampleNdjsonResponse}\n{"version":`,
          [`${sampleNdjsonResponse.split("\n")[0]}\n`, '{"version":'],
        ),
      );

      const artifact = await getArtifact("0.9.26", "aarch64", "apple-darwin");

      expect(artifact).toEqual({
        archiveFormat: "tar.gz",
        sha256:
          "fcf0a9ea6599c6ae28a4c854ac6da76f2c889354d7c36ce136ef071f7ab9721f",
        url: "https://github.com/astral-sh/uv/releases/download/0.9.26/uv-aarch64-apple-darwin.tar.gz",
      });
    });

    it("should find windows artifact", async () => {
      const artifact = await getArtifact("0.9.26", "x86_64", "pc-windows-msvc");

      expect(artifact).toEqual({
        archiveFormat: "zip",
        sha256:
          "eb02fd95d8e0eed462b4a67ecdd320d865b38c560bffcda9a0b87ec944bdf036",
        url: "https://github.com/astral-sh/uv/releases/download/0.9.26/uv-x86_64-pc-windows-msvc.zip",
      });
    });

    it("should prefer the default variant when multiple artifacts share a platform", async () => {
      mockFetch.mockResolvedValue(
        createMockResponse(true, 200, "OK", multiVariantNdjsonResponse),
      );

      const artifact = await getArtifact("0.9.26", "aarch64", "apple-darwin");

      expect(artifact).toEqual({
        archiveFormat: "zip",
        sha256: "default-checksum",
        url: "https://github.com/astral-sh/uv/releases/download/0.9.26/uv-aarch64-apple-darwin.zip",
      });
    });

    it("should return undefined for unknown version", async () => {
      const artifact = await getArtifact("0.0.1", "aarch64", "apple-darwin");

      expect(artifact).toBeUndefined();
    });

    it("should return undefined for unknown platform", async () => {
      const artifact = await getArtifact(
        "0.9.26",
        "aarch64",
        "unknown-linux-musl",
      );

      expect(artifact).toBeUndefined();
    });
  });

  describe("parseVersionData", () => {
    it("should throw for malformed NDJSON", () => {
      expect(() =>
        parseVersionData('{"version":"0.1.0"', "test-source"),
      ).toThrow("Failed to parse version data from test-source");
    });
  });
});