File size: 2,802 Bytes
21dd449 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 |
import { assert, it, describe, expect } from "vitest";
import { TEST_HUB_URL, TEST_ACCESS_TOKEN, TEST_USER } from "../test/consts";
import { insecureRandomString } from "../utils/insecureRandomString";
import { createRepo } from "./create-repo";
import { deleteRepo } from "./delete-repo";
import { downloadFile } from "./download-file";
describe("createRepo", () => {
it("should create a repo", async () => {
const repoName = `${TEST_USER}/TEST-${insecureRandomString()}`;
const result = await createRepo({
accessToken: TEST_ACCESS_TOKEN,
repo: {
name: repoName,
type: "model",
},
hubUrl: TEST_HUB_URL,
files: [{ path: ".gitattributes", content: new Blob(["*.html filter=lfs diff=lfs merge=lfs -text"]) }],
});
assert.deepStrictEqual(result, {
repoUrl: `${TEST_HUB_URL}/${repoName}`,
});
const content = await downloadFile({
repo: {
name: repoName,
type: "model",
},
path: ".gitattributes",
hubUrl: TEST_HUB_URL,
});
assert(content);
assert.strictEqual(await content.text(), "*.html filter=lfs diff=lfs merge=lfs -text");
await deleteRepo({
repo: {
name: repoName,
type: "model",
},
credentials: { accessToken: TEST_ACCESS_TOKEN },
hubUrl: TEST_HUB_URL,
});
});
it("should throw a client error when trying to create a repo without a fully-qualified name", async () => {
const tryCreate = createRepo({
repo: { name: "canonical", type: "model" },
credentials: { accessToken: TEST_ACCESS_TOKEN },
hubUrl: TEST_HUB_URL,
});
await expect(tryCreate).rejects.toBeInstanceOf(TypeError);
});
it("should create a model with a string as name", async () => {
const repoName = `${TEST_USER}/TEST-${insecureRandomString()}`;
const result = await createRepo({
accessToken: TEST_ACCESS_TOKEN,
hubUrl: TEST_HUB_URL,
repo: repoName,
files: [{ path: ".gitattributes", content: new Blob(["*.html filter=lfs diff=lfs merge=lfs -text"]) }],
});
assert.deepStrictEqual(result, {
repoUrl: `${TEST_HUB_URL}/${repoName}`,
});
await deleteRepo({
repo: {
name: repoName,
type: "model",
},
hubUrl: TEST_HUB_URL,
credentials: { accessToken: TEST_ACCESS_TOKEN },
});
});
it("should create a dataset with a string as name", async () => {
const repoName = `datasets/${TEST_USER}/TEST-${insecureRandomString()}`;
const result = await createRepo({
accessToken: TEST_ACCESS_TOKEN,
hubUrl: TEST_HUB_URL,
repo: repoName,
files: [{ path: ".gitattributes", content: new Blob(["*.html filter=lfs diff=lfs merge=lfs -text"]) }],
});
assert.deepStrictEqual(result, {
repoUrl: `${TEST_HUB_URL}/${repoName}`,
});
await deleteRepo({
repo: repoName,
hubUrl: TEST_HUB_URL,
credentials: { accessToken: TEST_ACCESS_TOKEN },
});
});
});
|