Spaces:
Running
Running
File size: 2,992 Bytes
287a0bc |
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 |
import {expect, test} from "@jest/globals";
import {ChromaClient} from "../src/ChromaClient";
import {chromaTokenDefault, chromaTokenBearer, chromaTokenXToken, cloudClient} from "./initClientWithAuth";
import chromaNoAuth from "./initClient";
test("it should get the version without auth needed", async () => {
const version = await chromaNoAuth.version();
expect(version).toBeDefined();
expect(version).toMatch(/^[0-9]+\.[0-9]+\.[0-9]+$/);
});
test("it should get the heartbeat without auth needed", async () => {
const heartbeat = await chromaNoAuth.heartbeat();
expect(heartbeat).toBeDefined();
expect(heartbeat).toBeGreaterThan(0);
});
test("it should raise error when non authenticated", async () => {
await expect(chromaNoAuth.listCollections()).rejects.toMatchObject({
status: 401
});
});
if (!process.env.XTOKEN_TEST) {
test('it should list collections with default token config', async () => {
await chromaTokenDefault.reset()
let collections = await chromaTokenDefault.listCollections()
expect(collections).toBeDefined()
expect(collections).toBeInstanceOf(Array)
expect(collections.length).toBe(0)
const collection = await chromaTokenDefault.createCollection({name: "test"});
collections = await chromaTokenDefault.listCollections()
expect(collections.length).toBe(1)
})
test('it should list collections with explicit bearer token config', async () => {
await chromaTokenBearer.reset()
let collections = await chromaTokenBearer.listCollections()
expect(collections).toBeDefined()
expect(collections).toBeInstanceOf(Array)
expect(collections.length).toBe(0)
const collection = await chromaTokenBearer.createCollection({name: "test"});
collections = await chromaTokenBearer.listCollections()
expect(collections.length).toBe(1)
})
} else {
test('it should list collections with explicit x-token token config', async () => {
await chromaTokenXToken.reset()
let collections = await chromaTokenXToken.listCollections()
expect(collections).toBeDefined()
expect(collections).toBeInstanceOf(Array)
expect(collections.length).toBe(0)
const collection = await chromaTokenXToken.createCollection({name: "test"});
collections = await chromaTokenXToken.listCollections()
expect(collections.length).toBe(1)
})
test('it should list collections with explicit x-token token config in CloudClient', async () => {
await cloudClient.reset()
let collections = await cloudClient.listCollections()
expect(collections).toBeDefined()
expect(collections).toBeInstanceOf(Array)
expect(collections.length).toBe(0)
const collection = await cloudClient.createCollection({name: "test"});
collections = await cloudClient.listCollections()
expect(collections.length).toBe(1)
})
}
|