diff --git a/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.test.tsx b/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.test.tsx index 0f57c5c0551..e2aa9d94339 100644 --- a/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.test.tsx +++ b/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.test.tsx @@ -1,7 +1,7 @@ /** * @vitest-environment jsdom */ -import { act, type ReactNode } from 'react' +import { act, type ComponentProps, type ReactNode } from 'react' import { createRoot, type Root } from 'react-dom/client' import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest' @@ -19,6 +19,11 @@ vi.mock('next/navigation', () => ({ })) vi.mock('@sim/emcn', () => ({ + Button: ({ + variant, + iconSize, + ...props + }: ComponentProps<'button'> & { variant?: string; iconSize?: string }) => + {copied ? 'Copied message' : 'Copy message'} @@ -180,27 +180,29 @@ export const MessageActions = memo(function MessageActions({ <> - + Good response - + Bad response @@ -209,15 +211,16 @@ export const MessageActions = memo(function MessageActions({ {canFork && ( - + Fork in new chat