Skip to content

Commit 49831d9

Browse files
MajorLiftmcmire
andcommitted
Apply ExtractAvailable{Action,Event} test utility types
Co-authored-by: Elliot Winkler <[email protected]>
1 parent c35943c commit 49831d9

File tree

1 file changed

+8
-11
lines changed

1 file changed

+8
-11
lines changed

packages/assets-controllers/src/AssetsContractController.test.ts

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -23,13 +23,12 @@ import { getDefaultPreferencesState } from '@metamask/preferences-controller';
2323
import assert from 'assert';
2424

2525
import { mockNetwork } from '../../../tests/mock-network';
26-
import { buildInfuraNetworkClientConfiguration } from '../../network-controller/tests/helpers';
2726
import type {
28-
AssetsContractControllerActions,
29-
AssetsContractControllerEvents,
30-
AllowedActions as AssetsContractAllowedActions,
31-
AllowedEvents as AssetsContractAllowedEvents,
32-
} from './AssetsContractController';
27+
ExtractAvailableAction,
28+
ExtractAvailableEvent,
29+
} from '../../base-controller/tests/helpers';
30+
import { buildInfuraNetworkClientConfiguration } from '../../network-controller/tests/helpers';
31+
import type { AssetsContractControllerMessenger } from './AssetsContractController';
3332
import {
3433
AssetsContractController,
3534
MISSING_PROVIDER_ERROR,
@@ -80,12 +79,10 @@ async function setupAssetContractControllers({
8079
let provider: Provider;
8180

8281
const controllerMessenger = new ControllerMessenger<
83-
| NetworkControllerActions
84-
| AssetsContractControllerActions
85-
| AssetsContractAllowedActions,
82+
| ExtractAvailableAction<AssetsContractControllerMessenger>
83+
| NetworkControllerActions,
84+
| ExtractAvailableEvent<AssetsContractControllerMessenger>
8685
| NetworkControllerEvents
87-
| AssetsContractControllerEvents
88-
| AssetsContractAllowedEvents
8986
>();
9087
const networkController = new NetworkController({
9188
infuraProjectId,

0 commit comments

Comments
 (0)