fix(gateway): filter projects by ownership — close data privacy leak (#202)
Some checks failed
ci/woodpecker/push/ci Pipeline failed
Some checks failed
ci/woodpecker/push/ci Pipeline failed
Co-authored-by: Jason Woltje <jason@diversecanvas.com> Co-committed-by: Jason Woltje <jason@diversecanvas.com>
This commit was merged in pull request #202.
This commit is contained in:
82
packages/brain/src/projects.spec.ts
Normal file
82
packages/brain/src/projects.spec.ts
Normal file
@@ -0,0 +1,82 @@
|
||||
import { describe, it, expect, vi } from 'vitest';
|
||||
import { createProjectsRepo } from './projects.js';
|
||||
|
||||
/**
|
||||
* Build a minimal Drizzle mock. Each call to db.select() returns a fresh
|
||||
* chain that resolves `where()` to the provided rows for that call.
|
||||
*
|
||||
* `calls` is an ordered list: the first item is returned for the first
|
||||
* db.select() call, the second for the second, and so on.
|
||||
*/
|
||||
function makeDb(calls: unknown[][]) {
|
||||
let callIndex = 0;
|
||||
const selectSpy = vi.fn(() => {
|
||||
const rows = calls[callIndex++] ?? [];
|
||||
const chain = {
|
||||
where: vi.fn().mockResolvedValue(rows),
|
||||
} as { where: ReturnType<typeof vi.fn>; from?: ReturnType<typeof vi.fn> };
|
||||
// from() returns the chain so .where() can be chained, but also resolves
|
||||
// directly (as a thenable) for queries with no .where() call.
|
||||
chain.from = vi.fn(() => Object.assign(Promise.resolve(rows), chain));
|
||||
return chain;
|
||||
});
|
||||
return { select: selectSpy };
|
||||
}
|
||||
|
||||
describe('createProjectsRepo — findAllForUser', () => {
|
||||
it('filters by userId when user has no team memberships', async () => {
|
||||
// First select: teamMembers query → empty
|
||||
// Second select: projects query → one owned project
|
||||
const db = makeDb([
|
||||
[], // teamMembers rows
|
||||
[{ id: 'p1', ownerId: 'user-1', teamId: null, ownerType: 'user' }],
|
||||
]);
|
||||
const repo = createProjectsRepo(db as never);
|
||||
|
||||
const result = await repo.findAllForUser('user-1');
|
||||
|
||||
expect(db.select).toHaveBeenCalledTimes(2);
|
||||
expect(result).toHaveLength(1);
|
||||
expect(result[0]?.id).toBe('p1');
|
||||
});
|
||||
|
||||
it('includes team projects when user is a team member', async () => {
|
||||
// First select: teamMembers → user belongs to one team
|
||||
// Second select: projects query → two projects (own + team)
|
||||
const db = makeDb([
|
||||
[{ teamId: 'team-1' }],
|
||||
[
|
||||
{ id: 'p1', ownerId: 'user-1', teamId: null, ownerType: 'user' },
|
||||
{ id: 'p2', ownerId: null, teamId: 'team-1', ownerType: 'team' },
|
||||
],
|
||||
]);
|
||||
const repo = createProjectsRepo(db as never);
|
||||
|
||||
const result = await repo.findAllForUser('user-1');
|
||||
|
||||
expect(db.select).toHaveBeenCalledTimes(2);
|
||||
expect(result).toHaveLength(2);
|
||||
});
|
||||
|
||||
it('returns empty array when user has no projects and no teams', async () => {
|
||||
const db = makeDb([[], []]);
|
||||
const repo = createProjectsRepo(db as never);
|
||||
|
||||
const result = await repo.findAllForUser('user-no-projects');
|
||||
expect(result).toHaveLength(0);
|
||||
});
|
||||
});
|
||||
|
||||
describe('createProjectsRepo — findAll', () => {
|
||||
it('returns all rows without any user filter', async () => {
|
||||
const rows = [
|
||||
{ id: 'p1', ownerId: 'user-1', teamId: null, ownerType: 'user' },
|
||||
{ id: 'p2', ownerId: 'user-2', teamId: null, ownerType: 'user' },
|
||||
];
|
||||
const db = makeDb([rows]);
|
||||
const repo = createProjectsRepo(db as never);
|
||||
|
||||
const result = await repo.findAll();
|
||||
expect(result).toHaveLength(2);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user