refactor: add prettier, rectify procedureTypes without hardcode string

PolymathNetwork · Nov 21, 2019 · c7a5889 · c7a5889
1 parent 35bfeb8
commit c7a5889
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 8 deletions.
diff --git a/src/procedures/__tests__/AssignSecurityTokenRole.ts b/src/procedures/__tests__/AssignSecurityTokenRole.ts
@@ -8,7 +8,13 @@ import * as tokenFactoryModule from '../../testUtils/MockedTokenFactoryModule';
 import { AssignSecurityTokenRole } from '../../procedures/AssignSecurityTokenRole';
 import { Procedure } from '../Procedure';
 import { PolymathError } from '../../PolymathError';
-import { ErrorCode, Feature, PolyTransactionTag, SecurityTokenRole } from '../../types';
+import {
+  ErrorCode,
+  Feature,
+  PolyTransactionTag,
+  ProcedureType,
+  SecurityTokenRole,
+} from '../../types';
 import * as securityTokenFactoryModule from '../../entities/factories/SecurityTokenFactory';
 import { mockFactories } from '../../testUtils/mockFactories';
 
@@ -76,7 +82,7 @@ describe('AssignSecurityTokenRole', () => {
   describe('Types', () => {
     test('should extend procedure and have AssignSecurityTokenRole type', async () => {
       expect(target instanceof Procedure).toBe(true);
-      expect(target.type).toBe('AssignSecurityTokenRole');
+      expect(target.type).toBe(ProcedureType.AssignSecurityTokenRole);
     });
   });
 

diff --git a/src/procedures/__tests__/AssignStoRole.ts b/src/procedures/__tests__/AssignStoRole.ts
@@ -7,7 +7,7 @@ import * as tokenFactoryModule from '../../testUtils/MockedTokenFactoryModule';
 import { AssignStoRole } from '../../procedures/AssignStoRole';
 import { Procedure } from '../Procedure';
 import { PolymathError } from '../../PolymathError';
-import { ErrorCode, PolyTransactionTag, StoRole } from '../../types';
+import { ErrorCode, PolyTransactionTag, ProcedureType, StoRole } from '../../types';
 
 const params = {
   symbol: 'TEST1',
@@ -46,7 +46,7 @@ describe('AssignStoRole', () => {
   describe('Types', () => {
     test('should extend procedure and have AssignStoRole type', async () => {
       expect(target instanceof Procedure).toBe(true);
-      expect(target.type).toBe('AssignStoRole');
+      expect(target.type).toBe(ProcedureType.AssignStoRole);
     });
   });
 

diff --git a/src/procedures/__tests__/EnableDividendManagers.ts b/src/procedures/__tests__/EnableDividendManagers.ts
@@ -8,7 +8,7 @@ import * as tokenFactoryModule from '../../testUtils/MockedTokenFactoryModule';
 import { EnableDividendManagers } from '../../procedures/EnableDividendManagers';
 import { Procedure } from '../Procedure';
 import { PolymathError } from '../../PolymathError';
-import { ErrorCode, PolyTransactionTag } from '../../types';
+import { ErrorCode, PolyTransactionTag, ProcedureType } from '../../types';
 
 const params = {
   symbol: 'TEST1',
@@ -57,7 +57,7 @@ describe('EnableDividendManagers', () => {
   describe('Types', () => {
     test('should extend procedure and have EnableDividendManagers type', async () => {
       expect(target instanceof Procedure).toBe(true);
-      expect(target.type).toBe('EnableDividendManagers');
+      expect(target.type).toBe(ProcedureType.EnableDividendManagers);
     });
   });
 

diff --git a/src/procedures/__tests__/EnableGeneralPermissionManager.ts b/src/procedures/__tests__/EnableGeneralPermissionManager.ts
@@ -8,7 +8,7 @@ import * as tokenFactoryModule from '../../testUtils/MockedTokenFactoryModule';
 import { EnableGeneralPermissionManager } from '../../procedures/EnableGeneralPermissionManager';
 import { Procedure } from '../Procedure';
 import { PolymathError } from '../../PolymathError';
-import { ErrorCode, PolyTransactionTag } from '../../types';
+import { ErrorCode, PolyTransactionTag, ProcedureType } from '../../types';
 
 const params = {
   symbol: 'TEST1',
@@ -56,7 +56,7 @@ describe('EnableGeneralPermissionManager', () => {
   describe('Types', () => {
     test('should extend procedure and have EnableGeneralPermissionManager type', async () => {
       expect(target instanceof Procedure).toBe(true);
-      expect(target.type).toBe('EnableGeneralPermissionManager');
+      expect(target.type).toBe(ProcedureType.EnableGeneralPermissionManager);
     });
   });
 

diff --git a/src/procedures/__tests__/PauseSto.ts b/src/procedures/__tests__/PauseSto.ts
@@ -100,6 +100,7 @@ describe('PauseSto', () => {
     // Instantiate PauseSto
     target = new PauseSto(cappedParams, contextMock.getMockInstance());
   });
+
   afterEach(() => {
     restore();
   });