Added PortfolioTest class

solvena · Mar 10, 2026 · 2aa2af0 · 2aa2af0
1 parent 32b5cb7
commit 2aa2af0
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/src/test/java/PortfolioTest.java b/src/test/java/PortfolioTest.java
@@ -0,0 +1,37 @@
+import static org.junit.jupiter.api.Assertions.*;
+
+import org.junit.jupiter.api.Test;
+import java.math.BigDecimal;
+
+public class PortfolioTest {
+
+    @Test
+    void testAddShare() {
+        Portfolio portfolio = new Portfolio();
+
+        Stock stock = new Stock("AAPL", "Apple", new BigDecimal("150"));
+        Share share = new Share(stock, new BigDecimal("10"), new BigDecimal("140"));
+
+        boolean result = portfolio.addShare(share);
+
+        assertTrue(result);
+        assertTrue(portfolio.contains(share));
+    }
+
+    @Test
+    void testRemoveShare() {
+        Portfolio portfolio = new Portfolio();
+
+        Stock stock = new Stock("AAPL", "Apple", new BigDecimal("150"));
+        Share share = new Share(stock, new BigDecimal("10"), new BigDecimal("140"));
+
+        portfolio.addShare(share);
+
+        boolean result = portfolio.removeShare(share);
+
+        assertTrue(result);
+        assertFalse(portfolio.contains(share));
+    }
+
+
+}