Files
Agent-System/internal/server/project_members_authz_test.go
T
weichenandCursor 3c2c7f9ccc test+arch: unblock production gate (import SSOT + isolated test DSN)
arch-check --mode block failed on frozen dev (23 NEW import findings):
modules materialindex/mineru/workflowscope and edges shipped via their
own accepted tasks were never registered in module-boundaries.yaml.
Register the intentional allowlist edges (precedent 97b868a9); no new
exceptions are added.

7 PG tests fell back to a hardcoded dev DB (127.0.0.1:5433) when
TEST_DATABASE_URL/CONTEXT_DB_DSN were unset, contradicting the isolated
test-DB contract (BUG-08) and polluting dev on a bare 'go test -short'.
Change the fallback to skip when no isolated DB env is provided.

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-09-22 18:04:02 +08:00

110 lines
2.9 KiB
Go

package server_test
import (
"context"
"net/http"
"net/http/httptest"
"os"
"testing"
"time"
"github.com/jackc/pgx/v5/pgxpool"
"chat-agent-system/internal/authz"
"chat-agent-system/internal/config"
"chat-agent-system/internal/server"
)
func openPGPool(t *testing.T) *pgxpool.Pool {
t.Helper()
dsn := os.Getenv("TEST_DATABASE_URL")
if dsn == "" {
t.Skip("TEST_DATABASE_URL must be set for PostgreSQL integration tests (refusing default dev database)")
return nil
}
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
pool, err := pgxpool.New(ctx, dsn)
if err != nil {
t.Skipf("postgres unavailable: %v", err)
return nil
}
if err := pool.Ping(ctx); err != nil {
pool.Close()
t.Skipf("postgres ping failed: %v", err)
}
return pool
}
func TestProjectMembers_CrossProjectForbidden(t *testing.T) {
pool := openPGPool(t)
if pool == nil {
return
}
defer pool.Close()
ctx := context.Background()
var projectA, projectB int64
err := pool.QueryRow(ctx, `SELECT id FROM projects ORDER BY id ASC LIMIT 1`).Scan(&projectA)
if err != nil || projectA <= 0 {
t.Skip("no projects in database for cross-project test")
}
err = pool.QueryRow(ctx, `SELECT id FROM projects WHERE id <> $1 ORDER BY id ASC LIMIT 1`, projectA).Scan(&projectB)
if err != nil || projectB <= 0 {
t.Skip("need at least two projects for cross-project test")
}
var outsiderID int64
err = pool.QueryRow(ctx, `
SELECT u.id FROM users u
WHERE NOT EXISTS (SELECT 1 FROM project_members pm WHERE pm.project_id = $1 AND pm.user_id = u.id AND pm.status = 'active')
AND NOT EXISTS (SELECT 1 FROM projects p WHERE p.id = $1 AND p.creator_user_id = u.id)
ORDER BY u.id ASC LIMIT 1
`, projectA).Scan(&outsiderID)
if err != nil || outsiderID <= 0 {
t.Skip("no outsider user available for cross-project test")
}
authzSvc := authz.NewAuthorizer(pool)
if err := authzSvc.RequireProjectAccess(ctx, projectA, outsiderID); err != authz.ErrForbidden {
t.Fatalf("expected outsider forbidden on project A, got %v", err)
}
}
func TestProjectMembers_ManageMembersPermissionKeyInCatalog(t *testing.T) {
found := false
for _, p := range authz.PermissionCatalog() {
if p.Key == authz.PermManageProjectMembers {
found = true
break
}
}
if !found {
t.Fatalf("expected %s in system permission catalog", authz.PermManageProjectMembers)
}
}
func TestProjectMembers_AnonymousReturns401(t *testing.T) {
cfg := &config.Config{
Env: "dev",
DataDir: t.TempDir(),
AgentBackend: "stub",
AllowInMemory: true,
AdminUsername: "admin",
AdminPassword: "admin123",
}
srv := server.NewServer(cfg, nil)
cfg.AllowInMemory = false
ts := httptest.NewServer(srv.Routes())
defer ts.Close()
resp, err := http.Get(ts.URL + "/api/v1/projects/1/members")
if err != nil {
t.Fatalf("request: %v", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusUnauthorized {
t.Fatalf("expected 401, got %d", resp.StatusCode)
}
}