@@ -11,11 +11,12 @@ import {
1111} from '@metamask/transaction-controller' ;
1212
1313import { ShieldController } from './ShieldController' ;
14- import { createMockBackend } from '../tests/mocks/backend' ;
14+ import { createMockBackend , MOCK_COVERAGE_ID } from '../tests/mocks/backend' ;
1515import { createMockMessenger } from '../tests/mocks/messenger' ;
1616import {
1717 generateMockSignatureRequest ,
1818 generateMockTxMeta ,
19+ setupCoverageResultReceived ,
1920} from '../tests/utils' ;
2021
2122/**
@@ -68,7 +69,7 @@ describe('ShieldController', () => {
6869 undefined as never ,
6970 ) ;
7071 expect ( await coverageResultReceived ) . toBeUndefined ( ) ;
71- expect ( backend . checkCoverage ) . toHaveBeenCalledWith ( txMeta ) ;
72+ expect ( backend . checkCoverage ) . toHaveBeenCalledWith ( { txMeta } ) ;
7273 } ) ;
7374
7475 it ( 'should no longer trigger checkCoverage when controller is stopped' , async ( ) => {
@@ -126,12 +127,7 @@ describe('ShieldController', () => {
126127 it ( 'should check coverage when a transaction is simulated' , async ( ) => {
127128 const { baseMessenger, backend } = setup ( ) ;
128129 const txMeta = generateMockTxMeta ( ) ;
129- const coverageResultReceived = new Promise < void > ( ( resolve ) => {
130- baseMessenger . subscribe (
131- 'ShieldController:coverageResultReceived' ,
132- ( _coverageResult ) => resolve ( ) ,
133- ) ;
134- } ) ;
130+ const coverageResultReceived = setupCoverageResultReceived ( baseMessenger ) ;
135131
136132 // Add transaction.
137133 baseMessenger . publish (
@@ -140,19 +136,25 @@ describe('ShieldController', () => {
140136 undefined as never ,
141137 ) ;
142138 expect ( await coverageResultReceived ) . toBeUndefined ( ) ;
143- expect ( backend . checkCoverage ) . toHaveBeenCalledWith ( txMeta ) ;
139+ expect ( backend . checkCoverage ) . toHaveBeenCalledWith ( { txMeta } ) ;
144140
145141 // Simulate transaction.
146- txMeta . simulationData = {
142+ const txMeta2 = { ...txMeta } ;
143+ txMeta2 . simulationData = {
147144 tokenBalanceChanges : [ ] ,
148145 } ;
146+ const coverageResultReceived2 =
147+ setupCoverageResultReceived ( baseMessenger ) ;
149148 baseMessenger . publish (
150149 'TransactionController:stateChange' ,
151- { transactions : [ txMeta ] } as TransactionControllerState ,
150+ { transactions : [ txMeta2 ] } as TransactionControllerState ,
152151 undefined as never ,
153152 ) ;
154- expect ( await coverageResultReceived ) . toBeUndefined ( ) ;
155- expect ( backend . checkCoverage ) . toHaveBeenCalledWith ( txMeta ) ;
153+ expect ( await coverageResultReceived2 ) . toBeUndefined ( ) ;
154+ expect ( backend . checkCoverage ) . toHaveBeenCalledWith ( {
155+ coverageId : MOCK_COVERAGE_ID ,
156+ txMeta : txMeta2 ,
157+ } ) ;
156158 } ) ;
157159
158160 it ( 'throws an error when the coverage ID has changed' , async ( ) => {
@@ -189,9 +191,9 @@ describe('ShieldController', () => {
189191 undefined as never ,
190192 ) ;
191193 expect ( await coverageResultReceived ) . toBeUndefined ( ) ;
192- expect ( backend . checkSignatureCoverage ) . toHaveBeenCalledWith (
194+ expect ( backend . checkSignatureCoverage ) . toHaveBeenCalledWith ( {
193195 signatureRequest,
194- ) ;
196+ } ) ;
195197 } ) ;
196198 } ) ;
197199
@@ -214,9 +216,9 @@ describe('ShieldController', () => {
214216 undefined as never ,
215217 ) ;
216218 expect ( await coverageResultReceived1 ) . toBeUndefined ( ) ;
217- expect ( backend . checkSignatureCoverage ) . toHaveBeenCalledWith (
218- signatureRequest1 ,
219- ) ;
219+ expect ( backend . checkSignatureCoverage ) . toHaveBeenCalledWith ( {
220+ signatureRequest : signatureRequest1 ,
221+ } ) ;
220222
221223 const signatureRequest2 = generateMockSignatureRequest ( ) ;
222224 const coverageResultReceived2 = new Promise < void > ( ( resolve ) => {
@@ -236,9 +238,9 @@ describe('ShieldController', () => {
236238 ) ;
237239
238240 expect ( await coverageResultReceived2 ) . toBeUndefined ( ) ;
239- expect ( backend . checkSignatureCoverage ) . toHaveBeenCalledWith (
240- signatureRequest2 ,
241- ) ;
241+ expect ( backend . checkSignatureCoverage ) . toHaveBeenCalledWith ( {
242+ signatureRequest : signatureRequest2 ,
243+ } ) ;
242244 } ) ;
243245
244246 describe ( 'logSignature' , ( ) => {
0 commit comments