@@ -23,13 +23,12 @@ import { getDefaultPreferencesState } from '@metamask/preferences-controller';
23
23
import assert from 'assert' ;
24
24
25
25
import { mockNetwork } from '../../../tests/mock-network' ;
26
- import { buildInfuraNetworkClientConfiguration } from '../../network-controller/tests/helpers' ;
27
26
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' ;
33
32
import {
34
33
AssetsContractController ,
35
34
MISSING_PROVIDER_ERROR ,
@@ -80,12 +79,10 @@ async function setupAssetContractControllers({
80
79
let provider : Provider ;
81
80
82
81
const controllerMessenger = new ControllerMessenger <
83
- | NetworkControllerActions
84
- | AssetsContractControllerActions
85
- | AssetsContractAllowedActions ,
82
+ | ExtractAvailableAction < AssetsContractControllerMessenger >
83
+ | NetworkControllerActions ,
84
+ | ExtractAvailableEvent < AssetsContractControllerMessenger >
86
85
| NetworkControllerEvents
87
- | AssetsContractControllerEvents
88
- | AssetsContractAllowedEvents
89
86
> ( ) ;
90
87
const networkController = new NetworkController ( {
91
88
infuraProjectId,
0 commit comments