pairing-dial.test.ts

1import { beforeEach, describe, expect, it, vi } from "vitest";23import { dialPairingChannel, type PairingSocket } from "./pairing-dial";4import { encodePairingCode } from "./pairing-code";56const enc = (o: unknown) => new TextEncoder().encode(JSON.stringify(o));7const dec = (b: Uint8Array) => JSON.parse(new TextDecoder().decode(b));89const LISTENER_EPHEMERAL_HEX = "a".repeat(64);10const CODE = encodePairingCode("192.168.4.77", 51555, Uint8Array.of(0xde, 0xad, 0xbe, 0xef));1112const FAKE_HANDSHAKE_HASH = Uint8Array.from([0x0f, 0xba, 0xb8, 0xcd, ...new Array(28).fill(0)]);1314// Skip the real handshake messages; encrypt/decrypt are identity transforms so15// the channel's framing can be asserted directly.16vi.mock("./noise-js", async (orig) => {17  const actual = await orig<typeof import("./noise-js")>();18  return {19    ...actual,20    noiseNKInitiator: vi.fn(async () => ({21      encrypt: (pt: Uint8Array) => pt,22      decrypt: (ct: Uint8Array) => ct,23      handshakeHash: FAKE_HANDSHAKE_HASH,24    })),25  };26});2728/** `null` in the queue means "the socket closes here with no data". */29function fakeSocket(inbound: Array<Uint8Array | null>) {30  const sent: Uint8Array[] = [];31  let closed = false;32  let lastWasClose = false;33  const queue = [...inbound];34  const sock: PairingSocket = {35    send(bytes) { sent.push(bytes); },36    async receive() {37      const next = queue.shift();38      if (next === undefined || next === null) {39        closed = true;40        lastWasClose = true;41        throw new Error("socket closed");42      }43      lastWasClose = false;44      return next;45    },46    close() { closed = true; },47    closedBeforeFrame() { return lastWasClose; },48  };49  return { sock, sent, isClosed: () => closed };50}5152function ackThen(...rest: Array<Uint8Array | null>) {53  return fakeSocket([54    enc({ type: "PEACH_HELLO_ACK", ephemeralPublicKeyHex: LISTENER_EPHEMERAL_HEX }),55    ...rest,56  ]);57}5859beforeEach(() => { vi.clearAllMocks(); });6061describe("dialPairingChannel", () => {62  it("dials the coded endpoint", async () => {63    const openSocket = vi.fn(async () => ackThen().sock);6465    const result = await dialPairingChannel(CODE, { openSocket });6667    expect(result.ok).toBe(true);68    expect(openSocket).toHaveBeenCalledWith("192.168.4.77", 51555);69  });7071  it("presents the token as hex and returns the shared SAS", async () => {72    const s = ackThen();73    const result = await dialPairingChannel(CODE, { openSocket: async () => s.sock });7475    expect(dec(s.sent[0])).toEqual({ type: "PEACH_HELLO", token: "deadbeef" });76    expect(result.ok && result.dialled.sas).toBe("895245");77    expect(result.ok && result.dialled.host).toBe("192.168.4.77");78  });7980  it("rejects an addressless legacy code before opening a socket", async () => {81    const openSocket = vi.fn(async () => ackThen().sock);8283    const result = await dialPairingChannel("123456", { openSocket });8485    expect(result).toEqual({ ok: false, error: "invalid_code" });86    expect(openSocket).not.toHaveBeenCalled();87  });8889  it("reads a silent close as invalid_code, not a transport failure", async () => {90    // A wrong or expired token gets no error frame — the listener just closes.91    const s = fakeSocket([null]);92    const result = await dialPairingChannel(CODE, { openSocket: async () => s.sock });9394    expect(result).toEqual({ ok: false, error: "invalid_code" });95    expect(s.isClosed()).toBe(true);96  });9798  it("treats a non-ACK first frame exactly like a wrong token", async () => {99    const s = fakeSocket([enc({ type: "SOMETHING_ELSE" })]);100    const result = await dialPairingChannel(CODE, { openSocket: async () => s.sock });101102    expect(result).toEqual({ ok: false, error: "invalid_code" });103  });104105  it("rejects an ACK whose ephemeral key is not 32-byte hex", async () => {106    const s = fakeSocket([enc({ type: "PEACH_HELLO_ACK", ephemeralPublicKeyHex: "nope" })]);107    const result = await dialPairingChannel(CODE, { openSocket: async () => s.sock });108109    expect(result).toEqual({ ok: false, error: "handshake_failed" });110    expect(s.isClosed()).toBe(true);111  });112113  it("reports no_host_found when the endpoint refuses the connection", async () => {114    const result = await dialPairingChannel(CODE, {115      openSocket: async () => { throw new Error("ECONNREFUSED"); },116    });117118    expect(result).toEqual({ ok: false, error: "no_host_found" });119  });120121  it("leaves the socket open on success so the caller owns its lifetime", async () => {122    const s = ackThen(enc({ hello: "world" }));123    const result = await dialPairingChannel(CODE, { openSocket: async () => s.sock });124    if (!result.ok) throw new Error("expected a dialled channel");125126    expect(s.isClosed()).toBe(false);127    await result.dialled.channel.send(enc({ type: "ANYTHING" }));128    // sent[0] is the plaintext hello; sent[1] went through the Noise channel.129    expect(dec(s.sent[1])).toEqual({ type: "ANYTHING" });130    expect(dec(await result.dialled.channel.receive())).toEqual({ hello: "world" });131132    result.dialled.socket.close();133    expect(s.isClosed()).toBe(true);134  });135});