@@ -53,7 +53,7 @@ describe('PauseSto', () => {
5353 let usdTieredStoFactoryMock : MockManager < usdTieredStoFactoryModule . UsdTieredStoFactory > ;
5454
5555 let securityTokenMock : MockManager < contractWrappersModule . SecurityToken_3_0_0 > ;
56- let moduleFactoryMock : MockManager < contractWrappersModule . ModuleFactory_3_0_0 > ;
56+
5757 let factoryMockSetup : Factories ;
5858 let securityTokenId : string ;
5959
@@ -73,15 +73,11 @@ describe('PauseSto', () => {
7373
7474 securityTokenMock = ImportMock . mockClass ( contractWrappersModule , 'SecurityToken_3_0_0' ) ;
7575
76- moduleFactoryMock = ImportMock . mockClass ( contractWrappersModule , 'ModuleFactory_3_0_0' ) ;
77-
7876 tokenFactoryMock . mock (
7977 'getSecurityTokenInstanceFromTicker' ,
8078 securityTokenMock . getMockInstance ( )
8179 ) ;
8280
83- moduleWrapperFactoryMock . mock ( 'getModuleInstance' , moduleFactoryMock . getMockInstance ( ) ) ;
84-
8581 cappedStoFactoryMock = ImportMock . mockClass ( cappedStoFactoryModule , 'CappedStoFactory' ) ;
8682
8783 usdTieredStoFactoryMock = ImportMock . mockClass (
@@ -99,6 +95,8 @@ describe('PauseSto', () => {
9995
10096 securityTokenId = SecurityToken . generateId ( { symbol : cappedParams . symbol } ) ;
10197
98+ moduleWrapperFactoryMock . mock ( 'getModuleInstance' , cappedStoMock . getMockInstance ( ) ) ;
99+
102100 // Instantiate PauseSto
103101 target = new PauseSto ( cappedParams , contextMock . getMockInstance ( ) ) ;
104102 } ) ;
@@ -116,10 +114,12 @@ describe('PauseSto', () => {
116114 describe ( 'PauseSto' , ( ) => {
117115 test ( 'should add the transaction to the queue to pause a capped sto' , async ( ) => {
118116 const addTransactionSpy = spy ( target , 'addTransaction' ) ;
117+ cappedStoMock . mock ( 'pause' , Promise . resolve ( 'Pause' ) ) ;
118+
119119 // Real call
120120 await target . prepareTransactions ( ) ;
121121
122- // Verifications
122+ // Verifications\
123123 expect (
124124 addTransactionSpy . getCall ( 0 ) . calledWith ( cappedStoMock . getMockInstance ( ) . pause )
125125 ) . toEqual ( true ) ;
@@ -128,9 +128,12 @@ describe('PauseSto', () => {
128128 } ) ;
129129
130130 test ( 'should add the transaction to the queue to pause a usdTiered sto' , async ( ) => {
131+ moduleWrapperFactoryMock . mock ( 'getModuleInstance' , usdTieredStoMock . getMockInstance ( ) ) ;
131132 target = new PauseSto ( usdTieredParams , contextMock . getMockInstance ( ) ) ;
132133
133134 const addTransactionSpy = spy ( target , 'addTransaction' ) ;
135+ usdTieredStoMock . mock ( 'pause' , Promise . resolve ( 'Pause' ) ) ;
136+
134137 // Real call
135138 await target . prepareTransactions ( ) ;
136139
0 commit comments