71 lines
2.0 KiB
TypeScript
71 lines
2.0 KiB
TypeScript
import { expect } from "vitest";
|
|
import type { KvPrimitives } from "./kv_primitives.ts";
|
|
|
|
import type { KV } from "../../plug-api/types/datastore.ts";
|
|
|
|
export async function allTests(db: KvPrimitives) {
|
|
await db.batchSet([
|
|
{ key: ["kv", "test2"], value: "Hello2" },
|
|
{ key: ["kv", "test1"], value: "Hello1" },
|
|
{ key: ["other", "random"], value: "Hello3" },
|
|
]);
|
|
const result = await db.batchGet([["kv", "test1"], ["kv", "test2"], [
|
|
"kv",
|
|
"test3",
|
|
]]);
|
|
expect(result.length).toEqual(3);
|
|
expect(result[0]).toEqual("Hello1");
|
|
expect(result[1]).toEqual("Hello2");
|
|
expect(result[2]).toEqual(undefined);
|
|
let counter = 0;
|
|
// Query all
|
|
for await (const _entry of db.query({})) {
|
|
counter++;
|
|
}
|
|
expect(counter).toEqual(3);
|
|
|
|
counter = 0;
|
|
// Query prefix
|
|
for await (const _entry of db.query({ prefix: ["kv"] })) {
|
|
counter++;
|
|
console.log(_entry);
|
|
}
|
|
expect(counter).toEqual(2);
|
|
|
|
// Delete a few keys
|
|
await db.batchDelete([["kv", "test1"], ["other", "random"]]);
|
|
const result2 = await db.batchGet([["kv", "test1"], ["kv", "test2"], [
|
|
"other",
|
|
"random",
|
|
]]);
|
|
expect(result2.length).toEqual(3);
|
|
expect(result2[0]).toEqual(undefined);
|
|
expect(result2[1]).toEqual("Hello2");
|
|
expect(result2[2]).toEqual(undefined);
|
|
|
|
// Update a key
|
|
await db.batchSet([{ key: ["kv", "test2"], value: "Hello2.1" }]);
|
|
const [val] = await db.batchGet([["kv", "test2"]]);
|
|
expect(val).toEqual("Hello2.1");
|
|
|
|
// Set a large batch
|
|
const largeBatch: KV[] = [];
|
|
for (let i = 0; i < 50; i++) {
|
|
largeBatch.push({ key: ["test", "test" + i], value: "Hello" });
|
|
}
|
|
await db.batchSet(largeBatch);
|
|
const largeBatchResult: KV[] = [];
|
|
for await (const entry of db.query({ prefix: ["test"] })) {
|
|
largeBatchResult.push(entry);
|
|
}
|
|
expect(largeBatchResult.length).toEqual(50);
|
|
|
|
// Delete the large batch
|
|
await db.batchDelete(largeBatch.map((e) => e.key));
|
|
|
|
// Make sure they're gone
|
|
for await (const _entry of db.query({ prefix: ["test"] })) {
|
|
throw new Error("This should not happen");
|
|
}
|
|
}
|