mirror of
https://github.com/paperclipai/paperclip.git
synced 2026-10-23 17:33:51 +02:00
Enforce private task and project access across direct reads, search, execution, files, plugins, live delivery, and sharing mutations. Preserve downward-only sharing, current responsible-user authorization, and audited emergency access. Bind historical draft assets with migration 0314. Co-Authored-By: Paperclip <noreply@paperclip.ing>
662 lines
23 KiB
TypeScript
662 lines
23 KiB
TypeScript
import { sql } from "drizzle-orm";
|
|
import express from "express";
|
|
import request from "supertest";
|
|
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
// Isolate runtime-management guards; real database privacy is covered by
|
|
// privacy-production-review.test.ts.
|
|
vi.mock("../services/authorization.js", async () => ({
|
|
...(await vi.importActual<typeof import("../services/authorization.js")>("../services/authorization.js")),
|
|
executionWorkspaceReadSqlCondition: async () => sql<boolean>`true`,
|
|
canActorReadExecutionWorkspace: async () => true,
|
|
}));
|
|
|
|
const mockProjectService = vi.hoisted(() => ({
|
|
create: vi.fn(),
|
|
createWorkspace: vi.fn(),
|
|
getById: vi.fn(),
|
|
listWorkspaces: vi.fn(),
|
|
resolveByReference: vi.fn(),
|
|
update: vi.fn(),
|
|
updateWorkspace: vi.fn(),
|
|
}));
|
|
|
|
const mockExecutionWorkspaceService = vi.hoisted(() => ({
|
|
getById: vi.fn(),
|
|
update: vi.fn(),
|
|
}));
|
|
|
|
const mockSecretService = vi.hoisted(() => ({
|
|
normalizeEnvBindingsForPersistence: vi.fn(),
|
|
}));
|
|
|
|
const mockEnvironmentService = vi.hoisted(() => ({
|
|
getById: vi.fn(),
|
|
}));
|
|
|
|
const mockWorkspaceOperationService = vi.hoisted(() => ({}));
|
|
const mockWorkspaceRuntimeLeaseService = vi.hoisted(() => ({
|
|
claim: vi.fn(async () => ({ outcome: "created", ownerKey: "issue:issue-1", lease: null, reclaimedFrom: null })),
|
|
release: vi.fn(async () => ({ released: false, ownerKey: null })),
|
|
get: vi.fn(async () => null),
|
|
}));
|
|
const mockHeartbeatService = vi.hoisted(() => ({}));
|
|
const mockLogActivity = vi.hoisted(() => vi.fn());
|
|
const mockGetTelemetryClient = vi.hoisted(() => vi.fn());
|
|
const mockAccessService = vi.hoisted(() => ({
|
|
decide: vi.fn(),
|
|
}));
|
|
const mockAssertCanManageProjectWorkspaceRuntimeServices = vi.hoisted(() => vi.fn());
|
|
const mockAssertCanManageExecutionWorkspaceRuntimeServices = vi.hoisted(() => vi.fn());
|
|
|
|
vi.mock("../telemetry.js", () => ({
|
|
getTelemetryClient: mockGetTelemetryClient,
|
|
}));
|
|
|
|
vi.mock("../services/index.js", () => ({
|
|
accessService: () => mockAccessService,
|
|
environmentService: () => mockEnvironmentService,
|
|
executionWorkspaceService: () => mockExecutionWorkspaceService,
|
|
heartbeatService: () => mockHeartbeatService,
|
|
logActivity: mockLogActivity,
|
|
projectService: () => mockProjectService,
|
|
secretService: () => mockSecretService,
|
|
workspaceOperationService: () => mockWorkspaceOperationService,
|
|
workspaceRuntimeLeaseService: () => mockWorkspaceRuntimeLeaseService,
|
|
LEASED_WORKSPACE_RUNTIME_ACTIONS: ["start", "stop", "restart", "repair"],
|
|
}));
|
|
|
|
vi.mock("../services/workspace-runtime.js", () => ({
|
|
cleanupExecutionWorkspaceArtifacts: vi.fn(),
|
|
startRuntimeServicesForWorkspaceControl: vi.fn(),
|
|
stopRuntimeServicesForExecutionWorkspace: vi.fn(),
|
|
stopRuntimeServicesForProjectWorkspace: vi.fn(),
|
|
}));
|
|
|
|
vi.mock("../routes/workspace-runtime-service-authz.js", () => ({
|
|
assertCanManageProjectWorkspaceRuntimeServices: mockAssertCanManageProjectWorkspaceRuntimeServices,
|
|
assertCanManageExecutionWorkspaceRuntimeServices: mockAssertCanManageExecutionWorkspaceRuntimeServices,
|
|
}));
|
|
|
|
function registerWorkspaceRouteMocks() {
|
|
vi.doMock("../telemetry.js", () => ({
|
|
getTelemetryClient: mockGetTelemetryClient,
|
|
}));
|
|
|
|
vi.doMock("../services/index.js", () => ({
|
|
accessService: () => mockAccessService,
|
|
environmentService: () => mockEnvironmentService,
|
|
executionWorkspaceService: () => mockExecutionWorkspaceService,
|
|
heartbeatService: () => mockHeartbeatService,
|
|
logActivity: mockLogActivity,
|
|
projectService: () => mockProjectService,
|
|
secretService: () => mockSecretService,
|
|
workspaceOperationService: () => mockWorkspaceOperationService,
|
|
workspaceRuntimeLeaseService: () => mockWorkspaceRuntimeLeaseService,
|
|
LEASED_WORKSPACE_RUNTIME_ACTIONS: ["start", "stop", "restart", "repair"],
|
|
}));
|
|
|
|
vi.doMock("../services/workspace-runtime.js", () => ({
|
|
cleanupExecutionWorkspaceArtifacts: vi.fn(),
|
|
startRuntimeServicesForWorkspaceControl: vi.fn(),
|
|
stopRuntimeServicesForExecutionWorkspace: vi.fn(),
|
|
stopRuntimeServicesForProjectWorkspace: vi.fn(),
|
|
}));
|
|
|
|
vi.doMock("../routes/workspace-runtime-service-authz.js", () => ({
|
|
assertCanManageProjectWorkspaceRuntimeServices: mockAssertCanManageProjectWorkspaceRuntimeServices,
|
|
assertCanManageExecutionWorkspaceRuntimeServices: mockAssertCanManageExecutionWorkspaceRuntimeServices,
|
|
}));
|
|
}
|
|
|
|
let appImportCounter = 0;
|
|
|
|
async function createProjectApp(actor: Record<string, unknown>) {
|
|
registerWorkspaceRouteMocks();
|
|
appImportCounter += 1;
|
|
const routeModulePath = `../routes/projects.js?workspace-runtime-routes-authz-${appImportCounter}`;
|
|
const middlewareModulePath = `../middleware/index.js?workspace-runtime-routes-authz-${appImportCounter}`;
|
|
const [{ projectRoutes }, { errorHandler }] = await Promise.all([
|
|
import(routeModulePath) as Promise<typeof import("../routes/projects.js")>,
|
|
import(middlewareModulePath) as Promise<typeof import("../middleware/index.js")>,
|
|
]);
|
|
const app = express();
|
|
app.use(express.json());
|
|
app.use((req, _res, next) => {
|
|
(req as any).actor = actor;
|
|
next();
|
|
});
|
|
app.use("/api", projectRoutes({} as any));
|
|
app.use(errorHandler);
|
|
return app;
|
|
}
|
|
|
|
async function createExecutionWorkspaceApp(actor: Record<string, unknown>) {
|
|
registerWorkspaceRouteMocks();
|
|
appImportCounter += 1;
|
|
const routeModulePath = `../routes/execution-workspaces.js?workspace-runtime-routes-authz-${appImportCounter}`;
|
|
const middlewareModulePath = `../middleware/index.js?workspace-runtime-routes-authz-${appImportCounter}`;
|
|
const [{ executionWorkspaceRoutes }, { errorHandler }] = await Promise.all([
|
|
import(routeModulePath) as Promise<typeof import("../routes/execution-workspaces.js")>,
|
|
import(middlewareModulePath) as Promise<typeof import("../middleware/index.js")>,
|
|
]);
|
|
const app = express();
|
|
app.use(express.json());
|
|
app.use((req, _res, next) => {
|
|
(req as any).actor = actor;
|
|
next();
|
|
});
|
|
app.use("/api", executionWorkspaceRoutes({} as any));
|
|
app.use(errorHandler);
|
|
return app;
|
|
}
|
|
|
|
function buildProject(overrides: Record<string, unknown> = {}) {
|
|
return {
|
|
id: "project-1",
|
|
companyId: "company-1",
|
|
urlKey: "project-1",
|
|
goalId: null,
|
|
goalIds: [],
|
|
goals: [],
|
|
name: "Project",
|
|
description: null,
|
|
status: "backlog",
|
|
leadAgentId: null,
|
|
targetDate: null,
|
|
color: null,
|
|
env: null,
|
|
pauseReason: null,
|
|
pausedAt: null,
|
|
executionWorkspacePolicy: null,
|
|
codebase: null,
|
|
workspaces: [],
|
|
primaryWorkspace: null,
|
|
archivedAt: null,
|
|
createdAt: new Date(),
|
|
updatedAt: new Date(),
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
function buildExecutionWorkspace(overrides: Record<string, unknown> = {}) {
|
|
return {
|
|
id: "workspace-1",
|
|
companyId: "company-1",
|
|
projectId: "project-1",
|
|
projectWorkspaceId: null,
|
|
sourceIssueId: null,
|
|
mode: "isolated_workspace",
|
|
strategyType: "git_worktree",
|
|
name: "Workspace",
|
|
status: "active",
|
|
cwd: "/tmp/workspace",
|
|
repoUrl: null,
|
|
baseRef: "main",
|
|
branchName: "feature/test",
|
|
providerType: "git_worktree",
|
|
providerRef: null,
|
|
derivedFromExecutionWorkspaceId: null,
|
|
lastUsedAt: new Date(),
|
|
openedAt: new Date(),
|
|
closedAt: null,
|
|
cleanupEligibleAt: null,
|
|
cleanupReason: null,
|
|
config: null,
|
|
metadata: null,
|
|
runtimeServices: [],
|
|
createdAt: new Date(),
|
|
updatedAt: new Date(),
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
describe("workspace runtime service route authorization", () => {
|
|
const projectId = "11111111-1111-4111-8111-111111111111";
|
|
const workspaceId = "22222222-2222-4222-8222-222222222222";
|
|
const executionWorkspaceId = "33333333-3333-4333-8333-333333333333";
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
mockAccessService.decide.mockResolvedValue({
|
|
allowed: true,
|
|
action: "company_scope:read",
|
|
reason: "allow_test",
|
|
explanation: "Allowed by test mock.",
|
|
});
|
|
mockEnvironmentService.getById.mockResolvedValue(null);
|
|
mockSecretService.normalizeEnvBindingsForPersistence.mockImplementation(async (_companyId, env) => env);
|
|
mockProjectService.resolveByReference.mockResolvedValue({ ambiguous: false, project: null });
|
|
mockProjectService.create.mockResolvedValue(buildProject());
|
|
mockProjectService.update.mockResolvedValue(buildProject());
|
|
mockProjectService.createWorkspace.mockResolvedValue({
|
|
id: workspaceId,
|
|
companyId: "company-1",
|
|
projectId,
|
|
name: "Workspace",
|
|
sourceType: "local_path",
|
|
cwd: "/tmp/project",
|
|
repoUrl: null,
|
|
repoRef: null,
|
|
defaultRef: null,
|
|
visibility: "default",
|
|
setupCommand: null,
|
|
cleanupCommand: null,
|
|
remoteProvider: null,
|
|
remoteWorkspaceRef: null,
|
|
sharedWorkspaceKey: null,
|
|
metadata: null,
|
|
runtimeConfig: null,
|
|
isPrimary: false,
|
|
runtimeServices: [],
|
|
createdAt: new Date(),
|
|
updatedAt: new Date(),
|
|
});
|
|
mockProjectService.listWorkspaces.mockResolvedValue([{
|
|
id: workspaceId,
|
|
companyId: "company-1",
|
|
projectId,
|
|
name: "Workspace",
|
|
sourceType: "local_path",
|
|
cwd: "/tmp/project",
|
|
repoUrl: null,
|
|
repoRef: null,
|
|
defaultRef: null,
|
|
visibility: "default",
|
|
setupCommand: null,
|
|
cleanupCommand: null,
|
|
remoteProvider: null,
|
|
remoteWorkspaceRef: null,
|
|
sharedWorkspaceKey: null,
|
|
metadata: null,
|
|
runtimeConfig: null,
|
|
isPrimary: false,
|
|
runtimeServices: [],
|
|
createdAt: new Date(),
|
|
updatedAt: new Date(),
|
|
}]);
|
|
mockProjectService.updateWorkspace.mockResolvedValue({
|
|
id: workspaceId,
|
|
companyId: "company-1",
|
|
projectId,
|
|
name: "Workspace",
|
|
sourceType: "local_path",
|
|
cwd: "/tmp/project",
|
|
repoUrl: null,
|
|
repoRef: null,
|
|
defaultRef: null,
|
|
visibility: "default",
|
|
setupCommand: null,
|
|
cleanupCommand: null,
|
|
remoteProvider: null,
|
|
remoteWorkspaceRef: null,
|
|
sharedWorkspaceKey: null,
|
|
metadata: null,
|
|
runtimeConfig: null,
|
|
isPrimary: false,
|
|
runtimeServices: [],
|
|
createdAt: new Date(),
|
|
updatedAt: new Date(),
|
|
});
|
|
mockExecutionWorkspaceService.update.mockResolvedValue(buildExecutionWorkspace());
|
|
mockAssertCanManageProjectWorkspaceRuntimeServices.mockResolvedValue(undefined);
|
|
mockAssertCanManageExecutionWorkspaceRuntimeServices.mockResolvedValue(undefined);
|
|
});
|
|
|
|
it("rejects agent callers for project workspace runtime service mutations when workspace auth denies access", async () => {
|
|
const { forbidden } = await import("../errors.js");
|
|
mockProjectService.getById.mockResolvedValue(buildProject({
|
|
id: projectId,
|
|
workspaces: [{
|
|
id: workspaceId,
|
|
companyId: "company-1",
|
|
projectId,
|
|
name: "Workspace",
|
|
sourceType: "local_path",
|
|
cwd: "/tmp/project",
|
|
repoUrl: null,
|
|
repoRef: null,
|
|
defaultRef: null,
|
|
visibility: "default",
|
|
setupCommand: null,
|
|
cleanupCommand: null,
|
|
remoteProvider: null,
|
|
remoteWorkspaceRef: null,
|
|
sharedWorkspaceKey: null,
|
|
metadata: null,
|
|
runtimeConfig: null,
|
|
isPrimary: false,
|
|
runtimeServices: [],
|
|
createdAt: new Date(),
|
|
updatedAt: new Date(),
|
|
}],
|
|
}));
|
|
mockAssertCanManageProjectWorkspaceRuntimeServices.mockRejectedValue(
|
|
forbidden("Missing permission to manage workspace runtime services"),
|
|
);
|
|
const app = await createProjectApp({
|
|
type: "agent",
|
|
agentId: "agent-1",
|
|
companyId: "company-1",
|
|
source: "agent_key",
|
|
runId: "run-1",
|
|
});
|
|
|
|
const res = await request(app)
|
|
.post(`/api/projects/${projectId}/workspaces/${workspaceId}/runtime-services/start`)
|
|
.send({});
|
|
|
|
expect(res.status).toBe(403);
|
|
expect(res.body.error).toContain("Missing permission");
|
|
expect(mockProjectService.getById).toHaveBeenCalledWith(projectId);
|
|
expect(mockAssertCanManageProjectWorkspaceRuntimeServices).toHaveBeenCalled();
|
|
}, 15000);
|
|
|
|
it.each(["skill_test", "task_bridge"])(
|
|
"rejects %s keys at the central runtime-manage gate for project and execution workspaces",
|
|
async (kind) => {
|
|
mockAccessService.decide.mockImplementation(async (input: { action?: string }) => ({
|
|
allowed: input.action !== "runtime:manage",
|
|
action: input.action,
|
|
reason: input.action === "runtime:manage" ? "deny_key_scope" : "allow_test",
|
|
explanation: input.action === "runtime:manage"
|
|
? "Restricted keys cannot manage workspace runtimes."
|
|
: "Allowed by test mock.",
|
|
}));
|
|
mockProjectService.getById.mockResolvedValue(buildProject({
|
|
id: projectId,
|
|
workspaces: [{
|
|
id: workspaceId,
|
|
companyId: "company-1",
|
|
projectId,
|
|
runtimeConfig: {
|
|
workspaceRuntime: { services: [{ name: "web", command: "pnpm dev" }] },
|
|
},
|
|
}],
|
|
}));
|
|
mockExecutionWorkspaceService.getById.mockResolvedValue(buildExecutionWorkspace({
|
|
id: executionWorkspaceId,
|
|
config: {
|
|
workspaceRuntime: { services: [{ name: "web", command: "pnpm dev" }] },
|
|
},
|
|
}));
|
|
const actor = {
|
|
type: "agent",
|
|
agentId: "agent-1",
|
|
companyId: "company-1",
|
|
source: "agent_key",
|
|
runId: "run-1",
|
|
keyScope: kind === "skill_test"
|
|
? { kind, issueId: "issue-1" }
|
|
: { kind, parentIssueId: "issue-1" },
|
|
};
|
|
|
|
const projectRes = await request(await createProjectApp(actor))
|
|
.post(`/api/projects/${projectId}/workspaces/${workspaceId}/runtime-services/start`)
|
|
.send({});
|
|
expect(projectRes.status, JSON.stringify(projectRes.body)).toBe(403);
|
|
expect(projectRes.body.error).toContain("authorization boundary");
|
|
|
|
const executionRes = await request(await createExecutionWorkspaceApp(actor))
|
|
.post(`/api/execution-workspaces/${executionWorkspaceId}/runtime-services/start`)
|
|
.send({});
|
|
expect(executionRes.status, JSON.stringify(executionRes.body)).toBe(403);
|
|
expect(executionRes.body.error).toContain("authorization boundary");
|
|
|
|
expect(mockAssertCanManageProjectWorkspaceRuntimeServices).not.toHaveBeenCalled();
|
|
expect(mockAssertCanManageExecutionWorkspaceRuntimeServices).not.toHaveBeenCalled();
|
|
expect(mockAccessService.decide).toHaveBeenCalledWith(expect.objectContaining({
|
|
action: "runtime:manage",
|
|
resource: { type: "company", companyId: "company-1" },
|
|
}));
|
|
},
|
|
);
|
|
|
|
it("blocks shared-project stop/restart requests from agents", async () => {
|
|
mockProjectService.getById.mockResolvedValue(buildProject({
|
|
id: projectId,
|
|
workspaces: [{
|
|
id: workspaceId,
|
|
companyId: "company-1",
|
|
projectId,
|
|
name: "Workspace",
|
|
sourceType: "local_path",
|
|
cwd: "/tmp/project",
|
|
repoUrl: null,
|
|
repoRef: null,
|
|
defaultRef: null,
|
|
visibility: "default",
|
|
setupCommand: null,
|
|
cleanupCommand: null,
|
|
remoteProvider: null,
|
|
remoteWorkspaceRef: null,
|
|
sharedWorkspaceKey: "shared-key",
|
|
metadata: null,
|
|
runtimeConfig: null,
|
|
isPrimary: false,
|
|
runtimeServices: [],
|
|
createdAt: new Date(),
|
|
updatedAt: new Date(),
|
|
}],
|
|
}));
|
|
const app = await createProjectApp({
|
|
type: "agent",
|
|
agentId: "agent-1",
|
|
companyId: "company-1",
|
|
source: "agent_key",
|
|
runId: "run-1",
|
|
});
|
|
|
|
const responses = await Promise.all([
|
|
request(app).post(`/api/projects/${projectId}/workspaces/${workspaceId}/runtime-services/stop`).send({}),
|
|
request(app).post(`/api/projects/${projectId}/workspaces/${workspaceId}/runtime-services/restart`).send({}),
|
|
]);
|
|
|
|
for (const res of responses) {
|
|
expect(res.status).toBe(403);
|
|
expect(res.body.error).toContain("Missing permission");
|
|
expect(mockProjectService.getById).toHaveBeenCalledWith(projectId);
|
|
expect(mockAssertCanManageProjectWorkspaceRuntimeServices).not.toHaveBeenCalled();
|
|
}
|
|
|
|
}, 15000);
|
|
|
|
it("rejects agent callers that create project execution workspace commands", async () => {
|
|
const app = await createProjectApp({
|
|
type: "agent",
|
|
agentId: "agent-1",
|
|
companyId: "company-1",
|
|
source: "agent_key",
|
|
runId: "run-1",
|
|
});
|
|
|
|
const res = await request(app)
|
|
.post("/api/companies/company-1/projects")
|
|
.send({
|
|
name: "Exploit",
|
|
executionWorkspacePolicy: {
|
|
enabled: true,
|
|
workspaceStrategy: {
|
|
type: "git_worktree",
|
|
provisionCommand: "touch /tmp/paperclip-rce",
|
|
runtimeProvisionCommand: "touch /tmp/paperclip-runtime-rce",
|
|
},
|
|
},
|
|
});
|
|
|
|
expect(res.status).toBe(403);
|
|
expect(res.body.error).toContain("host-executed workspace commands");
|
|
expect(mockProjectService.create).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("rejects agent callers that persist workspace-runtime service commands", async () => {
|
|
mockProjectService.getById.mockResolvedValue(buildProject());
|
|
const app = await createProjectApp({
|
|
type: "agent",
|
|
agentId: "agent-1",
|
|
companyId: "company-1",
|
|
source: "agent_key",
|
|
runId: "run-1",
|
|
});
|
|
|
|
const projectRes = await request(app)
|
|
.post("/api/companies/company-1/projects")
|
|
.send({
|
|
name: "Exploit",
|
|
executionWorkspacePolicy: {
|
|
enabled: true,
|
|
workspaceRuntime: {
|
|
services: [{ name: "web", command: "touch /tmp/paperclip-rce" }],
|
|
},
|
|
},
|
|
});
|
|
expect(projectRes.status).toBe(403);
|
|
expect(projectRes.body.error).toContain("executionWorkspacePolicy.workspaceRuntime.services[0].command");
|
|
|
|
const workspaceRes = await request(app)
|
|
.patch(`/api/projects/${projectId}/workspaces/${workspaceId}`)
|
|
.send({
|
|
runtimeConfig: {
|
|
workspaceRuntime: {
|
|
jobs: [{ name: "build", command: "touch /tmp/paperclip-rce" }],
|
|
},
|
|
},
|
|
});
|
|
expect(workspaceRes.status).toBe(403);
|
|
expect(workspaceRes.body.error).toContain("runtimeConfig.workspaceRuntime.jobs[0].command");
|
|
expect(mockProjectService.create).not.toHaveBeenCalled();
|
|
expect(mockProjectService.updateWorkspace).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("rejects agent callers that update project workspace cleanup commands", async () => {
|
|
mockProjectService.getById.mockResolvedValue(buildProject());
|
|
const app = await createProjectApp({
|
|
type: "agent",
|
|
agentId: "agent-1",
|
|
companyId: "company-1",
|
|
source: "agent_key",
|
|
runId: "run-1",
|
|
});
|
|
|
|
const res = await request(app)
|
|
.patch(`/api/projects/${projectId}/workspaces/${workspaceId}`)
|
|
.send({
|
|
cleanupCommand: "rm -rf /tmp/paperclip-rce",
|
|
});
|
|
|
|
expect(res.status).toBe(403);
|
|
expect(res.body.error).toContain("host-executed workspace commands");
|
|
expect(mockProjectService.updateWorkspace).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("allows board callers through the project workspace runtime auth gate", async () => {
|
|
mockProjectService.getById.mockResolvedValue(null);
|
|
const app = await createProjectApp({
|
|
type: "board",
|
|
userId: "board-1",
|
|
companyIds: ["company-1"],
|
|
source: "session",
|
|
isInstanceAdmin: false,
|
|
});
|
|
|
|
const res = await request(app)
|
|
.post(`/api/projects/${projectId}/workspaces/${workspaceId}/runtime-services/start`)
|
|
.send({});
|
|
|
|
expect(res.status).toBe(404);
|
|
expect(res.body.error).toContain("Project not found");
|
|
expect(mockProjectService.getById).toHaveBeenCalledWith(projectId);
|
|
});
|
|
|
|
it("rejects agent callers for execution workspace runtime service mutations when workspace auth denies access", async () => {
|
|
const { forbidden } = await import("../errors.js");
|
|
mockExecutionWorkspaceService.getById.mockResolvedValue(buildExecutionWorkspace({ id: executionWorkspaceId }));
|
|
mockAssertCanManageExecutionWorkspaceRuntimeServices.mockRejectedValue(
|
|
forbidden("Missing permission to manage workspace runtime services"),
|
|
);
|
|
const app = await createExecutionWorkspaceApp({
|
|
type: "agent",
|
|
agentId: "agent-1",
|
|
companyId: "company-1",
|
|
source: "agent_key",
|
|
runId: "run-1",
|
|
});
|
|
|
|
const res = await request(app)
|
|
.post(`/api/execution-workspaces/${executionWorkspaceId}/runtime-services/restart`)
|
|
.send({});
|
|
|
|
expect(res.status).toBe(403);
|
|
expect(res.body.error).toContain("Missing permission");
|
|
expect(mockExecutionWorkspaceService.getById).toHaveBeenCalledWith(executionWorkspaceId);
|
|
expect(mockAssertCanManageExecutionWorkspaceRuntimeServices).toHaveBeenCalled();
|
|
}, 15000);
|
|
|
|
it("rejects agent callers that patch execution workspace command config", async () => {
|
|
mockExecutionWorkspaceService.getById.mockResolvedValue(buildExecutionWorkspace({ id: executionWorkspaceId }));
|
|
const app = await createExecutionWorkspaceApp({
|
|
type: "agent",
|
|
agentId: "agent-1",
|
|
companyId: "company-1",
|
|
source: "agent_key",
|
|
runId: "run-1",
|
|
});
|
|
|
|
const res = await request(app)
|
|
.patch(`/api/execution-workspaces/${executionWorkspaceId}`)
|
|
.send({
|
|
config: {
|
|
cleanupCommand: "rm -rf /tmp/paperclip-rce",
|
|
},
|
|
});
|
|
|
|
expect(res.status).toBe(403);
|
|
expect(res.body.error).toContain("host-executed workspace commands");
|
|
expect(mockExecutionWorkspaceService.update).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("rejects agent callers that smuggle execution workspace commands through metadata.config", async () => {
|
|
mockExecutionWorkspaceService.getById.mockResolvedValue(buildExecutionWorkspace({ id: executionWorkspaceId }));
|
|
const app = await createExecutionWorkspaceApp({
|
|
type: "agent",
|
|
agentId: "agent-1",
|
|
companyId: "company-1",
|
|
source: "agent_key",
|
|
runId: "run-1",
|
|
});
|
|
|
|
const res = await request(app)
|
|
.patch(`/api/execution-workspaces/${executionWorkspaceId}`)
|
|
.send({
|
|
metadata: {
|
|
config: {
|
|
provisionCommand: "touch /tmp/paperclip-rce",
|
|
},
|
|
},
|
|
});
|
|
|
|
expect(res.status).toBe(403);
|
|
expect(res.body.error).toContain("host-executed workspace commands");
|
|
expect(mockExecutionWorkspaceService.update).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("allows board callers through the execution workspace runtime auth gate", async () => {
|
|
mockExecutionWorkspaceService.getById.mockResolvedValue(null);
|
|
const app = await createExecutionWorkspaceApp({
|
|
type: "board",
|
|
userId: "board-1",
|
|
companyIds: ["company-1"],
|
|
source: "session",
|
|
isInstanceAdmin: false,
|
|
});
|
|
|
|
const res = await request(app)
|
|
.post(`/api/execution-workspaces/${executionWorkspaceId}/runtime-services/restart`)
|
|
.send({});
|
|
|
|
expect(res.status).toBe(404);
|
|
expect(res.body.error).toContain("Execution workspace not found");
|
|
expect(mockExecutionWorkspaceService.getById).toHaveBeenCalledWith(executionWorkspaceId);
|
|
});
|
|
});
|