fix: export resolver method

PolymathNetwork · Nov 5, 2019 · a8e8fd9 · a8e8fd9
1 parent 5568ee2
commit a8e8fd9
Showing 1 changed file with 31 additions and 23 deletions.
diff --git a/src/procedures/SetDividendsWallet.ts b/src/procedures/SetDividendsWallet.ts
@@ -12,12 +12,42 @@ import {
   DividendType,
 } from '../types';
 import { PolymathError } from '../PolymathError';
+import { Factories } from '../Context';
 import {
   SecurityToken,
   Erc20DividendsManager,
   EthDividendsManager,
 } from '../entities';
 
+export const createResolver = async (
+  dividendType: DividendType,
+  factories: Factories,
+  symbol: string
+) => {
+  let refresh;
+  // eslint-disable-next-line default-case
+  switch (dividendType) {
+    case DividendType.Erc20: {
+      refresh = factories.erc20DividendsManagerFactory.refresh(
+        Erc20DividendsManager.generateId({
+          securityTokenId: SecurityToken.generateId({ symbol }),
+          dividendType,
+        })
+      );
+      break;
+    }
+    case DividendType.Eth: {
+      refresh = factories.ethDividendsManagerFactory.refresh(
+        EthDividendsManager.generateId({
+          securityTokenId: SecurityToken.generateId({ symbol }),
+          dividendType,
+        })
+      );
+    }
+  }
+  return refresh;
+};
+
 export class SetDividendsWallet extends Procedure<
   SetDividendsWalletProcedureArgs
 > {
@@ -72,29 +102,7 @@ export class SetDividendsWallet extends Procedure<
 
     await this.addTransaction(dividendModule.changeWallet, {
       tag: PolyTransactionTag.SetDividendsWallet,
-      resolver: async () => {
-        switch (dividendType) {
-          case DividendType.Erc20: {
-            return factories.erc20DividendsManagerFactory.refresh(
-              Erc20DividendsManager.generateId({
-                securityTokenId: SecurityToken.generateId({ symbol }),
-                dividendType,
-              })
-            );
-          }
-          case DividendType.Eth: {
-            return factories.ethDividendsManagerFactory.refresh(
-              EthDividendsManager.generateId({
-                securityTokenId: SecurityToken.generateId({ symbol }),
-                dividendType,
-              })
-            );
-          }
-          default: {
-            break;
-          }
-        }
-      },
+      resolver: () => createResolver(dividendType, factories, symbol),
     })({ wallet: address });
   }
 }