feat: refresh st factory when there is a redemption

Author: Victor Wiebe

src/procedures/ControllerRedeem.ts

@@ -67,10 +67,21 @@ export class ControllerRedeem extends Procedure<ControllerRedeemProcedureArgs> {
      */
     await this.addTransaction(securityToken.controllerRedeem, {
       tag: PolyTransactionTag.ControllerRedeem,
-      resolvers: [createControllerRedeemResolver(factories, symbol, from)],
+      resolvers: [
+        createControllerRedeemResolver(factories, symbol, from),
+        refreshSecurityTokenFactoryResolver(factories, symbol),
+      ],
     })({ from, value: amount, data, operatorData: reason });
   }
 }
+
+export const refreshSecurityTokenFactoryResolver = (
+  factories: Factories,
+  symbol: string
+) => async () => {
+  return factories.securityTokenFactory.refresh(SecurityToken.generateId({ symbol }));
+};
+
 export const createControllerRedeemResolver = (
   factories: Factories,
   symbol: string,

src/procedures/__tests__/ControllerRedeem.ts

@@ -18,6 +18,7 @@ import {
 } from '../../types';
 import { mockFactories } from '../../testUtils/mockFactories';
 import * as shareholderFactoryModule from '../../entities/factories/ShareholderFactory';
+import * as securityTokenFactoryModule from '../../entities/factories/SecurityTokenFactory';
 import { Factories } from '../../Context';
 import { SecurityToken, Shareholder } from '../../entities';
 
@@ -37,6 +38,7 @@ describe('ControllerRedeem', () => {
   let tokenFactoryMock: MockManager<tokenFactoryModule.MockedTokenFactoryModule>;
   let securityTokenMock: MockManager<contractWrappersModule.SecurityToken_3_0_0>;
   let shareholderFactoryMock: MockManager<shareholderFactoryModule.ShareholderFactory>;
+  let securityTokenFactoryMock: MockManager<securityTokenFactoryModule.SecurityTokenFactory>;
   let factoriesMockedSetup: Factories;
 
   beforeEach(() => {
@@ -58,8 +60,13 @@ describe('ControllerRedeem', () => {
     contextMock.set('contractWrappers', wrappersMock.getMockInstance());
     wrappersMock.set('tokenFactory', tokenFactoryMock.getMockInstance());
     shareholderFactoryMock = ImportMock.mockClass(shareholderFactoryModule, 'ShareholderFactory');
+    securityTokenFactoryMock = ImportMock.mockClass(
+      securityTokenFactoryModule,
+      'SecurityTokenFactory'
+    );
     factoriesMockedSetup = mockFactories();
     factoriesMockedSetup.shareholderFactory = shareholderFactoryMock.getMockInstance();
+    factoriesMockedSetup.securityTokenFactory = securityTokenFactoryMock.getMockInstance();
     contextMock.set('factories', factoriesMockedSetup);
 
     // Instantiate ControllerRedeem
@@ -166,4 +173,19 @@ describe('ControllerRedeem', () => {
     expect(resolverValue).toEqual(undefined);
     expect(refreshStub.callCount).toEqual(1);
   });
+
+  test('should refresh the security token factory with resolver', async () => {
+    const refreshStub = securityTokenFactoryMock.mock('refresh', Promise.resolve(undefined));
+
+    const resolverValue = await controllerRedeemModule.refreshSecurityTokenFactoryResolver(
+      factoriesMockedSetup,
+      params.symbol
+    )();
+
+    expect(
+      refreshStub.getCall(0).calledWithExactly(SecurityToken.generateId({ symbol: params.symbol }))
+    ).toEqual(true);
+    expect(resolverValue).toEqual(undefined);
+    expect(refreshStub.callCount).toEqual(1);
+  });
 });