Files
colbymchenry--codegraph/__tests__/pinia-store-synthesizer.test.ts
T
wehub-resource-sync 5f98960d22
Deploy site to GitHub Pages / build (push) Failing after 1s
Deploy site to GitHub Pages / deploy (push) Has been skipped
chore: import upstream snapshot with attribution
2026-07-13 12:02:56 +08:00

109 lines
3.8 KiB
TypeScript

/**
* Pinia `useStore().action()` dispatch bridge.
*
* A Pinia store factory `export const useXStore = defineStore(...)` exposes its
* actions as methods on the store instance; a consumer does `const s = useXStore()`
* then `s.action()`. That method-on-instance call has no static edge to the action
* (which lives in the store module). This bridges consumer → action by binding the
* store var to its factory's file and resolving `s.method()` to a function node IN
* THAT FILE — so it covers both the options and setup store forms, stays precise
* (a Pinia built-in like `$patch`, or an unrelated same-named method, resolves to
* nothing), and fires only when a `defineStore` factory actually exists.
*/
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
import * as fs from 'node:fs';
import * as path from 'node:path';
import * as os from 'node:os';
import { CodeGraph } from '../src';
describe('pinia-store synthesizer', () => {
let dir: string;
beforeEach(() => { dir = fs.mkdtempSync(path.join(os.tmpdir(), 'pinia-store-')); });
afterEach(() => { fs.rmSync(dir, { recursive: true, force: true }); });
it('bridges `const s = useXStore(); s.action()` to the action, across options + setup forms', async () => {
// Options-form store.
fs.writeFileSync(
path.join(dir, 'authStore.ts'),
`import { defineStore } from 'pinia';
export const useAuthStore = defineStore({
id: 'auth',
state: () => ({ token: '' }),
actions: {
async getMenu() { return loadMenu(); },
setToken(t: string) { this.token = t; },
},
});
`
);
// Setup-form store.
fs.writeFileSync(
path.join(dir, 'chatStore.ts'),
`import { defineStore } from 'pinia';
export const useChatStore = defineStore('chat', () => {
const getList = async () => { return fetchList(); };
return { getList };
});
`
);
// Consumer binds both stores and calls their actions (plus a Pinia built-in).
fs.writeFileSync(
path.join(dir, 'init.ts'),
`import { useAuthStore } from './authStore';
import { useChatStore } from './chatStore';
export function init() {
const authStore = useAuthStore();
const chatStore = useChatStore();
authStore.getMenu();
authStore.setToken('x');
authStore.$patch({}); // Pinia built-in — must not bridge
chatStore.getList();
}
`
);
const cg = await CodeGraph.init(dir, { silent: true });
await cg.indexAll();
const db = (cg as any).db.db;
const edges = db
.prepare(
`SELECT s.name source, t.name target, t.file_path tf
FROM edges e JOIN nodes s ON s.id = e.source JOIN nodes t ON t.id = e.target
WHERE json_extract(e.metadata,'$.synthesizedBy') = 'pinia-store'`
)
.all();
const pairs = edges.map((r: any) => `${r.source}->${r.target}`).sort();
// Exactly the three real actions, all from `init`.
expect(pairs).toEqual(['init->getList', 'init->getMenu', 'init->setToken']);
// Each target is the action in its own store file (cross-file, store-scoped).
expect(edges.every((r: any) => /Store\.ts$/.test(r.tf))).toBe(true);
// The Pinia built-in `$patch` produced no edge.
expect(pairs.some((p: string) => p.includes('patch'))).toBe(false);
cg.close?.();
});
it('produces nothing when there is no defineStore factory (not a Pinia store)', async () => {
fs.writeFileSync(
path.join(dir, 'thing.ts'),
`function useThing() { return { run() { return 1; } }; }
export function go() {
const thing = useThing();
thing.run();
}
`
);
const cg = await CodeGraph.init(dir, { silent: true });
await cg.indexAll();
const db = (cg as any).db.db;
const c = db
.prepare(`SELECT count(*) c FROM edges WHERE json_extract(metadata,'$.synthesizedBy') = 'pinia-store'`)
.get().c;
expect(c).toBe(0);
cg.close?.();
});
});