update more tests

MetaMask · Nov 4, 2024 · 47ffbf8 · 47ffbf8
1 parent 50e864f
commit 47ffbf8
Show file tree

Hide file tree

Showing 2 changed files with 124 additions and 0 deletions.
diff --git a/ui/pages/confirmations/components/confirm/info/typed-sign-v1/typed-sign-v1.test.tsx b/ui/pages/confirmations/components/confirm/info/typed-sign-v1/typed-sign-v1.test.tsx
@@ -5,6 +5,7 @@ import { TransactionType } from '@metamask/transaction-controller';
 import { renderWithConfirmContextProvider } from '../../../../../../../test/lib/confirmations/render-helpers';
 import { getMockTypedSignConfirmStateForRequest } from '../../../../../../../test/data/confirmations/helper';
 import { unapprovedTypedSignMsgV1 } from '../../../../../../../test/data/confirmations/typed_sign';
+import * as snapUtils from '../../../../../../helpers/utils/snaps';
 import TypedSignInfoV1 from './typed-sign-v1';
 
 jest.mock(
@@ -16,6 +17,21 @@ jest.mock(
   }),
 );
 
+jest.mock('../../../../../../../node_modules/@metamask/snaps-utils', () => {
+  const originalUtils = jest.requireActual(
+    '../../../../../../../node_modules/@metamask/snaps-utils',
+  );
+  return {
+    ...originalUtils,
+    stripSnapPrefix: jest.fn().mockReturnValue('@metamask/examplesnap'),
+    getSnapPrefix: jest.fn().mockReturnValue('npm:'),
+  };
+});
+
+jest.mock('../../../../../../helpers/utils/snaps', () => ({
+  isSnapId: jest.fn(),
+}));
+
 describe('TypedSignInfo', () => {
   it('correctly renders typed sign data request', () => {
     const mockState = getMockTypedSignConfirmStateForRequest(
@@ -42,4 +58,50 @@ describe('TypedSignInfo', () => {
     );
     expect(container).toMatchInlineSnapshot(`<div />`);
   });
+
+  it('displays "requestFromInfoSnap" tooltip when origin is a snap', async () => {
+    const mockState = getMockTypedSignConfirmStateForRequest({
+      id: '123',
+      type: TransactionType.signTypedData,
+      chainId: '0x5',
+    });
+    (snapUtils.isSnapId as jest.Mock).mockReturnValue(true);
+    const mockStore = configureMockStore([])(mockState);
+    const { queryByText } = renderWithConfirmContextProvider(
+      <TypedSignInfoV1 />,
+      mockStore,
+    );
+
+    const requestFromLabel = queryByText('Request from');
+
+    await requestFromLabel?.dispatchEvent(
+      new MouseEvent('mouseenter', { bubbles: true }),
+    );
+    expect(
+      queryByText('This is the Snap asking for your signature.'),
+    ).toBeDefined();
+  });
+
+  it('displays "requestFromInfo" tooltip when origin is not a snap', async () => {
+    const mockState = getMockTypedSignConfirmStateForRequest({
+      id: '123',
+      type: TransactionType.signTypedData,
+      chainId: '0x5',
+    });
+    (snapUtils.isSnapId as jest.Mock).mockReturnValue(false);
+    const mockStore = configureMockStore([])(mockState);
+    const { queryByText } = renderWithConfirmContextProvider(
+      <TypedSignInfoV1 />,
+      mockStore,
+    );
+
+    const requestFromLabel = queryByText('Request from');
+
+    await requestFromLabel?.dispatchEvent(
+      new MouseEvent('mouseenter', { bubbles: true }),
+    );
+    expect(
+      queryByText('This is the site asking for your signature.'),
+    ).toBeDefined();
+  });
 });
diff --git a/ui/pages/confirmations/components/confirm/info/typed-sign/typed-sign.test.tsx b/ui/pages/confirmations/components/confirm/info/typed-sign/typed-sign.test.tsx
@@ -16,6 +16,7 @@ import {
   unapprovedTypedSignMsgV3,
 } from '../../../../../../../test/data/confirmations/typed_sign';
 import { renderWithConfirmContextProvider } from '../../../../../../../test/lib/confirmations/render-helpers';
+import * as snapUtils from '../../../../../../helpers/utils/snaps';
 import TypedSignInfo from './typed-sign';
 
 jest.mock(
@@ -33,6 +34,21 @@ jest.mock('../../../../../../store/actions', () => {
   };
 });
 
+jest.mock('../../../../../../../node_modules/@metamask/snaps-utils', () => {
+  const originalUtils = jest.requireActual(
+    '../../../../../../../node_modules/@metamask/snaps-utils',
+  );
+  return {
+    ...originalUtils,
+    stripSnapPrefix: jest.fn().mockReturnValue('@metamask/examplesnap'),
+    getSnapPrefix: jest.fn().mockReturnValue('npm:'),
+  };
+});
+
+jest.mock('../../../../../../helpers/utils/snaps', () => ({
+  isSnapId: jest.fn(),
+}));
+
 describe('TypedSignInfo', () => {
   it('renders origin for typed sign data request', () => {
     const state = getMockTypedSignConfirmState();
@@ -127,4 +143,50 @@ describe('TypedSignInfo', () => {
     );
     expect(container).toMatchSnapshot();
   });
+
+  it('displays "requestFromInfoSnap" tooltip when origin is a snap', async () => {
+    const mockState = getMockTypedSignConfirmStateForRequest({
+      id: '123',
+      type: TransactionType.signTypedData,
+      chainId: '0x5',
+    });
+    (snapUtils.isSnapId as jest.Mock).mockReturnValue(true);
+    const mockStore = configureMockStore([])(mockState);
+    const { queryByText } = renderWithConfirmContextProvider(
+      <TypedSignInfo />,
+      mockStore,
+    );
+
+    const requestFromLabel = queryByText('Request from');
+
+    await requestFromLabel?.dispatchEvent(
+      new MouseEvent('mouseenter', { bubbles: true }),
+    );
+    expect(
+      queryByText('This is the Snap asking for your signature.'),
+    ).toBeDefined();
+  });
+
+  it('displays "requestFromInfo" tooltip when origin is not a snap', async () => {
+    const mockState = getMockTypedSignConfirmStateForRequest({
+      id: '123',
+      type: TransactionType.signTypedData,
+      chainId: '0x5',
+    });
+    (snapUtils.isSnapId as jest.Mock).mockReturnValue(false);
+    const mockStore = configureMockStore([])(mockState);
+    const { queryByText } = renderWithConfirmContextProvider(
+      <TypedSignInfo />,
+      mockStore,
+    );
+
+    const requestFromLabel = queryByText('Request from');
+
+    await requestFromLabel?.dispatchEvent(
+      new MouseEvent('mouseenter', { bubbles: true }),
+    );
+    expect(
+      queryByText('This is the site asking for your signature.'),
+    ).toBeDefined();
+  });
 });