Function recordDesktopActions

coderd/x/chatd/chattool/computeruse_test.go:1022–1047

(
	t testing.TB,
	mockConn *agentconnmock.MockAgentConn,
	geometry workspacesdk.DesktopGeometry,
	times int,
	screenshotPNG string,
)

}

func recordDesktopActions(
	t testing.TB,
	mockConn *agentconnmock.MockAgentConn,
	geometry workspacesdk.DesktopGeometry,
	times int,
	screenshotPNG string,
) *[]workspacesdk.DesktopAction {
	t.Helper()
	actions := make([]workspacesdk.DesktopAction, 0, times)
	mockConn.EXPECT().ExecuteDesktopAction(
		gomock.Any(),
		gomock.AssignableToTypeOf(workspacesdk.DesktopAction{}),
	).DoAndReturn(func(_ context.Context, action workspacesdk.DesktopAction) (workspacesdk.DesktopActionResponse, error) {
		actions = append(actions, action)
		if action.Action == "screenshot" {
			return workspacesdk.DesktopActionResponse{
				Output:           "screenshot",
				ScreenshotData:   screenshotPNG,
				ScreenshotWidth:  geometry.DeclaredWidth,
				ScreenshotHeight: geometry.DeclaredHeight,
			}, nil
		}
		return workspacesdk.DesktopActionResponse{Output: action.Action + " performed"}, nil
	}).Times(times)
	return &actions
}

func assertDesktopActionScaled(
	t testing.TB,