From 4212e9e400ca821bb875cfb08f3802be1d5c31ea Mon Sep 17 00:00:00 2001
From: Thomas <4159106+dtslvr@users.noreply.github.com>
Date: Fri, 22 Jul 2022 14:56:44 +0200
Subject: [PATCH] Add tests for getInvestments()

---
 .../portfolio-calculator-baln-buy-and-sell.spec.ts         | 7 +++++++
 .../app/portfolio/portfolio-calculator-baln-buy.spec.ts    | 6 ++++++
 .../app/portfolio/portfolio-calculator-no-orders.spec.ts   | 4 ++++
 ...ortfolio-calculator-novn-buy-and-sell-partially.spec.ts | 7 +++++++
 4 files changed, 24 insertions(+)

diff --git a/apps/api/src/app/portfolio/portfolio-calculator-baln-buy-and-sell.spec.ts b/apps/api/src/app/portfolio/portfolio-calculator-baln-buy-and-sell.spec.ts
index e19616ab4..d826e1d0e 100644
--- a/apps/api/src/app/portfolio/portfolio-calculator-baln-buy-and-sell.spec.ts
+++ b/apps/api/src/app/portfolio/portfolio-calculator-baln-buy-and-sell.spec.ts
@@ -62,6 +62,8 @@ describe('PortfolioCalculator', () => {
         parseDate('2021-11-22')
       );
 
+      const investments = portfolioCalculator.getInvestments();
+
       const investmentsByMonth = portfolioCalculator.getInvestmentsByMonth();
 
       spy.mockRestore();
@@ -94,6 +96,11 @@ describe('PortfolioCalculator', () => {
         totalInvestment: new Big('0')
       });
 
+      expect(investments).toEqual([
+        { date: '2021-11-22', investment: new Big('285.8') },
+        { date: '2021-11-30', investment: new Big('0') }
+      ]);
+
       expect(investmentsByMonth).toEqual([
         { date: '2021-11-01', investment: new Big('12.6') }
       ]);
diff --git a/apps/api/src/app/portfolio/portfolio-calculator-baln-buy.spec.ts b/apps/api/src/app/portfolio/portfolio-calculator-baln-buy.spec.ts
index 5ce20cf57..b8cc6050a 100644
--- a/apps/api/src/app/portfolio/portfolio-calculator-baln-buy.spec.ts
+++ b/apps/api/src/app/portfolio/portfolio-calculator-baln-buy.spec.ts
@@ -51,6 +51,8 @@ describe('PortfolioCalculator', () => {
         parseDate('2021-11-30')
       );
 
+      const investments = portfolioCalculator.getInvestments();
+
       const investmentsByMonth = portfolioCalculator.getInvestmentsByMonth();
 
       spy.mockRestore();
@@ -83,6 +85,10 @@ describe('PortfolioCalculator', () => {
         totalInvestment: new Big('273.2')
       });
 
+      expect(investments).toEqual([
+        { date: '2021-11-30', investment: new Big('273.2') }
+      ]);
+
       expect(investmentsByMonth).toEqual([
         { date: '2021-11-01', investment: new Big('273.2') }
       ]);
diff --git a/apps/api/src/app/portfolio/portfolio-calculator-no-orders.spec.ts b/apps/api/src/app/portfolio/portfolio-calculator-no-orders.spec.ts
index c10037dda..32935a20e 100644
--- a/apps/api/src/app/portfolio/portfolio-calculator-no-orders.spec.ts
+++ b/apps/api/src/app/portfolio/portfolio-calculator-no-orders.spec.ts
@@ -39,6 +39,8 @@ describe('PortfolioCalculator', () => {
         new Date()
       );
 
+      const investments = portfolioCalculator.getInvestments();
+
       const investmentsByMonth = portfolioCalculator.getInvestmentsByMonth();
 
       spy.mockRestore();
@@ -54,6 +56,8 @@ describe('PortfolioCalculator', () => {
         totalInvestment: new Big(0)
       });
 
+      expect(investments).toEqual([]);
+
       expect(investmentsByMonth).toEqual([]);
     });
   });
diff --git a/apps/api/src/app/portfolio/portfolio-calculator-novn-buy-and-sell-partially.spec.ts b/apps/api/src/app/portfolio/portfolio-calculator-novn-buy-and-sell-partially.spec.ts
index ef0bdf98c..745dc9c39 100644
--- a/apps/api/src/app/portfolio/portfolio-calculator-novn-buy-and-sell-partially.spec.ts
+++ b/apps/api/src/app/portfolio/portfolio-calculator-novn-buy-and-sell-partially.spec.ts
@@ -62,6 +62,8 @@ describe('PortfolioCalculator', () => {
         parseDate('2022-03-07')
       );
 
+      const investments = portfolioCalculator.getInvestments();
+
       const investmentsByMonth = portfolioCalculator.getInvestmentsByMonth();
 
       spy.mockRestore();
@@ -94,6 +96,11 @@ describe('PortfolioCalculator', () => {
         totalInvestment: new Big('75.80')
       });
 
+      expect(investments).toEqual([
+        { date: '2022-03-07', investment: new Big('151.6') },
+        { date: '2022-04-08', investment: new Big('75.8') }
+      ]);
+
       expect(investmentsByMonth).toEqual([
         { date: '2022-03-01', investment: new Big('151.6') },
         { date: '2022-04-01', investment: new Big('-85.73') }