@huggingface/hub/src/lib/create-collection.spec.ts

Utilities to interact with the Hugging Face hub

import { it, describe, expect } from "vitest";

import { TEST_HUB_URL, TEST_ACCESS_TOKEN, TEST_USER } from "../test/consts";
import { createCollection } from "./create-collection";
import { deleteCollection } from "./delete-collection";

describe("createCollection", () => {
	it("should create a collection", async () => {
		let slug: string = "";
		const randomString = crypto.randomUUID();
		const title = `Test Collection ${randomString}`;

		try {
			const result = await createCollection({
				collection: {
					title,
					namespace: TEST_USER,
					description: "This is a test collection",
					private: false,
				},
				accessToken: TEST_ACCESS_TOKEN,
				hubUrl: TEST_HUB_URL,
			});

			expect(result.slug.startsWith(`${TEST_USER}/test-collection-${randomString}`)).toBe(true);

			slug = result.slug;
		} finally {
			if (slug) {
				await deleteCollection({
					slug,
					accessToken: TEST_ACCESS_TOKEN,
					hubUrl: TEST_HUB_URL,
				});
			}
		}
	});
});