fix: entity tests were broken from recent changes

BREAKING CHANGE: security token now has many more variables
PolymathNetwork · Dec 16, 2019 · fb9d3cf · fb9d3cf
1 parent ef552c0
commit fb9d3cf
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/src/entities/__tests__/SecurityToken.ts b/src/entities/__tests__/SecurityToken.ts
@@ -1,3 +1,4 @@
+import { BigNumber } from '@polymathnetwork/contract-wrappers';
 import { SecurityToken } from '../SecurityToken';
 import { Context } from '../../Context';
 
@@ -6,13 +7,25 @@ const params1 = {
   name: 'Test Token 1',
   address: '0x1',
   owner: '0x3',
+  tokenDetails: 'details',
+  version: [new BigNumber(1)],
+  granularity: 3,
+  totalSupply: new BigNumber(1000),
+  currentCheckpoint: 2,
+  treasuryWallet: '0x3',
 };
 
 const params2 = {
   symbol: 'TEST2',
   name: 'Test Token 2',
   address: '0x2',
   owner: '0x4',
+  tokenDetails: 'details2',
+  version: [new BigNumber(1)],
+  granularity: 3,
+  totalSupply: new BigNumber(1000),
+  currentCheckpoint: 2,
+  treasuryWallet: '0x3',
 };
 
 const context = {} as Context;

diff --git a/src/entities/__tests__/Shareholders.ts b/src/entities/__tests__/Shareholders.ts
@@ -1,17 +1,23 @@
 import * as sinon from 'sinon';
 import { ImportMock, MockManager } from 'ts-mock-imports';
+import { BigNumber } from '@polymathnetwork/contract-wrappers';
 import * as contextModule from '../../Context';
 import * as createCheckpointProcedure from '../../procedures/CreateCheckpoint';
-
-import { Shareholders } from '~/entities/SecurityToken/Shareholders';
-import { SubModule } from '~/entities/SecurityToken/SubModule';
+import { Shareholders } from '../../entities/SecurityToken/Shareholders';
+import { SubModule } from '../../entities/SecurityToken/SubModule';
 import { SecurityToken } from '../SecurityToken';
 
 const params1 = {
   symbol: 'TEST1',
   name: 'Test Token 1',
   address: '0x1',
   owner: '0x3',
+  tokenDetails: 'details',
+  version: [new BigNumber(1)],
+  granularity: 3,
+  totalSupply: new BigNumber(1000),
+  currentCheckpoint: 2,
+  treasuryWallet: '0x3',
 };
 
 describe('Shareholders', () => {