Function TestMCPHTTP_E2E_ToolWithWorkspace

coderd/mcp/mcp_e2e_test.go:223–295

(t *testing.T)

}

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

	coderClient, closer, api := coderdtest.NewWithAPI(t, nil)
	defer closer.Close()

	user := coderdtest.CreateFirstUser(t, coderClient)
	r := dbfake.WorkspaceBuild(t, api.Database, database.WorkspaceTable{
		Name:           "myworkspace",
		OrganizationID: user.OrganizationID,
		OwnerID:        user.UserID,
	}).WithAgent().Do()

	fs := afero.NewMemMapFs()
	tmpdir := os.TempDir()
	require.NoError(t, fs.MkdirAll(tmpdir, 0o755))
	filePath := filepath.Join(tmpdir, "mcp-http-test.txt")
	require.NoError(t, afero.WriteFile(fs, filePath, []byte("hello from mcp"), 0o644))

	_ = agenttest.New(t, coderClient.URL, r.AgentToken, func(opts *agent.Options) {
		opts.Filesystem = fs
	})
	coderdtest.NewWorkspaceAgentWaiter(t, coderClient, r.Workspace.ID).Wait()

	mcpURL := api.AccessURL.String() + mcpserver.MCPEndpoint
	mcpClient, err := mcpclient.NewStreamableHttpClient(mcpURL,
		transport.WithHTTPHeaders(map[string]string{
			"Authorization": "Bearer " + coderClient.SessionToken(),
		}))
	require.NoError(t, err)
	defer func() {
		if closeErr := mcpClient.Close(); closeErr != nil {
			t.Logf("Failed to close MCP client: %v", closeErr)
		}
	}()

	ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
	defer cancel()

	require.NoError(t, mcpClient.Start(ctx))
	_, err = mcpClient.Initialize(ctx, mcp.InitializeRequest{
		Params: mcp.InitializeParams{
			ProtocolVersion: mcp.LATEST_PROTOCOL_VERSION,
			ClientInfo: mcp.Implementation{
				Name:    "test-client-workspace",
				Version: "1.0.0",
			},
		},
	})
	require.NoError(t, err)

	toolResult, err := mcpClient.CallTool(ctx, mcp.CallToolRequest{
		Params: mcp.CallToolParams{
			Name: toolsdk.ToolNameWorkspaceLS,
			Arguments: map[string]any{
				"workspace": r.Workspace.Name,
				"path":      tmpdir,
			},