mirror of
https://github.com/BerriAI/litellm.git
synced 2026-11-04 07:27:51 +00:00
test(ui): move vector store inventory tests next to ragIngestCall
Both this branch and main (#48478) appended a describe block at the end of networking.test.ts, so the PR conflicted with main. Keeping the vector store tests next to the other RAG and vector store test avoids that
This commit is contained in:
Tanishq
parent
1188ea351c
commit
0b0a7e8833
1 changed files with 87 additions and 87 deletions
|
|
@ -999,6 +999,93 @@ describe("ragIngestCall", () => {
|
|||
});
|
||||
});
|
||||
|
||||
describe("vectorStoreListCall complete dashboard inventory", () => {
|
||||
const stores = Array.from({ length: 203 }, (_, index) => ({
|
||||
vector_store_id: `vs_${String(index).padStart(3, "0")}`,
|
||||
custom_llm_provider: "openai",
|
||||
created_at: "2026-01-01T00:00:00Z",
|
||||
updated_at: "2026-01-01T00:00:00Z",
|
||||
}));
|
||||
|
||||
afterEach(() => {
|
||||
vi.unstubAllGlobals();
|
||||
Networking.setGlobalLitellmHeaderName("Authorization");
|
||||
});
|
||||
|
||||
it("loads every page with the configured authentication header", async () => {
|
||||
const requests: { path: string; page: string | null; pageSize: string | null; auth: string | null }[] = [];
|
||||
const mockFetch = vi.fn<typeof fetch>().mockImplementation(async (input, init) => {
|
||||
const url = new URL(input instanceof Request ? input.url : String(input), "http://localhost");
|
||||
const request = {
|
||||
path: url.pathname,
|
||||
page: url.searchParams.get("page"),
|
||||
pageSize: url.searchParams.get("page_size"),
|
||||
auth: new Headers(init?.headers).get("x-litellm-key"),
|
||||
};
|
||||
requests.push(request);
|
||||
const page = requests.length;
|
||||
const inventory = {
|
||||
object: "list",
|
||||
data: stores.slice((page - 1) * 100, page * 100),
|
||||
current_page: page,
|
||||
total_count: stores.length,
|
||||
total_pages: 3,
|
||||
};
|
||||
return new Response(JSON.stringify(inventory));
|
||||
});
|
||||
vi.stubGlobal("fetch", mockFetch);
|
||||
Networking.setGlobalLitellmHeaderName("x-litellm-key");
|
||||
|
||||
const response = await Networking.vectorStoreListCall("inventory-token");
|
||||
|
||||
expect(response.data).toEqual(stores);
|
||||
expect(requests).toEqual([
|
||||
{ path: "/vector_store/list", page: "1", pageSize: "100", auth: "Bearer inventory-token" },
|
||||
{ path: "/vector_store/list", page: "2", pageSize: "100", auth: "Bearer inventory-token" },
|
||||
{ path: "/vector_store/list", page: "3", pageSize: "100", auth: "Bearer inventory-token" },
|
||||
]);
|
||||
});
|
||||
|
||||
it("does not duplicate an older backend's already-complete first response", async () => {
|
||||
const inventory = { object: "list", data: stores, total_count: stores.length, current_page: 1, total_pages: 3 };
|
||||
const mockFetch = vi.fn<typeof fetch>().mockResolvedValue(new Response(JSON.stringify(inventory)));
|
||||
vi.stubGlobal("fetch", mockFetch);
|
||||
|
||||
const response = await Networking.vectorStoreListCall("inventory-token");
|
||||
|
||||
expect(response.data).toEqual(stores);
|
||||
expect(mockFetch).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it("keeps an empty inventory empty without requesting another page", async () => {
|
||||
const inventory = { object: "list", data: [], total_count: 0, current_page: 1, total_pages: 0 };
|
||||
const mockFetch = vi.fn<typeof fetch>().mockResolvedValue(new Response(JSON.stringify(inventory)));
|
||||
vi.stubGlobal("fetch", mockFetch);
|
||||
|
||||
const response = await Networking.vectorStoreListCall("inventory-token");
|
||||
|
||||
expect(response.data).toEqual([]);
|
||||
expect(mockFetch).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it("reports a later page error instead of returning a partial inventory", async () => {
|
||||
const inventory = {
|
||||
object: "list",
|
||||
data: stores.slice(0, 100),
|
||||
total_count: stores.length,
|
||||
current_page: 1,
|
||||
total_pages: 3,
|
||||
};
|
||||
const mockFetch = vi
|
||||
.fn<typeof fetch>()
|
||||
.mockResolvedValueOnce(new Response(JSON.stringify(inventory)))
|
||||
.mockResolvedValueOnce(new Response(JSON.stringify({ detail: "Catalog page unavailable" }), { status: 503 }));
|
||||
vi.stubGlobal("fetch", mockFetch);
|
||||
|
||||
await expect(Networking.vectorStoreListCall("inventory-token")).rejects.toThrow("Catalog page unavailable");
|
||||
});
|
||||
});
|
||||
|
||||
describe("getUiConfig runtime backend discovery", () => {
|
||||
const discoveryConfig = () => ({
|
||||
server_root_path: "/",
|
||||
|
|
@ -1191,90 +1278,3 @@ describe("getUiConfig runtime backend discovery", () => {
|
|||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe("vectorStoreListCall complete dashboard inventory", () => {
|
||||
const stores = Array.from({ length: 203 }, (_, index) => ({
|
||||
vector_store_id: `vs_${String(index).padStart(3, "0")}`,
|
||||
custom_llm_provider: "openai",
|
||||
created_at: "2026-01-01T00:00:00Z",
|
||||
updated_at: "2026-01-01T00:00:00Z",
|
||||
}));
|
||||
|
||||
afterEach(() => {
|
||||
vi.unstubAllGlobals();
|
||||
Networking.setGlobalLitellmHeaderName("Authorization");
|
||||
});
|
||||
|
||||
it("loads every page with the configured authentication header", async () => {
|
||||
const requests: { path: string; page: string | null; pageSize: string | null; auth: string | null }[] = [];
|
||||
const mockFetch = vi.fn<typeof fetch>().mockImplementation(async (input, init) => {
|
||||
const url = new URL(input instanceof Request ? input.url : String(input), "http://localhost");
|
||||
const request = {
|
||||
path: url.pathname,
|
||||
page: url.searchParams.get("page"),
|
||||
pageSize: url.searchParams.get("page_size"),
|
||||
auth: new Headers(init?.headers).get("x-litellm-key"),
|
||||
};
|
||||
requests.push(request);
|
||||
const page = requests.length;
|
||||
const inventory = {
|
||||
object: "list",
|
||||
data: stores.slice((page - 1) * 100, page * 100),
|
||||
current_page: page,
|
||||
total_count: stores.length,
|
||||
total_pages: 3,
|
||||
};
|
||||
return new Response(JSON.stringify(inventory));
|
||||
});
|
||||
vi.stubGlobal("fetch", mockFetch);
|
||||
Networking.setGlobalLitellmHeaderName("x-litellm-key");
|
||||
|
||||
const response = await Networking.vectorStoreListCall("inventory-token");
|
||||
|
||||
expect(response.data).toEqual(stores);
|
||||
expect(requests).toEqual([
|
||||
{ path: "/vector_store/list", page: "1", pageSize: "100", auth: "Bearer inventory-token" },
|
||||
{ path: "/vector_store/list", page: "2", pageSize: "100", auth: "Bearer inventory-token" },
|
||||
{ path: "/vector_store/list", page: "3", pageSize: "100", auth: "Bearer inventory-token" },
|
||||
]);
|
||||
});
|
||||
|
||||
it("does not duplicate an older backend's already-complete first response", async () => {
|
||||
const inventory = { object: "list", data: stores, total_count: stores.length, current_page: 1, total_pages: 3 };
|
||||
const mockFetch = vi.fn<typeof fetch>().mockResolvedValue(new Response(JSON.stringify(inventory)));
|
||||
vi.stubGlobal("fetch", mockFetch);
|
||||
|
||||
const response = await Networking.vectorStoreListCall("inventory-token");
|
||||
|
||||
expect(response.data).toEqual(stores);
|
||||
expect(mockFetch).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it("keeps an empty inventory empty without requesting another page", async () => {
|
||||
const inventory = { object: "list", data: [], total_count: 0, current_page: 1, total_pages: 0 };
|
||||
const mockFetch = vi.fn<typeof fetch>().mockResolvedValue(new Response(JSON.stringify(inventory)));
|
||||
vi.stubGlobal("fetch", mockFetch);
|
||||
|
||||
const response = await Networking.vectorStoreListCall("inventory-token");
|
||||
|
||||
expect(response.data).toEqual([]);
|
||||
expect(mockFetch).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it("reports a later page error instead of returning a partial inventory", async () => {
|
||||
const inventory = {
|
||||
object: "list",
|
||||
data: stores.slice(0, 100),
|
||||
total_count: stores.length,
|
||||
current_page: 1,
|
||||
total_pages: 3,
|
||||
};
|
||||
const mockFetch = vi
|
||||
.fn<typeof fetch>()
|
||||
.mockResolvedValueOnce(new Response(JSON.stringify(inventory)))
|
||||
.mockResolvedValueOnce(new Response(JSON.stringify({ detail: "Catalog page unavailable" }), { status: 503 }));
|
||||
vi.stubGlobal("fetch", mockFetch);
|
||||
|
||||
await expect(Networking.vectorStoreListCall("inventory-token")).rejects.toThrow("Catalog page unavailable");
|
||||
});
|
||||
});
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue