mocks

Author: HowardBraham

app/scripts/lib/snap-keyring/keyring-snaps-permissions.test.ts

@@ -8,6 +8,12 @@ import {
   keyringSnapPermissionsBuilder,
 } from './keyring-snaps-permissions';
 
+// Run these tests as if we were in a Flask build
+jest.mock('../../../../shared/lib/build-types', () => ({
+  ...jest.requireActual('../../../../shared/lib/build-types'),
+  isFlask: jest.fn().mockReturnValue(true),
+}));
+
 const PORTFOLIO_ORIGINS: string[] = [
   'https://app.metamask.io',
   'https://dev.app.metamask.io',

app/scripts/lib/snap-keyring/utils/isBlockedUrl.test.ts

@@ -8,6 +8,12 @@ import { Messenger } from '@metamask/messenger';
 import { getRootMessenger } from '../../messenger';
 import { isBlockedUrl } from './isBlockedUrl';
 
+// Run these tests as if we were in a Flask build
+jest.mock('../../../../../shared/lib/build-types', () => ({
+  ...jest.requireActual('../../../../../shared/lib/build-types'),
+  isFlask: jest.fn().mockReturnValue(true),
+}));
+
 describe('isBlockedUrl', () => {
   const messenger = getRootMessenger<
     PhishingControllerActions,

ui/pages/onboarding-flow/onboarding-flow.test.tsx

@@ -29,6 +29,12 @@ import { mockNetworkState } from '../../../test/stub/networks';
 import { FirstTimeFlowType } from '../../../shared/constants/onboarding';
 import OnboardingFlow from './onboarding-flow';
 
+// Run these tests as if we were in a Flask build
+jest.mock('../../../../shared/lib/build-types', () => ({
+  ...jest.requireActual('../../../../shared/lib/build-types'),
+  isFlask: jest.fn().mockReturnValue(true),
+}));
+
 const mockUseNavigate = jest.fn();
 
 jest.mock('react-router-dom', () => ({

ui/pages/settings/experimental-tab/experimental-tab.test.tsx

@@ -5,8 +5,17 @@ import configureStore from '../../../store/store';
 import mockState from '../../../../test/data/mock-state.json';
 import { LegacyMetaMetricsProvider } from '../../../contexts/metametrics';
 import * as storeActions from '../../../store/actions';
+import { isFlask } from '../../../../shared/lib/build-types';
 import ExperimentalTab from './experimental-tab';
 
+// Allow each test to set whether it's Flask or not
+jest.mock('../../../../shared/lib/build-types', () => ({
+  ...jest.requireActual('../../../../shared/lib/build-types'),
+  isFlask: jest.fn(),
+}));
+
+const mockIsFlask = jest.mocked(isFlask);
+
 jest.mock('../../../store/actions', () => ({
   ...jest.requireActual('../../../store/actions'),
   setAddSnapAccountEnabled: jest.fn().mockReturnValue(async () => undefined),
@@ -29,28 +38,30 @@ const render = (overrideMetaMaskState = {}) => {
 
 describe('ExperimentalTab', () => {
   it('renders ExperimentalTab component without error', () => {
+    mockIsFlask.mockReturnValue(false);
     expect(() => {
       render();
     }).not.toThrow();
   });
 
   it('renders one toggle option when build type is main', () => {
-    process.env.METAMASK_BUILD_TYPE = 'main';
+    mockIsFlask.mockReturnValue(false);
     const { getAllByRole } = render();
     const toggle = getAllByRole('checkbox');
 
     expect(toggle).toHaveLength(1);
   });
 
   it('renders two toggle options when build type is flask', () => {
-    process.env.METAMASK_BUILD_TYPE = 'flask';
+    mockIsFlask.mockReturnValue(true);
     const { getAllByRole } = render();
     const toggle = getAllByRole('checkbox');
 
     expect(toggle).toHaveLength(2);
   });
 
   it('enables add account snap', async () => {
+    mockIsFlask.mockReturnValue(true);
     const { getByTestId } = render();
 
     const toggle = getByTestId('add-account-snap-toggle-button');