diff --git a/packages/phishing-controller/src/PhishingController.test.ts b/packages/phishing-controller/src/PhishingController.test.ts index 810f7a179a8..35fdfb27014 100644 --- a/packages/phishing-controller/src/PhishingController.test.ts +++ b/packages/phishing-controller/src/PhishingController.test.ts @@ -1,4 +1,4 @@ -import { ControllerMessenger } from '@metamask/base-controller'; +import { Messenger } from '@metamask/base-controller'; import { strict as assert } from 'assert'; import nock from 'nock'; import * as sinon from 'sinon'; @@ -21,23 +21,18 @@ import { getHostnameFromUrl } from './utils'; const controllerName = 'PhishingController'; /** - * Constructs a restricted controller messenger. + * Constructs a restricted messenger. * - * @returns A restricted controller messenger. + * @returns A restricted messenger. */ function getRestrictedMessenger() { - const controllerMessenger = new ControllerMessenger< - PhishingControllerActions, - never - >(); + const messenger = new Messenger(); - const messenger = controllerMessenger.getRestricted({ + return messenger.getRestricted({ name: controllerName, allowedActions: [], allowedEvents: [], }); - - return messenger; } /** diff --git a/packages/phishing-controller/src/PhishingController.ts b/packages/phishing-controller/src/PhishingController.ts index cdf8eaab235..d62ebaac87c 100644 --- a/packages/phishing-controller/src/PhishingController.ts +++ b/packages/phishing-controller/src/PhishingController.ts @@ -1,7 +1,7 @@ import type { ControllerGetStateAction, ControllerStateChangeEvent, - RestrictedControllerMessenger, + RestrictedMessenger, } from '@metamask/base-controller'; import { BaseController } from '@metamask/base-controller'; import { safelyExecute } from '@metamask/controller-utils'; @@ -268,7 +268,7 @@ export type PhishingControllerStateChangeEvent = ControllerStateChangeEvent< export type PhishingControllerEvents = PhishingControllerStateChangeEvent; -export type PhishingControllerMessenger = RestrictedControllerMessenger< +export type PhishingControllerMessenger = RestrictedMessenger< typeof controllerName, PhishingControllerActions, PhishingControllerEvents,