Changeset 0.16.0 (#74)

server/database/testhelper.go

@@ -165,6 +165,9 @@ func TruncateAll(t *testing.T) {
 	}
 	// Order matters due to foreign keys — truncate with CASCADE
 	tables := []string{
+		"resource_grants",
+		"group_members",
+		"groups",
 		"usage_log",
 		"model_pricing",
 		"notes",
@@ -256,6 +259,64 @@ func SeedTestChannel(t *testing.T, userID, title string) string {
 
 // ── Helpers ─────────────────────────────────
 
+// SeedTestTeam creates a test team and returns the team ID.
+func SeedTestTeam(t *testing.T, name, createdBy string) string {
+	t.Helper()
+	var id string
+	err := DB.QueryRow(`
+		INSERT INTO teams (name, created_by)
+		VALUES ($1, $2)
+		RETURNING id
+	`, name, createdBy).Scan(&id)
+	if err != nil {
+		t.Fatalf("SeedTestTeam: %v", err)
+	}
+	return id
+}
+
+// SeedTestTeamMember adds a user to a team.
+func SeedTestTeamMember(t *testing.T, teamID, userID, role string) {
+	t.Helper()
+	_, err := DB.Exec(`
+		INSERT INTO team_members (team_id, user_id, role)
+		VALUES ($1, $2, $3)
+	`, teamID, userID, role)
+	if err != nil {
+		t.Fatalf("SeedTestTeamMember: %v", err)
+	}
+}
+
+// SeedTestGroup creates a test group and returns the group ID.
+func SeedTestGroup(t *testing.T, name, scope string, teamID *string, createdBy string) string {
+	t.Helper()
+	var id string
+	var teamArg interface{}
+	if teamID != nil {
+		teamArg = *teamID
+	}
+	err := DB.QueryRow(`
+		INSERT INTO groups (name, scope, team_id, created_by)
+		VALUES ($1, $2, $3, $4)
+		RETURNING id
+	`, name, scope, teamArg, createdBy).Scan(&id)
+	if err != nil {
+		t.Fatalf("SeedTestGroup: %v", err)
+	}
+	return id
+}
+
+// SeedGroupMember adds a user to a group.
+func SeedGroupMember(t *testing.T, groupID, userID, addedBy string) {
+	t.Helper()
+	_, err := DB.Exec(`
+		INSERT INTO group_members (group_id, user_id, added_by)
+		VALUES ($1, $2, $3)
+	`, groupID, userID, addedBy)
+	if err != nil {
+		t.Fatalf("SeedGroupMember: %v", err)
+	}
+}
+
 func envOr(key, fallback string) string {
 	if v := os.Getenv(key); v != "" {
 		return v