@pinecone-database/pinecone
Version:
This is the official Node.js SDK for [Pinecone](https://www.pinecone.io), written in TypeScript.
62 lines • 2.93 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
const list_1 = require("../../vectors/list");
const setupListResponse = (response, isSuccess = true) => {
const fakeList = jest
.fn()
.mockImplementation(() => isSuccess ? Promise.resolve(response) : Promise.reject(response));
const VOA = { listVectors: fakeList };
const VectorProvider = {
provide: async () => VOA,
};
return { VOA: VOA, VectorProvider: VectorProvider };
};
describe('list', () => {
test('listPaginated calls the openapi list endpoint, passing target namespace with ListOptions', async () => {
const listResponse = {
vectors: [
{ id: 'prefix-1', values: [0.2, 0.4] },
{ id: 'prefix-2', values: [0.3, 0.5] },
{ id: 'prefix-3', values: [0.4, 0.6] },
],
pagination: { next: 'fake-pagination-token-123123123' },
namespace: 'list-namespace',
usage: { readUnits: 1 },
};
const { VectorProvider, VOA } = setupListResponse(listResponse);
const listPaginatedFn = (0, list_1.listPaginated)(VectorProvider, 'list-namespace');
const returned = await listPaginatedFn({ prefix: 'prefix-' });
expect(returned).toBe(listResponse);
expect(VOA.listVectors).toHaveBeenCalledWith({
prefix: 'prefix-',
namespace: 'list-namespace',
});
});
test('Throw error if pass in empty prefix', async () => {
const { VectorProvider } = setupListResponse({});
const listPaginatedFn = (0, list_1.listPaginated)(VectorProvider, 'list-namespace');
const toThrow = async () => {
await listPaginatedFn({ limit: -3 });
};
await expect(toThrow()).rejects.toThrow('`limit` property must be greater than 0');
});
test('Throw error if misspell property', async () => {
const { VectorProvider } = setupListResponse({});
const listPaginatedFn = (0, list_1.listPaginated)(VectorProvider, 'list-namespace');
const toThrow = async () => {
// @ts-ignore
await listPaginatedFn({ limitgadsf: -3 });
};
await expect(toThrow()).rejects.toThrow('Object contained invalid properties: limitgadsf. Valid properties include prefix, limit, paginationToken.');
});
test('Throw error if add unknown property', async () => {
const { VectorProvider } = setupListResponse({});
const listPaginatedFn = (0, list_1.listPaginated)(VectorProvider, 'list-namespace');
const toThrow = async () => {
// @ts-ignore
await listPaginatedFn({ limit: 3, testy: 'test' });
};
await expect(toThrow()).rejects.toThrow('Object contained invalid properties: testy. Valid properties include prefix, limit, paginationToken.');
});
});
//# sourceMappingURL=list.test.js.map