feat(ai): add mcp-metadata (#13150)

Co-authored-by: Félix Malfait <felix@twenty.com>
2025-07-16 21:32:32 +02:00
parent b25f50e288
commit 11abe5440b
50 changed files with 1288 additions and 230 deletions
@@ -1,44 +1,51 @@
-import { act, renderHook } from '@testing-library/react';
+import { act, renderHook, waitFor } from '@testing-library/react';
 import { RecoilRoot } from 'recoil';
-import * as ReactRouterDom from 'react-router-dom';

 import { useRequestFreshCaptchaToken } from '@/captcha/hooks/useRequestFreshCaptchaToken';
+import { isRequestingCaptchaTokenState } from '@/captcha/states/isRequestingCaptchaTokenState';
+import { isCaptchaRequiredForPath } from '@/captcha/utils/isCaptchaRequiredForPath';
+import { captchaState } from '@/client-config/states/captchaState';
+import { CaptchaDriverType } from '~/generated-metadata/graphql';

-jest.mock('react-router-dom', () => ({
-  ...jest.requireActual('react-router-dom'),
-  useLocation: jest.fn(),
-}));
+jest.mock('@/captcha/utils/isCaptchaRequiredForPath');

 describe('useRequestFreshCaptchaToken', () => {
+  const mockGrecaptchaExecute = jest.fn();
+  const mockTurnstileRender = jest.fn();
+  const mockTurnstileExecute = jest.fn();
+
  beforeEach(() => {
-    // Mock useLocation to return a path that requires captcha
-    (ReactRouterDom.useLocation as jest.Mock).mockReturnValue({
-      pathname: '/sign-in',
-    });
-
-    // Mock window.grecaptcha
-    window.grecaptcha = {
-      execute: jest.fn().mockImplementation(() => {
-        return Promise.resolve('google-recaptcha-token');
-      }),
-    };
-
-    // Mock window.turnstile
-    window.turnstile = {
-      render: jest.fn().mockReturnValue('turnstile-widget-id'),
-      execute: jest.fn().mockImplementation((widgetId, options) => {
-        return options?.callback('turnstile-token');
-      }),
-    };
-  });
-
-  afterEach(() => {
    jest.clearAllMocks();
-    delete window.grecaptcha;
-    delete window.turnstile;
+
+    window.grecaptcha = {
+      execute: mockGrecaptchaExecute,
+    };
+
+    window.turnstile = {
+      render: mockTurnstileRender,
+      execute: mockTurnstileExecute,
+    };
+
+    (isCaptchaRequiredForPath as jest.Mock).mockReturnValue(true);
+
+    mockGrecaptchaExecute.mockImplementation((_siteKey, _options) => {
+      return Promise.resolve('google-recaptcha-token');
+    });
+
+    mockTurnstileRender.mockImplementation((_selector, _options) => {
+      return 'turnstile-widget-id';
+    });
+
+    mockTurnstileExecute.mockImplementation((widgetId, options) => {
+      if (options !== undefined && typeof options.callback === 'function') {
+        options.callback('turnstile-token');
+      }
+    });
  });

-  it('should not request a token if captcha is not required for the path', async () => {
+  it('should not request a token if captcha is not required for the current path', async () => {
+    (isCaptchaRequiredForPath as jest.Mock).mockReturnValue(false);
+
    const { result } = renderHook(() => useRequestFreshCaptchaToken(), {
      wrapper: RecoilRoot,
    });
@@ -47,11 +54,12 @@ describe('useRequestFreshCaptchaToken', () => {
      await result.current.requestFreshCaptchaToken();
    });

-    expect(window.grecaptcha.execute).not.toHaveBeenCalled();
-    expect(window.turnstile.execute).not.toHaveBeenCalled();
+    expect(mockGrecaptchaExecute).not.toHaveBeenCalled();
+    expect(mockTurnstileRender).not.toHaveBeenCalled();
+    expect(mockTurnstileExecute).not.toHaveBeenCalled();
  });

-  it('should not request a token if captcha provider is not defined', async () => {
+  it('should not request a token if captcha provider is undefined', async () => {
    const { result } = renderHook(() => useRequestFreshCaptchaToken(), {
      wrapper: RecoilRoot,
    });
@@ -60,7 +68,67 @@ describe('useRequestFreshCaptchaToken', () => {
      await result.current.requestFreshCaptchaToken();
    });

-    expect(window.grecaptcha.execute).not.toHaveBeenCalled();
-    expect(window.turnstile.execute).not.toHaveBeenCalled();
+    expect(mockGrecaptchaExecute).not.toHaveBeenCalled();
+    expect(mockTurnstileRender).not.toHaveBeenCalled();
+    expect(mockTurnstileExecute).not.toHaveBeenCalled();
+  });
+
+  it('should request a token from Google reCAPTCHA when provider is GOOGLE_RECAPTCHA', async () => {
+    const { result } = renderHook(() => useRequestFreshCaptchaToken(), {
+      wrapper: ({ children }) => (
+        <RecoilRoot
+          initializeState={({ set }) => {
+            set(captchaState, {
+              provider: CaptchaDriverType.GOOGLE_RECAPTCHA,
+              siteKey: 'google-site-key',
+            });
+            set(isRequestingCaptchaTokenState, false);
+          }}
+        >
+          {children}
+        </RecoilRoot>
+      ),
+    });
+
+    await act(async () => {
+      await result.current.requestFreshCaptchaToken();
+    });
+
+    expect(mockGrecaptchaExecute).toHaveBeenCalledWith('google-site-key', {
+      action: 'submit',
+    });
+
+    await waitFor(() => {
+      expect(mockGrecaptchaExecute).toHaveBeenCalled();
+    });
+  });
+
+  it('should request a token from Turnstile when provider is TURNSTILE', async () => {
+    const { result } = renderHook(() => useRequestFreshCaptchaToken(), {
+      wrapper: ({ children }) => (
+        <RecoilRoot
+          initializeState={({ set }) => {
+            set(captchaState, {
+              provider: CaptchaDriverType.TURNSTILE,
+              siteKey: 'turnstile-site-key',
+            });
+            set(isRequestingCaptchaTokenState, false);
+          }}
+        >
+          {children}
+        </RecoilRoot>
+      ),
+    });
+
+    await act(async () => {
+      await result.current.requestFreshCaptchaToken();
+    });
+
+    expect(mockTurnstileRender).toHaveBeenCalledWith('#captcha-widget', {
+      sitekey: 'turnstile-site-key',
+    });
+    expect(mockTurnstileExecute).toHaveBeenCalledWith('turnstile-widget-id', {
+      callback: expect.any(Function),
+    });
  });
 });