From de5a0f3e940a7461d359d4de09befcbf087c2a04 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Einar=20Sk=C3=B8ien?= <einaskoi@stud.ntnu.no>
Date: Wed, 25 Feb 2026 10:28:14 +0100
Subject: [PATCH] implement test for portfolio and fix small bug in Stock class

---
 .../idi/idatt2003/gruppe42/Model/Stock.java   |  2 +-
 .../idi/idatt2003/gruppe42/PortfolioTest.java | 41 +++++++++++++++++++
 2 files changed, 42 insertions(+), 1 deletion(-)
 create mode 100644 src/test/java/edu/ntnu/idi/idatt2003/gruppe42/PortfolioTest.java

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
index 084ac3a..bd4192c 100644
--- 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
new file mode 100644
index 0000000..5bb56d8
--- /dev/null
+++ 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));
+  }
+}