Function TestUnixSocketClientToolMethods

internal/cli/client_test.go:557–721

(t *testing.T)

}

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

	client := &unixSocketClient{
		socketPath: "/tmp/agh.sock",
		httpClient: &http.Client{
			Transport: roundTripperFunc(func(req *http.Request) (*http.Response, error) {
				switch {
				case req.Method == http.MethodGet && req.URL.Path == "/api/tools":
					if req.URL.Query().Get("workspace_id") != "ws-1" ||
						req.URL.Query().Get("session_id") != "sess-1" ||
						req.URL.Query().Get("agent_name") != "coder" {
						t.Fatalf("tool list query = %s, want scoped query", req.URL.RawQuery)
					}
					return newHTTPResponse(http.StatusOK, string(mustJSON(t, sampleToolsResponse()))), nil
				case req.Method == http.MethodPost && req.URL.Path == "/api/tools/search":
					var request ToolSearchRequest
					if err := json.NewDecoder(req.Body).Decode(&request); err != nil {
						t.Fatalf("decode search body: %v", err)
					}
					if request.Query != "skill" || request.Limit != 2 || request.WorkspaceID != "ws-1" {
						t.Fatalf("tool search request = %#v, want scoped search", request)
					}
					return newHTTPResponse(http.StatusOK, string(mustJSON(t, sampleToolsResponse()))), nil
				case req.Method == http.MethodGet && req.URL.Path == "/api/tools/agh__skill_view":
					if req.URL.Query().Get("workspace_id") != "ws-1" {
						t.Fatalf("tool info query = %s, want workspace_id=ws-1", req.URL.RawQuery)
					}
					return newHTTPResponse(
						http.StatusOK,
						string(mustJSON(t, ToolResponseRecord{Tool: sampleToolsResponse().Tools[0]})),
					), nil
				case req.Method == http.MethodPost && req.URL.Path == "/api/tools/agh__tool_info/invoke":
					var request ToolInvokeRequest
					if err := json.NewDecoder(req.Body).Decode(&request); err != nil {
						t.Fatalf("decode invoke body: %v", err)
					}
					if string(request.Input) != `{"tool_id":"agh__skill_view"}` ||
						request.SessionID != "sess-1" ||
						len(request.SensitiveInputFields) != 1 ||
						request.SensitiveInputFields[0] != "token" {
						t.Fatalf("invoke request = %#v, want scoped tool input", request)
					}
					response := sampleInvokeResponse()
					response.Result.Preview = "token=agh_claim_secret"
					response.Result.Structured = json.RawMessage(
						`{"access_token":"super-secret","completion_tokens":9}`,
					)
					return newHTTPResponse(http.StatusOK, string(mustJSON(t, response))), nil
				case req.Method == http.MethodGet && req.URL.Path == "/api/toolsets":
					if req.URL.Query().Get("agent_name") != "coder" {
						t.Fatalf("toolsets query = %s, want agent_name=coder", req.URL.RawQuery)
					}
					return newHTTPResponse(http.StatusOK, string(mustJSON(t, sampleToolsetsResponse()))), nil
				case req.Method == http.MethodGet && req.URL.Path == "/api/toolsets/agh__catalog":
					if req.URL.Query().Get("session_id") != "sess-1" {
						t.Fatalf("toolset info query = %s, want session_id=sess-1", req.URL.RawQuery)
					}