Move domain stuff to sub directory, clean up base classes

2026-06-19 10:42:58 -07:00
parent bfd91f2d39
commit 1bd62d3024
95 changed files with 4956 additions and 1353 deletions
@@ -0,0 +1,123 @@
+import {describe, it, expect} from "vitest"
+import {makeSecret, POLL, NOTE} from "@welshman/util"
+import type {TrustedEvent} from "@welshman/util"
+import {Nip01Signer} from "@welshman/signer"
+import {Poll, PollBuilder} from "../src/kinds/Poll"
+
+const signer = new Nip01Signer(makeSecret())
+const pubkey = "ee".repeat(32)
+
+const makeEvent = (overrides: Partial<TrustedEvent> = {}): TrustedEvent =>
+  ({
+    id: "ff".repeat(32),
+    pubkey,
+    created_at: 0,
+    kind: POLL,
+    tags: [],
+    content: "",
+    sig: "00".repeat(64),
+    ...overrides,
+  }) as TrustedEvent
+
+describe("Poll", () => {
+  it("parses the represented tags and plain-text title", async () => {
+    const event = makeEvent({
+      content: "Favorite color?",
+      tags: [
+        ["option", "1", "Red"],
+        ["option", "2", "Blue"],
+        ["polltype", "multiplechoice"],
+        ["endsAt", "1234"],
+        ["relay", "wss://relay.one"],
+        ["relay", "wss://relay.two"],
+        ["alt", "x"],
+      ],
+    })
+
+    const poll = await Poll.fromEvent(event)
+
+    expect(poll.title()).toBe("Favorite color?")
+    expect(poll.options()).toEqual([
+      {id: "1", label: "Red"},
+      {id: "2", label: "Blue"},
+    ])
+    expect(poll.pollType()).toBe("multiplechoice")
+    expect(poll.endsAt()).toBe(1234)
+    expect(poll.relays()).toEqual(["wss://relay.one", "wss://relay.two"])
+  })
+
+  it("tallies results from response events", async () => {
+    const poll = await Poll.fromEvent(
+      makeEvent({
+        content: "Pick one",
+        tags: [
+          ["option", "1", "Red"],
+          ["option", "2", "Blue"],
+        ],
+      }),
+    )
+
+    const responses = [
+      {pubkey: "a", created_at: 1, tags: [["response", "1"]]},
+      {pubkey: "b", created_at: 1, tags: [["response", "2"]]},
+      // Latest response per pubkey wins.
+      {pubkey: "a", created_at: 2, tags: [["response", "2"]]},
+    ] as TrustedEvent[]
+
+    const result = poll.results(responses)
+
+    expect(result.voters).toBe(2)
+    expect(result.options.find(o => o.id === "1")!.votes).toBe(0)
+    expect(result.options.find(o => o.id === "2")!.votes).toBe(2)
+  })
+
+  it("round-trips with no duplication", async () => {
+    const event = makeEvent({
+      content: "Favorite color?",
+      tags: [
+        ["option", "1", "Red"],
+        ["option", "2", "Blue"],
+        ["polltype", "multiplechoice"],
+        ["endsAt", "1234"],
+        ["relay", "wss://relay.one"],
+        ["alt", "x"],
+      ],
+    })
+
+    const tmpl = await (await Poll.fromEvent(event)).builder().toTemplate(signer)
+
+    expect(tmpl.content).toBe("Favorite color?")
+    expect(tmpl.tags.filter(t => t[0] === "option").length).toBe(2)
+    expect(tmpl.tags.filter(t => t[0] === "polltype").length).toBe(1)
+    expect(tmpl.tags.filter(t => t[0] === "endsAt").length).toBe(1)
+    expect(tmpl.tags.filter(t => t[0] === "relay").length).toBe(1)
+    // Unknown tag survives the round-trip.
+    expect(tmpl.tags).toContainEqual(["alt", "x"])
+  })
+
+  it("builds from a fresh builder", async () => {
+    const tmpl = await new PollBuilder()
+      .setTitle("Q?")
+      .addOption("Red", "1")
+      .addOption("Blue", "2")
+      .setPollType("multiplechoice")
+      .setEndsAt(9999)
+      .setRelays(["wss://relay.one"])
+      .toTemplate(signer)
+
+    expect(tmpl.kind).toBe(POLL)
+    expect(tmpl.content).toBe("Q?")
+    expect(tmpl.tags).toContainEqual(["option", "1", "Red"])
+    expect(tmpl.tags).toContainEqual(["polltype", "multiplechoice"])
+    expect(tmpl.tags).toContainEqual(["endsAt", "9999"])
+    expect(tmpl.tags).toContainEqual(["relay", "wss://relay.one"])
+  })
+
+  it("requires at least one option", async () => {
+    await expect(new PollBuilder().setTitle("Q?").toTemplate(signer)).rejects.toThrow()
+  })
+
+  it("throws on the wrong kind", async () => {
+    await expect(Poll.fromEvent(makeEvent({kind: NOTE}))).rejects.toThrow()
+  })
+})