Function runForegroundWithOutput

coderd/x/chatd/chattool/execute_internal_test.go:67–100

runForegroundWithOutput runs a foreground command through the Execute tool with a mock that returns the given output, and returns the parsed result.

(t *testing.T, output string)

// Execute tool with a mock that returns the given output, and
// returns the parsed result.
func runForegroundWithOutput(t *testing.T, output string) ExecuteResult {
	t.Helper()
	ctrl := gomock.NewController(t)
	mockConn := agentconnmock.NewMockAgentConn(ctrl)

	mockConn.EXPECT().
		StartProcess(gomock.Any(), gomock.Any()).
		Return(workspacesdk.StartProcessResponse{ID: "proc-1"}, nil)
	exitCode := 0
	mockConn.EXPECT().
		ProcessOutput(gomock.Any(), "proc-1", gomock.Any()).
		Return(workspacesdk.ProcessOutputResponse{
			Running:  false,
			ExitCode: &exitCode,
			Output:   output,
		}, nil)

	tool := Execute(ExecuteOptions{
		GetWorkspaceConn: func(_ context.Context) (workspacesdk.AgentConn, error) {
			return mockConn, nil
		},
	})
	ctx := testutil.Context(t, testutil.WaitMedium)
	resp, err := tool.Run(ctx, fantasy.ToolCall{
		ID:    "call-1",
		Name:  "execute",
		Input: `{"command":"echo test"}`,
	})
	require.NoError(t, err)

	var result ExecuteResult
	require.NoError(t, json.Unmarshal([]byte(resp.Content), &result))
	return result
}