From 030a6f968fa64a95b9517d86e7c31fdba119885c Mon Sep 17 00:00:00 2001
From: andreafspeziale <andreaf.speziale@gmail.com>
Date: Fri, 22 Feb 2019 17:23:06 +0100
Subject: [PATCH] #20 Add get trading wallet asset balance test

---
 .../getTradingWalletBalanceCommand.test.js    | 88 +++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 tests/unit/commands/getTradingWalletBalanceCommand.test.js

diff --git a/tests/unit/commands/getTradingWalletBalanceCommand.test.js b/tests/unit/commands/getTradingWalletBalanceCommand.test.js
new file mode 100644
index 0000000..a8ea958
--- /dev/null
+++ b/tests/unit/commands/getTradingWalletBalanceCommand.test.js
@@ -0,0 +1,88 @@
+/* eslint-env node, jest */
+const sandbox = require('sinon').createSandbox()
+
+const { PrivateKeyService } = require('../../../src/services/PrivateKeyService')
+
+const logger = require('../../../src/logger')
+const TradingWalletServiceBuilder = require('../../../src/factories/TradingWalletServiceBuilder')
+const PrivateKeyValidator = require('../../../src/validators/PrivateKeyValidator')
+
+const GetTradingWalletBalanceCommandValidator = require('../../../src/validators/GetTradingWalletBalanceCommandValidator')
+const GetTradingWalletBalanceCommand = require('../../../src/commands/GetBalanceCommand')
+
+const getTradingWalletBalanceCommandValidator = new GetTradingWalletBalanceCommandValidator(logger)
+
+const tradingWalletService = TradingWalletServiceBuilder.build()
+
+const privateKeyValidator = new PrivateKeyValidator(logger)
+const privateKeyService = new PrivateKeyService(logger)
+
+const getTradingWalletBalanceCommand = new GetTradingWalletBalanceCommand(logger, tradingWalletService,
+  getTradingWalletBalanceCommandValidator, privateKeyService, privateKeyValidator)
+
+afterEach(() => {
+  sandbox.restore()
+})
+
+describe('tws get-balance', () => {
+  const from = '0x9c858489661158d1721a66319f8683925d5a8b70'
+  const to = '0x230cd1dc412c44bb95aa39018e2a2aed28ebadfc'
+  const token = '0x9727e2fb13f7f42d5a6f1a4a9877d4a7e0404d6a'
+
+  describe('should raise Error if from wallet address was not defined as expected', () => {
+    const invalidFromAddresses = ['NotValidEthereumAddress', [], {}, 21]
+    test.each(invalidFromAddresses)(`from 
+      is not a valid ethereum address: %o`, async (invalidFrom) => {
+      const expectedResult = [
+        {
+          code: 'ValidationError',
+          field: 'from',
+          message: 'from needs to be an ethereum address',
+        },
+      ]
+
+      const result = await getTradingWalletBalanceCommand
+        .executeAsync({ from: invalidFrom, to, token })
+
+      expect(result).toMatchObject(expectedResult)
+    })
+
+    test('from is not defined', async () => {
+      const from = undefined
+      const expectedResult = [{
+        code: 'ValidationError',
+        field: 'from',
+        message: 'from is required',
+      }]
+
+      const result = await getTradingWalletBalanceCommand.executeAsync({ from, to, token })
+      expect(result).toMatchObject(expectedResult)
+    })
+  })
+
+  describe('should execute GetTradingWalletBalanceCommand as expected', () => {
+    const draftValues = [false, undefined]
+    test.each(draftValues)('should return the tradingWallet asset balance with draft value = %o', async (draft) => {
+      const expectedTradinWalletAssetBalance = '1'
+      sandbox.stub(tradingWalletService.transactionLib.ethApiClient, 'transactionCallAsync')
+        .returns(expectedTradinWalletAssetBalance)
+
+      const result = await getTradingWalletBalanceCommand.executeAsync({ from, to, token, draft })
+      expect(result).toBe(expectedTradinWalletAssetBalance)
+    })
+
+    test('should return the transactionObjectDraft for tradingWallet asset balance', async () => {
+      const expectedTransactionObjectDraft = {
+        data: '0xf6b1b18b0000000000000000000000009727e2fb13f7f42d5a6f1a4a9877d4a7e0404d6a',
+        from,
+        to,
+        value: '0x0',
+      }
+
+      const result = await getTradingWalletBalanceCommand
+        .executeAsync({ from, to, token, draft: true })
+
+      expect(result).toMatchObject(expectedTransactionObjectDraft)
+    })
+  })
+})