Function TestTaskManagerExecutionProfiles

internal/task/manager_profile_test.go:9–115

(t *testing.T)

)

func TestTaskManagerExecutionProfiles(t *testing.T) {
	t.Parallel()

	t.Run("Should return default inherit profile when no row is persisted", func(t *testing.T) {
		t.Parallel()

		store := newInMemoryManagerStore()
		store.tasks["task-1"] = validProfileTask("task-1")
		manager := newTaskManagerForTest(t, store)

		profile, err := manager.GetExecutionProfile(context.Background(), "task-1", validActorContext())
		if err != nil {
			t.Fatalf("GetExecutionProfile() error = %v", err)
		}

		if got, want := profile.Coordinator.Mode, CoordinatorModeInherit; got != want {
			t.Fatalf("Coordinator.Mode = %q, want %q", got, want)
		}
		if got, want := profile.Worker.Mode, WorkerModeInherit; got != want {
			t.Fatalf("Worker.Mode = %q, want %q", got, want)
		}
		if got, want := profile.Sandbox.Mode, SandboxModeInherit; got != want {
			t.Fatalf("Sandbox.Mode = %q, want %q", got, want)
		}
	})

	t.Run("Should persist validated profile and emit audit event", func(t *testing.T) {
		t.Parallel()

		store := newInMemoryManagerStore()
		store.tasks["task-1"] = validProfileTask("task-1")
		manager := newTaskManagerForTest(t, store)

		profile, err := manager.SetExecutionProfile(context.Background(), "task-1", &ExecutionProfile{
			Coordinator: CoordinatorProfile{Mode: CoordinatorModeGuided, Guidance: "Keep the worker focused."},
			Worker: WorkerProfile{
				Mode:                  WorkerModeSelect,
				AgentName:             "coder",
				AllowedAgentNames:     []string{"coder"},
				RequiredCapabilities:  []string{"go"},
				PreferredCapabilities: []string{"tests"},
			},
			Review: ReviewProfile{
				AgentName:         "reviewer",
				AllowedAgentNames: []string{"reviewer"},
			},
			Sandbox: SandboxPolicy{Mode: SandboxModeRef, SandboxRef: "workspace"},
		}, validActorContext())
		if err != nil {
			t.Fatalf("SetExecutionProfile() error = %v", err)
		}

		if got, want := profile.TaskID, "task-1"; got != want {
			t.Fatalf("TaskID = %q, want %q", got, want)
		}
		if got, want := store.profiles["task-1"].Worker.AgentName, "coder"; got != want {
			t.Fatalf("stored Worker.AgentName = %q, want %q", got, want)
		}