Update Portfolio class with exceptions

solvena · May 24, 2026 · 3bb7de3 · 3bb7de3
1 parent bd8d34a
commit 3bb7de3
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/main/java/Model/Portfolio.java b/src/main/java/Model/Portfolio.java
@@ -12,10 +12,16 @@ public Portfolio() {
     }
 
     public boolean addShare(Share share) {
+        if (share == null) {
+            throw new IllegalArgumentException("Share cannot be null");
+        }
         return shares.add(share);
     }
 
     public boolean removeShare(Share share) {
+        if (share == null) {
+            throw new IllegalArgumentException("Share cannot be null");
+        }
         return shares.remove(share);
     }
 
@@ -24,6 +30,9 @@ public List<Share> getShares() {
     }
 
     public List<Share> getShares(String symbol) {
+        if (symbol == null || symbol.isBlank()) {
+            throw new IllegalArgumentException("Symbol cannot be null or blank");
+        }
         List<Share> result = new ArrayList<>();
         for (Share share : shares) {
             if (share.getStock().getSymbol().equals(symbol)) {
@@ -34,6 +43,9 @@ public List<Share> getShares(String symbol) {
     }
 
     public boolean contains(Share share) {
+        if (share == null) {
+            throw new IllegalArgumentException("Share cannot be null");
+        }
         return shares.contains(share);
     }