implement test for portfolio and fix small bug in Stock class

einaskoi · Feb 25, 2026 · de5a0f3 · de5a0f3
1 parent 4b688cb
commit de5a0f3
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 1 deletion.
diff --git a/src/main/java/edu/ntnu/idi/idatt2003/gruppe42/Model/Stock.java b/src/main/java/edu/ntnu/idi/idatt2003/gruppe42/Model/Stock.java
@@ -24,7 +24,7 @@ public String getCompany() {
   }
 
   public BigDecimal getSalesPrice() {
-    return prices.getLast();
+    return prices.get(prices.size() - 1);
   }
 
   public void addNewSalesPrices(BigDecimal salesPrice) {

diff --git a/src/test/java/edu/ntnu/idi/idatt2003/gruppe42/PortfolioTest.java b/src/test/java/edu/ntnu/idi/idatt2003/gruppe42/PortfolioTest.java
@@ -0,0 +1,41 @@
+package edu.ntnu.idi.idatt2003.gruppe42;
+
+import edu.ntnu.idi.idatt2003.gruppe42.Model.Portfolio;
+import edu.ntnu.idi.idatt2003.gruppe42.Model.Share;
+import edu.ntnu.idi.idatt2003.gruppe42.Model.Stock;
+import org.junit.jupiter.api.Test;
+
+import java.math.BigDecimal;
+
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+public class PortfolioTest {
+
+  @Test
+  public void addShareTest() {
+    Portfolio portfolio = new Portfolio();
+    Stock stock = new Stock("AAPL", "Apple", new BigDecimal("10000"));
+    Share share = new Share(stock, new BigDecimal("10"), new BigDecimal("15000"));
+
+    boolean result = portfolio.addShare(share);
+    assertTrue(portfolio.contains(share));
+  }
+
+  @Test
+  public void removeShareTest() {
+    Portfolio portfolio = new Portfolio();
+    Stock stock = new Stock("AAPL", "Apple", new BigDecimal("10000"));
+    Share share = new Share(stock, new BigDecimal("10"), new BigDecimal("15000"));
+
+    portfolio.addShare(share);
+    boolean result = portfolio.removeShare(share);
+    assertFalse(portfolio.contains(share));
+  }
+
+  @Test
+  void containsTest() {
+    Portfolio portfolio = new Portfolio();
+    assertFalse(portfolio.contains(null));
+  }
+}