forked from Kieran/snort
205 lines
5.6 KiB
TypeScript
205 lines
5.6 KiB
TypeScript
import { RelayCache } from "../src/gossip-model";
|
|
import { RequestBuilder, RequestStrategy } from "../src/request-builder";
|
|
import { describe, expect } from "@jest/globals";
|
|
import { expandFilter } from "../src/request-expander";
|
|
import { bytesToHex } from "@noble/curves/abstract/utils";
|
|
import { unixNow, unixNowMs } from "@snort/shared";
|
|
|
|
const DummyCache = {
|
|
getFromCache: (pk?: string) => {
|
|
if (!pk) return undefined;
|
|
|
|
return {
|
|
pubkey: pk,
|
|
created_at: unixNow(),
|
|
relays: [
|
|
{
|
|
url: `wss://${pk}.com/`,
|
|
settings: {
|
|
read: true,
|
|
write: true,
|
|
},
|
|
},
|
|
],
|
|
};
|
|
},
|
|
} as RelayCache;
|
|
|
|
describe("RequestBuilder", () => {
|
|
describe("basic", () => {
|
|
test("empty filter", () => {
|
|
const b = new RequestBuilder("test");
|
|
b.withFilter();
|
|
expect(b.buildRaw()).toEqual([{}]);
|
|
});
|
|
test("only kind", () => {
|
|
const b = new RequestBuilder("test");
|
|
b.withFilter().kinds([0]);
|
|
expect(b.buildRaw()).toMatchObject([{ kinds: [0] }]);
|
|
});
|
|
test("empty authors", () => {
|
|
const b = new RequestBuilder("test");
|
|
b.withFilter().authors([]);
|
|
expect(b.buildRaw()).toMatchObject([{ authors: [] }]);
|
|
});
|
|
test("authors/kinds/ids", () => {
|
|
const authors = ["a1", "a2"];
|
|
const kinds = [0, 1, 2, 3];
|
|
const ids = ["id1", "id2", "id3"];
|
|
const b = new RequestBuilder("test");
|
|
b.withFilter().authors(authors).kinds(kinds).ids(ids);
|
|
expect(b.buildRaw()).toMatchObject([{ ids, authors, kinds }]);
|
|
});
|
|
test("authors and kinds, duplicates removed", () => {
|
|
const authors = ["a1", "a2"];
|
|
const kinds = [0, 1, 2, 3];
|
|
const ids = ["id1", "id2", "id3"];
|
|
const b = new RequestBuilder("test");
|
|
b.withFilter().ids(ids).authors(authors).kinds(kinds).ids(ids).authors(authors).kinds(kinds);
|
|
expect(b.buildRaw()).toMatchObject([{ ids, authors, kinds }]);
|
|
});
|
|
test("search", () => {
|
|
const b = new RequestBuilder("test");
|
|
b.withFilter().kinds([1]).search("test-search");
|
|
expect(b.buildRaw()).toMatchObject([{ kinds: [1], search: "test-search" }]);
|
|
});
|
|
test("timeline", () => {
|
|
const authors = ["a1", "a2"];
|
|
const kinds = [0, 1, 2, 3];
|
|
const until = 10;
|
|
const since = 5;
|
|
const b = new RequestBuilder("test");
|
|
b.withFilter().kinds(kinds).authors(authors).since(since).until(until);
|
|
expect(b.buildRaw()).toMatchObject([{ kinds, authors, until, since }]);
|
|
});
|
|
test("multi-filter timeline", () => {
|
|
const authors = ["a1", "a2"];
|
|
const kinds = [0, 1, 2, 3];
|
|
const until = 10;
|
|
const since = 5;
|
|
const b = new RequestBuilder("test");
|
|
b.withFilter().kinds(kinds).authors(authors).since(since).until(until);
|
|
b.withFilter().kinds(kinds).authors(authors).since(since).until(until);
|
|
expect(b.buildRaw()).toMatchObject([
|
|
{ kinds, authors, until, since },
|
|
{ kinds, authors, until, since },
|
|
]);
|
|
});
|
|
});
|
|
|
|
describe("diff basic", () => {
|
|
const rb = new RequestBuilder("test");
|
|
const f0 = rb.withFilter();
|
|
|
|
const a = rb.buildRaw();
|
|
f0.authors(["a"]);
|
|
expect(a).toEqual([{}]);
|
|
|
|
const b = rb.buildDiff(DummyCache, a.flatMap(expandFilter));
|
|
expect(b).toMatchObject([
|
|
{
|
|
filters: [{ authors: ["a"] }],
|
|
},
|
|
]);
|
|
});
|
|
|
|
describe("build gossip simply", () => {
|
|
const rb = new RequestBuilder("test");
|
|
rb.withFilter().authors(["a", "b"]).kinds([0]);
|
|
|
|
const a = rb.build(DummyCache);
|
|
expect(a).toEqual([
|
|
{
|
|
strategy: RequestStrategy.AuthorsRelays,
|
|
relay: "wss://a.com/",
|
|
filters: [
|
|
{
|
|
kinds: [0],
|
|
authors: ["a"],
|
|
},
|
|
],
|
|
},
|
|
{
|
|
strategy: RequestStrategy.AuthorsRelays,
|
|
relay: "wss://b.com/",
|
|
filters: [
|
|
{
|
|
kinds: [0],
|
|
authors: ["b"],
|
|
},
|
|
],
|
|
},
|
|
]);
|
|
});
|
|
|
|
describe("build gossip merged similar filters", () => {
|
|
const rb = new RequestBuilder("test");
|
|
rb.withFilter().authors(["a", "b"]).kinds([0]);
|
|
rb.withFilter().authors(["a", "b"]).kinds([10002]);
|
|
rb.withFilter().authors(["a"]).limit(10).kinds([4]);
|
|
|
|
const a = rb.build(DummyCache);
|
|
expect(a).toEqual([
|
|
{
|
|
strategy: RequestStrategy.AuthorsRelays,
|
|
relay: "wss://a.com/",
|
|
filters: [
|
|
{
|
|
kinds: [0, 10002],
|
|
authors: ["a"],
|
|
},
|
|
{
|
|
kinds: [4],
|
|
authors: ["a"],
|
|
limit: 10,
|
|
},
|
|
],
|
|
},
|
|
{
|
|
strategy: RequestStrategy.AuthorsRelays,
|
|
relay: "wss://b.com/",
|
|
filters: [
|
|
{
|
|
kinds: [0, 10002],
|
|
authors: ["b"],
|
|
},
|
|
],
|
|
},
|
|
]);
|
|
});
|
|
});
|
|
|
|
describe("build diff, large follow list", () => {
|
|
const f = [];
|
|
for (let x = 0; x < 2500; x++) {
|
|
const bytes = crypto.getRandomValues(new Uint8Array(32));
|
|
f.push(bytesToHex(bytes));
|
|
}
|
|
|
|
const rb = new RequestBuilder("test");
|
|
rb.withFilter().authors(f).kinds([1, 6, 10002, 3, 6969]);
|
|
|
|
const start = unixNowMs();
|
|
const a = rb.build(DummyCache);
|
|
expect(a).toEqual(
|
|
f.map(a => {
|
|
return {
|
|
strategy: RequestStrategy.AuthorsRelays,
|
|
relay: `wss://${a}.com/`,
|
|
filters: [
|
|
{
|
|
kinds: [1, 6, 10002, 3, 6969],
|
|
authors: [a],
|
|
},
|
|
],
|
|
};
|
|
})
|
|
);
|
|
expect(unixNowMs() - start).toBeLessThan(500);
|
|
|
|
const start2 = unixNowMs();
|
|
const b = rb.buildDiff(DummyCache, rb.buildRaw().flatMap(expandFilter));
|
|
expect(b).toEqual([]);
|
|
expect(unixNowMs() - start2).toBeLessThan(100);
|
|
});
|