Function TestUnixSocketClientAgentTaskMethods

internal/cli/client_test.go:289–467

(t *testing.T)

}

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

	credentials := agentidentity.Credentials{
		SessionID:   "sess-1",
		AgentName:   "coder",
		WorkspaceID: "ws-1",
	}
	var sawClaim bool
	var sawNoWork bool
	var sawHeartbeat bool
	var sawComplete bool
	var sawFail bool
	var sawRelease bool

	client := &unixSocketClient{
		socketPath: "/tmp/agh.sock",
		httpClient: &http.Client{
			Transport: roundTripperFunc(func(req *http.Request) (*http.Response, error) {
				assertAgentRequestHeaders(t, req, credentials)
				switch {
				case req.Method == http.MethodPost && req.URL.Path == "/api/agent/tasks/claim-next":
					var payload contract.AgentTaskClaimNextRequest
					if err := json.NewDecoder(req.Body).Decode(&payload); err != nil {
						t.Fatalf("json.Decode(claim-next body) error = %v", err)
					}
					if payload.WorkspaceID == "empty" {
						sawNoWork = true
						return newHTTPResponse(http.StatusNoContent, ""), nil
					}
					sawClaim = true
					if payload.WorkspaceID != "ws-1" ||
						payload.PriorityMin != 2 ||
						payload.LeaseSeconds != 120 ||
						!payload.Wait ||
						len(payload.RequiredCapabilities) != 1 ||
						payload.RequiredCapabilities[0] != "go" {
						t.Fatalf("claim-next body = %#v, want parsed request", payload)
					}
					return newHTTPResponse(
						http.StatusOK,
						`{"claim":{"task":{"id":"task-1","title":"Run task","status":"in_progress","scope":"workspace","workspace_id":"ws-1"},"run":{"id":"run-1","task_id":"task-1","status":"claimed","attempt":1,"session_id":"sess-1","queued_at":"2026-04-03T12:00:00Z"},"lease":{"task_id":"task-1","run_id":"run-1","status":"claimed","session_id":"sess-1","coordination_channel_id":"builders","claim_token_hash":"sha256:aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"},"coordination_channel":{"id":"builders","channel":"builders","display_name":"Builders","allowed_message_kinds":["status"]}}}`,
					), nil
				case req.Method == http.MethodPost && req.URL.Path == "/api/agent/tasks/run-1/heartbeat":
					sawHeartbeat = true
					var payload contract.AgentTaskHeartbeatRequest
					if err := json.NewDecoder(req.Body).Decode(&payload); err != nil {
						t.Fatalf("json.Decode(heartbeat body) error = %v", err)
					}
					if payload.LeaseSeconds != 60 {
						t.Fatalf("heartbeat body = %#v, want lease duration only", payload)
					}
					return agentTaskLeaseHTTPResponse(taskpkg.TaskRunStatusClaimed), nil
				case req.Method == http.MethodPost && req.URL.Path == "/api/agent/tasks/run-1/complete":
					sawComplete = true
					var payload contract.AgentTaskCompleteRequest
					if err := json.NewDecoder(req.Body).Decode(&payload); err != nil {
						t.Fatalf("json.Decode(complete body) error = %v", err)