|
| 1 | +import { seedlessOnboardingControllerInit } from '.'; |
| 2 | +import { ExtendedControllerMessenger } from '../../../ExtendedControllerMessenger'; |
| 3 | +import { buildControllerInitRequestMock } from '../../utils/test-utils'; |
| 4 | +import { ControllerInitRequest } from '../../types'; |
| 5 | +import { |
| 6 | + SeedlessOnboardingController, |
| 7 | + SeedlessOnboardingControllerMessenger, |
| 8 | + SeedlessOnboardingControllerState, |
| 9 | +} from '@metamask/seedless-onboarding-controller'; |
| 10 | + |
| 11 | +jest.mock('@metamask/seedless-onboarding-controller', () => { |
| 12 | + const actualSeedlessOnboardingController = jest.requireActual( |
| 13 | + '@metamask/seedless-onboarding-controller', |
| 14 | + ); |
| 15 | + return { |
| 16 | + controllerName: actualSeedlessOnboardingController.controllerName, |
| 17 | + getDefaultSeedlessOnboardingControllerState: |
| 18 | + actualSeedlessOnboardingController.getDefaultSeedlessOnboardingControllerState, |
| 19 | + SeedlessOnboardingController: jest.fn(), |
| 20 | + Web3AuthNetwork: actualSeedlessOnboardingController.Web3AuthNetwork, |
| 21 | + }; |
| 22 | +}); |
| 23 | + |
| 24 | +describe('seedless onboarding controller init', () => { |
| 25 | + const seedlessOnboardingControllerClassMock = jest.mocked( |
| 26 | + SeedlessOnboardingController, |
| 27 | + ); |
| 28 | + let initRequestMock: jest.Mocked< |
| 29 | + ControllerInitRequest<SeedlessOnboardingControllerMessenger> |
| 30 | + >; |
| 31 | + |
| 32 | + beforeEach(() => { |
| 33 | + jest.resetAllMocks(); |
| 34 | + const baseControllerMessenger = new ExtendedControllerMessenger(); |
| 35 | + // Create controller init request mock |
| 36 | + initRequestMock = buildControllerInitRequestMock(baseControllerMessenger); |
| 37 | + }); |
| 38 | + |
| 39 | + it('returns controller instance', () => { |
| 40 | + expect( |
| 41 | + seedlessOnboardingControllerInit(initRequestMock).controller, |
| 42 | + ).toBeInstanceOf(SeedlessOnboardingController); |
| 43 | + }); |
| 44 | + |
| 45 | + it('controller state should be default state when no initial state is passed in', () => { |
| 46 | + const defaultSeedlessOnboardingControllerState = jest |
| 47 | + .requireActual('@metamask/seedless-onboarding-controller') |
| 48 | + .getDefaultSeedlessOnboardingControllerState(); |
| 49 | + |
| 50 | + seedlessOnboardingControllerInit(initRequestMock); |
| 51 | + |
| 52 | + const seedlessOnboardingControllerState = |
| 53 | + seedlessOnboardingControllerClassMock.mock.calls[0][0].state; |
| 54 | + |
| 55 | + expect(seedlessOnboardingControllerState).toEqual( |
| 56 | + defaultSeedlessOnboardingControllerState, |
| 57 | + ); |
| 58 | + }); |
| 59 | + |
| 60 | + it('controller state should be initial state when initial state is passed in', () => { |
| 61 | + const initialSeedlessOnboardingControllerState: Partial<SeedlessOnboardingControllerState> = |
| 62 | + { |
| 63 | + vault: undefined, |
| 64 | + nodeAuthTokens: undefined, |
| 65 | + }; |
| 66 | + |
| 67 | + initRequestMock.persistedState = { |
| 68 | + ...initRequestMock.persistedState, |
| 69 | + SeedlessOnboardingController: initialSeedlessOnboardingControllerState, |
| 70 | + }; |
| 71 | + |
| 72 | + seedlessOnboardingControllerInit(initRequestMock); |
| 73 | + |
| 74 | + const seedlessOnboardingControllerState = |
| 75 | + seedlessOnboardingControllerClassMock.mock.calls[0][0].state; |
| 76 | + |
| 77 | + expect(seedlessOnboardingControllerState).toStrictEqual( |
| 78 | + initialSeedlessOnboardingControllerState, |
| 79 | + ); |
| 80 | + }); |
| 81 | +}); |
0 commit comments