@sundaeswap/sprinkles 0.1.1

package/src/Sprinkle/__tests__/tx-dialog.test.ts

@@ -0,0 +1,293 @@
+import { describe, test, expect, mock, beforeEach } from "bun:test";
+import { Type } from "@sinclair/typebox";
+
+// Track select calls
+let selectCalls: any[] = [];
+let selectResponses: any[] = [];
+
+// Mock @inquirer/prompts
+mock.module("@inquirer/prompts", () => ({
+  input: mock(async () => "test"),
+  password: mock(async () => "secret"),
+  select: mock(async (opts: any) => {
+    selectCalls.push(opts);
+    const response = selectResponses.shift();
+    if (response === undefined) {
+      // Default: return -1 (Back/Exit)
+      return -1;
+    }
+    return response;
+  }),
+  search: mock(async () => "result"),
+}));
+
+// Mock clipboardy
+let clipboardContent = "";
+let clipboardShouldFail = false;
+mock.module("clipboardy", () => ({
+  default: {
+    writeSync: (content: string) => {
+      if (clipboardShouldFail) throw new Error("clipboard not available");
+      clipboardContent = content;
+    },
+  },
+}));
+
+const { Sprinkle } = await import("../../index.js");
+
+const TestSchema = Type.Object({
+  name: Type.String(),
+});
+
+describe("TxDialog (2.6)", () => {
+  beforeEach(() => {
+    selectCalls = [];
+    selectResponses = [];
+    clipboardContent = "";
+    clipboardShouldFail = false;
+  });
+
+  test("shows truncated CBOR by default", async () => {
+    const sprinkle = new Sprinkle(TestSchema, "/tmp/test-txdialog");
+    const logs: string[] = [];
+    const origLog = console.log;
+    console.log = (...args: any[]) => logs.push(args.join(" "));
+
+    // Just exit immediately
+    selectResponses = [-1];
+
+    const mockTx = {
+      toCbor: () => "a".repeat(100),
+      body: () => ({
+        requiredSigners: () => null,
+        certs: () => null,
+        withdrawals: () => null,
+      }),
+    };
+
+    const mockBlaze = {
+      wallet: { getUsedAddresses: async () => [] },
+    };
+
+    await sprinkle.TxDialog(mockBlaze as any, mockTx as any);
+    console.log = origLog;
+
+    // Should show truncated CBOR (50 chars + ...)
+    const cborLog = logs.find(l => l.includes("Transaction CBOR:"));
+    expect(cborLog).toBeDefined();
+    expect(cborLog!.length).toBeLessThan(100);
+    expect(cborLog).toContain("...");
+  });
+
+  test("menu items for non-HotWallet include Expand, Copy, no Sign", async () => {
+    const sprinkle = new Sprinkle(TestSchema, "/tmp/test-txdialog2");
+    const origLog = console.log;
+    console.log = () => {};
+
+    selectResponses = [-1];
+
+    const mockTx = {
+      toCbor: () => "deadbeef",
+      body: () => ({
+        requiredSigners: () => null,
+        certs: () => null,
+        withdrawals: () => null,
+      }),
+    };
+
+    const mockBlaze = {
+      wallet: { getUsedAddresses: async () => [] },
+    };
+
+    await sprinkle.TxDialog(mockBlaze as any, mockTx as any);
+    console.log = origLog;
+
+    // Check the select call had Expand and Copy but not Sign
+    const menuCall = selectCalls[0];
+    expect(menuCall).toBeDefined();
+    const choiceNames = menuCall.choices.map((c: any) => c.name);
+    expect(choiceNames).toContain("Expand CBOR");
+    expect(choiceNames).toContain("Copy CBOR to clipboard");
+    expect(choiceNames).not.toContain("Sign and submit transaction");
+  });
+
+  test("menu items for HotWallet include Sign", async () => {
+    const sprinkle = new Sprinkle(TestSchema, "/tmp/test-txdialog3");
+    const origLog = console.log;
+    console.log = () => {};
+
+    // Need to import HotWallet to use instanceof
+    const { HotWallet } = await import("@blaze-cardano/sdk");
+
+    selectResponses = [-1];
+
+    const mockTx = {
+      toCbor: () => "deadbeef",
+      body: () => ({
+        requiredSigners: () => null,
+        certs: () => null,
+        withdrawals: () => null,
+      }),
+    };
+
+    // Create a mock that passes instanceof HotWallet check
+    // We need to use Object.create to set the prototype
+    const hotWalletProto = HotWallet.prototype;
+    const mockWallet = Object.create(hotWalletProto);
+    mockWallet.getUsedAddresses = async () => [];
+
+    const mockBlaze = {
+      wallet: mockWallet,
+    };
+
+    await sprinkle.TxDialog(mockBlaze as any, mockTx as any);
+    console.log = origLog;
+
+    const menuCall = selectCalls[0];
+    expect(menuCall).toBeDefined();
+    const choiceNames = menuCall.choices.map((c: any) => c.name);
+    expect(choiceNames).toContain("Sign and submit transaction");
+  });
+
+  test("copy CBOR to clipboard works", async () => {
+    const sprinkle = new Sprinkle(TestSchema, "/tmp/test-txdialog4");
+    const origLog = console.log;
+    const logs: string[] = [];
+    console.log = (...args: any[]) => logs.push(args.join(" "));
+
+    // Select "Copy CBOR to clipboard" (index 1), then Back (-1)
+    selectResponses = [1, -1];
+
+    const txCbor = "deadbeefcafebabe";
+    const mockTx = {
+      toCbor: () => txCbor,
+      body: () => ({
+        requiredSigners: () => null,
+        certs: () => null,
+        withdrawals: () => null,
+      }),
+    };
+
+    const mockBlaze = {
+      wallet: { getUsedAddresses: async () => [] },
+    };
+
+    await sprinkle.TxDialog(mockBlaze as any, mockTx as any);
+    console.log = origLog;
+
+    expect(clipboardContent).toBe(txCbor);
+    expect(logs.some(l => l.includes("copied to clipboard"))).toBe(true);
+  });
+
+  test("clipboard failure falls back to expanded view", async () => {
+    const sprinkle = new Sprinkle(TestSchema, "/tmp/test-txdialog5");
+    const origLog = console.log;
+    const logs: string[] = [];
+    console.log = (...args: any[]) => logs.push(args.join(" "));
+
+    clipboardShouldFail = true;
+
+    // Select "Copy CBOR to clipboard" (index 1), 
+    // It will fail and call showDialog again with expanded=true
+    // Then Back (-1) from the expanded menu, then Back from outer
+    selectResponses = [1, -1, -1];
+
+    const txCbor = "a".repeat(100);
+    const mockTx = {
+      toCbor: () => txCbor,
+      body: () => ({
+        requiredSigners: () => null,
+        certs: () => null,
+        withdrawals: () => null,
+      }),
+    };
+
+    const mockBlaze = {
+      wallet: { getUsedAddresses: async () => [] },
+    };
+
+    await sprinkle.TxDialog(mockBlaze as any, mockTx as any);
+    console.log = origLog;
+
+    expect(logs.some(l => l.includes("Failed to copy"))).toBe(true);
+    // After failure it should show the full CBOR without truncation
+    expect(logs.some(l => l.includes(txCbor) && !l.includes("..."))).toBe(true);
+  });
+
+  test("expand CBOR shows full content", async () => {
+    const sprinkle = new Sprinkle(TestSchema, "/tmp/test-txdialog6");
+    const origLog = console.log;
+    const logs: string[] = [];
+    console.log = (...args: any[]) => logs.push(args.join(" "));
+
+    // Select "Expand CBOR" (index 0), then Back (-1) from expanded, then Back from outer
+    selectResponses = [0, -1, -1];
+
+    const txCbor = "b".repeat(100);
+    const mockTx = {
+      toCbor: () => txCbor,
+      body: () => ({
+        requiredSigners: () => null,
+        certs: () => null,
+        withdrawals: () => null,
+      }),
+    };
+
+    const mockBlaze = {
+      wallet: { getUsedAddresses: async () => [] },
+    };
+
+    await sprinkle.TxDialog(mockBlaze as any, mockTx as any);
+    console.log = origLog;
+
+    // First call shows truncated, second shows full
+    const truncatedLog = logs.find(l => l.includes("...") && l.includes("Transaction CBOR:"));
+    const fullLog = logs.find(l => l.includes(txCbor) && !l.includes("..."));
+    expect(truncatedLog).toBeDefined();
+    expect(fullLog).toBeDefined();
+  });
+
+  test("beforeSign hook is called before signing", async () => {
+    const sprinkle = new Sprinkle(TestSchema, "/tmp/test-txdialog7");
+    const origLog = console.log;
+    const origWarn = console.warn;
+    console.log = () => {};
+    console.warn = () => {};
+
+    const { HotWallet } = await import("@blaze-cardano/sdk");
+
+    let beforeSignCalled = false;
+
+    // Select "Sign and submit" (index 2 - after Expand and Copy), then Back
+    selectResponses = [2, -1];
+
+    const mockTx = {
+      toCbor: () => "deadbeef",
+      body: () => ({
+        requiredSigners: () => null,
+        certs: () => null,
+        withdrawals: () => null,
+      }),
+    };
+
+    const hotWalletProto = HotWallet.prototype;
+    const mockWallet = Object.create(hotWalletProto);
+    mockWallet.getUsedAddresses = async () => [];
+
+    const mockBlaze = {
+      wallet: mockWallet,
+      signTransaction: async (tx: any) => tx,
+      submitTransaction: async () => "txhash123",
+    };
+
+    await sprinkle.TxDialog(mockBlaze as any, mockTx as any, {
+      beforeSign: async () => {
+        beforeSignCalled = true;
+      },
+    });
+    console.log = origLog;
+    console.warn = origWarn;
+
+    expect(beforeSignCalled).toBe(true);
+  });
+});