diff --git a/src/main/java/entities/CommonUser.java b/src/main/java/entities/CommonUser.java index e0551dd..017d7ff 100644 --- a/src/main/java/entities/CommonUser.java +++ b/src/main/java/entities/CommonUser.java @@ -82,4 +82,9 @@ public Double getBalance() { return balance; } + @Override + public void clearPortfolio() { + this.portfolio.clear(); + } + } diff --git a/src/main/java/entities/User.java b/src/main/java/entities/User.java index 46afb93..3153d05 100644 --- a/src/main/java/entities/User.java +++ b/src/main/java/entities/User.java @@ -16,4 +16,5 @@ public interface User { void setBalance(Double amount); void addBalance(Double addition); Double getBalance(); + void clearPortfolio(); } diff --git a/src/main/java/use_cases/ResetBalance/ResetBalanceInteractor.java b/src/main/java/use_cases/ResetBalance/ResetBalanceInteractor.java index 8937751..f84cc8e 100644 --- a/src/main/java/use_cases/ResetBalance/ResetBalanceInteractor.java +++ b/src/main/java/use_cases/ResetBalance/ResetBalanceInteractor.java @@ -25,6 +25,7 @@ public void execute() { Double curBalance = user.getBalance(); Double amountToAdd = 10000.0; user.setBalance(amountToAdd); + user.clearPortfolio(); // honestly, maybe consider adding a TopupTransaction HashMap userHistory = user.getHistory();