prompt-hash/src/hooks/useContractSync.test.ts at main · ghzhost/prompt-hash · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import { describe, it, expect, vi } from "vitest";
import type { QueryClient } from "@tanstack/react-query";

vi.mock("./useSubscription", () => ({ useSubscription: vi.fn() }));
vi.mock("@/lib/stellar/browserConfig", () => ({
  browserStellarConfig: { promptHashContractId: "test-contract-id" },
}));

import { invalidateAllPromptQueries } from "./useContractSync";

function mockQueryClient() {
  return {
    invalidateQueries: vi.fn().mockResolvedValue(undefined),
  } as unknown as QueryClient;
}

const EXPECTED_KEYS = [
  ["marketplace-prompts"],
  ["created-prompts"],
  ["purchased-prompts"],
  ["saved-prompts"],
  ["prompt-access"],
];

describe("invalidateAllPromptQueries", () => {
  it("invalidates all prompt-related query keys", async () => {
    const queryClient = mockQueryClient();
    await invalidateAllPromptQueries(queryClient);

    expect(queryClient.invalidateQueries).toHaveBeenCalledTimes(5);
    for (const queryKey of EXPECTED_KEYS) {
      expect(queryClient.invalidateQueries).toHaveBeenCalledWith({ queryKey });
    }
  });

  it("includes marketplace-prompts so the browse grid refreshes after any TX", async () => {
    const queryClient = mockQueryClient();
    await invalidateAllPromptQueries(queryClient);

    expect(queryClient.invalidateQueries).toHaveBeenCalledWith({
      queryKey: ["marketplace-prompts"],
    });
  });

  it("includes created-prompts so creator sales counts refresh after a purchase", async () => {
    const queryClient = mockQueryClient();
    await invalidateAllPromptQueries(queryClient);

    expect(queryClient.invalidateQueries).toHaveBeenCalledWith({
      queryKey: ["created-prompts"],
    });
  });

  it("includes purchased-prompts so buyer library refreshes after buy", async () => {
    const queryClient = mockQueryClient();
    await invalidateAllPromptQueries(queryClient);

    expect(queryClient.invalidateQueries).toHaveBeenCalledWith({
      queryKey: ["purchased-prompts"],
    });
  });

  it("includes saved-prompts so buyer saved listings refresh after mutations", async () => {
    const queryClient = mockQueryClient();
    await invalidateAllPromptQueries(queryClient);

    expect(queryClient.invalidateQueries).toHaveBeenCalledWith({
      queryKey: ["saved-prompts"],
    });
  });

  it("includes prompt-access so access checks refresh after a purchase", async () => {
    const queryClient = mockQueryClient();
    await invalidateAllPromptQueries(queryClient);

    expect(queryClient.invalidateQueries).toHaveBeenCalledWith({
      queryKey: ["prompt-access"],
    });
  });

  it("awaits all invalidations in parallel before resolving", async () => {
    const settled: string[] = [];
    const queryClient = {
      invalidateQueries: vi.fn().mockImplementation(
        ({ queryKey }: { queryKey: string[] }) =>
          new Promise<void>((resolve) =>
            setTimeout(() => {
              settled.push(queryKey[0]);
              resolve();
            }, 5),
          ),
      ),
    } as unknown as QueryClient;

    await invalidateAllPromptQueries(queryClient);

    expect(settled).toHaveLength(5);
    expect(settled).toContain("marketplace-prompts");
    expect(settled).toContain("created-prompts");
    expect(settled).toContain("purchased-prompts");
    expect(settled).toContain("saved-prompts");
    expect(settled).toContain("prompt-access");
  });

  it("can be called multiple times without error", async () => {
    const queryClient = mockQueryClient();
    await invalidateAllPromptQueries(queryClient);
    await invalidateAllPromptQueries(queryClient);

    expect(queryClient.invalidateQueries).toHaveBeenCalledTimes(10);
  });
});