fix: entity tests for version change

PolymathNetwork · Dec 17, 2019 · ee52431 · ee52431
1 parent 8f0e914
commit ee52431
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/entities/__tests__/SecurityToken.ts b/src/entities/__tests__/SecurityToken.ts
@@ -1,14 +1,15 @@
 import { BigNumber } from '@polymathnetwork/contract-wrappers';
 import { SecurityToken } from '../SecurityToken';
 import { Context } from '../../Context';
+import { Version } from '~/types';
 
 const params1 = {
   symbol: 'TEST1',
   name: 'Test Token 1',
   address: '0x1',
   owner: '0x3',
   tokenDetails: 'details',
-  version: [new BigNumber(1)],
+  version: Version.V3_1_0,
   granularity: 3,
   totalSupply: new BigNumber(1000),
   currentCheckpoint: 2,

diff --git a/src/entities/__tests__/Shareholders.ts b/src/entities/__tests__/Shareholders.ts
@@ -6,14 +6,15 @@ import * as createCheckpointProcedure from '../../procedures/CreateCheckpoint';
 import { Shareholders } from '../../entities/SecurityToken/Shareholders';
 import { SubModule } from '../../entities/SecurityToken/SubModule';
 import { SecurityToken } from '../SecurityToken';
+import { Version } from '../../types';
 
 const params1 = {
   symbol: 'TEST1',
   name: 'Test Token 1',
   address: '0x1',
   owner: '0x3',
   tokenDetails: 'details',
-  version: [new BigNumber(1)],
+  version: Version.V3_1_0,
   granularity: 3,
   totalSupply: new BigNumber(1000),
   currentCheckpoint: 2,