implement logic in Portfolio class

einaskoi · Feb 23, 2026 · 0bafdcb · 0bafdcb
1 parent 8067564
commit 0bafdcb
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/main/java/edu/ntnu/idi/idatt2003/gruppe42/Model/Portfolio.java b/src/main/java/edu/ntnu/idi/idatt2003/gruppe42/Model/Portfolio.java
@@ -23,7 +23,11 @@ public Portfolio() {
    * @return true or false based on if the operation was successful or not.
    */
   public boolean addShare(Share share) {
-    return true;
+    if (share == null) {
+      return false;
+    } else {
+      return shares.add(share);
+    }
   }
 
   /**
@@ -32,7 +36,11 @@ public boolean addShare(Share share) {
    * @return true or false based on if the operation was successful or not.
    */
   public boolean removeShare(Share share) {
-    return false;
+    if (share == null || !shares.contains(share)) {
+      return false;
+    } else {
+      return shares.remove(share);
+    }
   }
 
   public List<Share> getShares() {
@@ -45,6 +53,6 @@ public List<Share> getShares() {
    * @return true or false based on if the share was found or not.
    */
   public boolean contains(Share share) {
-    return true;
+    return share != null && shares.contains(share);
   }
 }