fix: add extra mocks to ensure that calledWith is valid

PolymathNetwork · Nov 20, 2019 · 47675d9 · 47675d9
1 parent 1774545
commit 47675d9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/procedures/__tests__/TransferErc20.ts b/src/procedures/__tests__/TransferErc20.ts
@@ -112,6 +112,7 @@ describe('TransferErc20', () => {
   describe('TransferErc20', () => {
     test('should add a transaction to the queue to transfer an erc20 token with specified token address to a specified receiving address', async () => {
       const addTransactionSpy = spy(target, 'addTransaction');
+      erc20Mock.mock('transfer', Promise.resolve('Transfer'));
       // Real call
       await target.prepareTransactions();
 
@@ -153,6 +154,8 @@ describe('TransferErc20', () => {
       wrappersMock.mock('isTestnet', Promise.resolve(true));
       polyTokenMock.mock('address', Promise.resolve(params.tokenAddress));
       erc20Mock.mock('balanceOf', Promise.resolve(new BigNumber(2)));
+      erc20Mock.mock('transfer', Promise.resolve('Transfer'));
+      wrappersMock.mock('getPolyTokens', Promise.resolve('GetPolyTokens'));
       const addTransactionSpy = spy(target, 'addTransaction');
       // Real call
       await target.prepareTransactions();