58 lines
1.3 KiB
TypeScript
58 lines
1.3 KiB
TypeScript
import { Connection } from "@snort/nostr";
|
|
import { Query } from "./Query";
|
|
import { getRandomValues } from "crypto";
|
|
|
|
window.crypto = {} as any;
|
|
window.crypto.getRandomValues = getRandomValues as any;
|
|
|
|
describe("query", () => {
|
|
test("progress", () => {
|
|
const q = new Query("test", [
|
|
{
|
|
kinds: [1],
|
|
authors: ["test"],
|
|
},
|
|
]);
|
|
const opt = {
|
|
read: true,
|
|
write: true,
|
|
};
|
|
const c1 = new Connection("wss://one.com", opt);
|
|
c1.Down = false;
|
|
const c2 = new Connection("wss://two.com", opt);
|
|
c2.Down = false;
|
|
const c3 = new Connection("wss://three.com", opt);
|
|
c3.Down = false;
|
|
|
|
q.sendToRelay(c1);
|
|
q.sendToRelay(c2);
|
|
q.sendToRelay(c3);
|
|
|
|
expect(q.progress).toBe(0);
|
|
q.eose(q.id, c1);
|
|
expect(q.progress).toBe(1 / 3);
|
|
q.eose(q.id, c1);
|
|
expect(q.progress).toBe(1 / 3);
|
|
q.eose(q.id, c2);
|
|
expect(q.progress).toBe(2 / 3);
|
|
q.eose(q.id, c3);
|
|
expect(q.progress).toBe(1);
|
|
|
|
const qs = new Query("test-1", [
|
|
{
|
|
kinds: [1],
|
|
authors: ["test-sub"],
|
|
},
|
|
]);
|
|
q.subQueries.push(qs);
|
|
qs.sendToRelay(c1);
|
|
|
|
expect(q.progress).toBe(0.5);
|
|
q.eose(qs.id, c1);
|
|
expect(q.progress).toBe(1);
|
|
qs.sendToRelay(c2);
|
|
// 1 + 0.5 (1/2 sent sub query)
|
|
expect(q.progress).toBe(1.5 / 2);
|
|
});
|
|
});
|