@@ -53,7 +53,7 @@ describe('PauseSto', () => {
53
53
let usdTieredStoFactoryMock : MockManager < usdTieredStoFactoryModule . UsdTieredStoFactory > ;
54
54
55
55
let securityTokenMock : MockManager < contractWrappersModule . SecurityToken_3_0_0 > ;
56
- let moduleFactoryMock : MockManager < contractWrappersModule . ModuleFactory_3_0_0 > ;
56
+
57
57
let factoryMockSetup : Factories ;
58
58
let securityTokenId : string ;
59
59
@@ -73,15 +73,11 @@ describe('PauseSto', () => {
73
73
74
74
securityTokenMock = ImportMock . mockClass ( contractWrappersModule , 'SecurityToken_3_0_0' ) ;
75
75
76
- moduleFactoryMock = ImportMock . mockClass ( contractWrappersModule , 'ModuleFactory_3_0_0' ) ;
77
-
78
76
tokenFactoryMock . mock (
79
77
'getSecurityTokenInstanceFromTicker' ,
80
78
securityTokenMock . getMockInstance ( )
81
79
) ;
82
80
83
- moduleWrapperFactoryMock . mock ( 'getModuleInstance' , moduleFactoryMock . getMockInstance ( ) ) ;
84
-
85
81
cappedStoFactoryMock = ImportMock . mockClass ( cappedStoFactoryModule , 'CappedStoFactory' ) ;
86
82
87
83
usdTieredStoFactoryMock = ImportMock . mockClass (
@@ -99,6 +95,8 @@ describe('PauseSto', () => {
99
95
100
96
securityTokenId = SecurityToken . generateId ( { symbol : cappedParams . symbol } ) ;
101
97
98
+ moduleWrapperFactoryMock . mock ( 'getModuleInstance' , cappedStoMock . getMockInstance ( ) ) ;
99
+
102
100
// Instantiate PauseSto
103
101
target = new PauseSto ( cappedParams , contextMock . getMockInstance ( ) ) ;
104
102
} ) ;
@@ -116,10 +114,12 @@ describe('PauseSto', () => {
116
114
describe ( 'PauseSto' , ( ) => {
117
115
test ( 'should add the transaction to the queue to pause a capped sto' , async ( ) => {
118
116
const addTransactionSpy = spy ( target , 'addTransaction' ) ;
117
+ cappedStoMock . mock ( 'pause' , Promise . resolve ( 'Pause' ) ) ;
118
+
119
119
// Real call
120
120
await target . prepareTransactions ( ) ;
121
121
122
- // Verifications
122
+ // Verifications\
123
123
expect (
124
124
addTransactionSpy . getCall ( 0 ) . calledWith ( cappedStoMock . getMockInstance ( ) . pause )
125
125
) . toEqual ( true ) ;
@@ -128,9 +128,12 @@ describe('PauseSto', () => {
128
128
} ) ;
129
129
130
130
test ( 'should add the transaction to the queue to pause a usdTiered sto' , async ( ) => {
131
+ moduleWrapperFactoryMock . mock ( 'getModuleInstance' , usdTieredStoMock . getMockInstance ( ) ) ;
131
132
target = new PauseSto ( usdTieredParams , contextMock . getMockInstance ( ) ) ;
132
133
133
134
const addTransactionSpy = spy ( target , 'addTransaction' ) ;
135
+ usdTieredStoMock . mock ( 'pause' , Promise . resolve ( 'Pause' ) ) ;
136
+
134
137
// Real call
135
138
await target . prepareTransactions ( ) ;
136
139
0 commit comments