From 65fb560b7fd1234bc7f0ca54b697cb6182f3bbce Mon Sep 17 00:00:00 2001
From: Thomas <4159106+dtslvr@users.noreply.github.com>
Date: Mon, 10 May 2021 21:02:32 +0200
Subject: [PATCH] Add mock functions

* get()
* initialize()
* toCurrency()
---
 apps/api/src/models/portfolio.spec.ts | 44 ++++++++++++++++++++++++---
 1 file changed, 39 insertions(+), 5 deletions(-)

diff --git a/apps/api/src/models/portfolio.spec.ts b/apps/api/src/models/portfolio.spec.ts
index c1006e8f9..aa4b207ce 100644
--- a/apps/api/src/models/portfolio.spec.ts
+++ b/apps/api/src/models/portfolio.spec.ts
@@ -18,18 +18,52 @@ jest.mock('../services/data-provider.service', () => {
   return {
     DataProviderService: jest.fn().mockImplementation(() => {
       const today = format(new Date(), 'yyyy-MM-dd');
+
       return {
-        getHistorical: () =>
-          Promise.resolve({
+        get: () => {
+          return Promise.resolve({
+            BTCUSD: {
+              currency: Currency.USD,
+              dataSource: DataSource.YAHOO,
+              marketPrice: 57973.008,
+              marketState: MarketState.open,
+              name: 'Bitcoin USD'
+            },
+            ETHUSD: {
+              currency: Currency.USD,
+              dataSource: DataSource.YAHOO,
+              marketPrice: 3915.337,
+              marketState: MarketState.open,
+              name: 'Ethereum USD'
+            }
+          });
+        },
+        getHistorical: () => {
+          return Promise.resolve({
             BTCUSD: {
-              today: 57973.008
+              [today]: 57973.008
+            },
+            ETHUSD: {
+              [today]: 3915.337
             }
-          })
+          });
+        }
       };
     })
   };
 });
-jest.mock('../services/exchange-rate-data.service');
+
+jest.mock('../services/exchange-rate-data.service', () => {
+  return {
+    ExchangeRateDataService: jest.fn().mockImplementation(() => {
+      return {
+        initialize: () => Promise.resolve(),
+        toCurrency: () => Promise.resolve(1)
+      };
+    })
+  };
+});
+
 jest.mock('../services/rules.service');
 
 const DEFAULT_ACCOUNT_ID = '693a834b-eb89-42c9-ae47-35196c25d269';