mirror of
https://github.com/moltbot/moltbot.git
synced 2026-05-13 15:47:28 +00:00
test: speed up sqlite vec loader test
This commit is contained in:
@@ -1,4 +1,5 @@
|
|||||||
import { afterEach, describe, expect, it, vi } from "vitest";
|
import { afterEach, describe, expect, it, vi } from "vitest";
|
||||||
|
import { loadSqliteVecExtension } from "./sqlite-vec.js";
|
||||||
|
|
||||||
function mockMissingSqliteVecPackage(): void {
|
function mockMissingSqliteVecPackage(): void {
|
||||||
vi.doMock("sqlite-vec", () => {
|
vi.doMock("sqlite-vec", () => {
|
||||||
@@ -8,19 +9,13 @@ function mockMissingSqliteVecPackage(): void {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
async function importLoader() {
|
|
||||||
return import("./sqlite-vec.js");
|
|
||||||
}
|
|
||||||
|
|
||||||
afterEach(() => {
|
afterEach(() => {
|
||||||
vi.doUnmock("sqlite-vec");
|
vi.doUnmock("sqlite-vec");
|
||||||
vi.resetModules();
|
|
||||||
});
|
});
|
||||||
|
|
||||||
describe("loadSqliteVecExtension", () => {
|
describe("loadSqliteVecExtension", () => {
|
||||||
it("loads explicit extensionPath without importing bundled sqlite-vec", async () => {
|
it("loads explicit extensionPath without importing bundled sqlite-vec", async () => {
|
||||||
mockMissingSqliteVecPackage();
|
mockMissingSqliteVecPackage();
|
||||||
const { loadSqliteVecExtension } = await importLoader();
|
|
||||||
const db = {
|
const db = {
|
||||||
enableLoadExtension: vi.fn(),
|
enableLoadExtension: vi.fn(),
|
||||||
loadExtension: vi.fn(),
|
loadExtension: vi.fn(),
|
||||||
@@ -38,7 +33,6 @@ describe("loadSqliteVecExtension", () => {
|
|||||||
|
|
||||||
it("returns a valid memorySearch extensionPath hint when sqlite-vec is absent", async () => {
|
it("returns a valid memorySearch extensionPath hint when sqlite-vec is absent", async () => {
|
||||||
mockMissingSqliteVecPackage();
|
mockMissingSqliteVecPackage();
|
||||||
const { loadSqliteVecExtension } = await importLoader();
|
|
||||||
const db = {
|
const db = {
|
||||||
enableLoadExtension: vi.fn(),
|
enableLoadExtension: vi.fn(),
|
||||||
loadExtension: vi.fn(),
|
loadExtension: vi.fn(),
|
||||||
|
|||||||
Reference in New Issue
Block a user