add test for portfolio value change

taveras · Apr 17, 2024 · 5263355 · 5263355
1 parent 8287314
commit 5263355
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/test/edu/berkeley/aep/PortfolioTest.java b/test/edu/berkeley/aep/PortfolioTest.java
@@ -18,4 +18,23 @@ public void shouldBeAbleToBuyStock() {
         p.buyStock(company, 1);
         assertEquals(50, p.currentValue());
     }
+    @Test
+    public void currentValueShouldIncreaseWhenStockIncreases() {
+        Portfolio p = new Portfolio();
+        Stock company = new Stock("XYZ", 50);
+
+        p.buyStock(company, 1);
+        company.increasePrice(5);
+        assertEquals(55, p.currentValue());
+    }
+
+    @Test
+    public void currentValueShouldDecreaseWhenStockIncreases() {
+        Portfolio p = new Portfolio();
+        Stock company = new Stock("XYZ", 50);
+
+        p.buyStock(company, 1);
+        company.decreasePrice(5);
+        assertEquals(45, p.currentValue());
+    }
 }