Skip to content

Commit 5ee5b02

Browse files
author
Victor Wiebe
committed
fix: bug in pauseSto, calledWith args, prettier
1 parent 4a311be commit 5ee5b02

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

src/procedures/__tests__/PauseSto.ts

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)