diff --git a/src/main/java/edu/ntnu/idi/idatt2003/g40/mappe/Exchange.java b/src/main/java/edu/ntnu/idi/idatt2003/g40/mappe/Exchange.java index 2083d6a..6276925 100644 --- a/src/main/java/edu/ntnu/idi/idatt2003/g40/mappe/Exchange.java +++ b/src/main/java/edu/ntnu/idi/idatt2003/g40/mappe/Exchange.java @@ -55,12 +55,14 @@ public Transaction buy(String symbol, BigDecimal quantity, Player player) { Stock stock = stockMap.get(symbol); Share share = new Share(stock, quantity, stock.getSalesPrice()); TransactionCalculator calculator = new PurchaseCalculator(share); - return new Purchase(share, week, calculator, player); + player.withdrawMoney(calculator.calculateTotal()); + return new Purchase(share, week, calculator); } public Transaction sell(Share share, Player player) { TransactionCalculator calculator = new SaleCalculator(share); - return new Sale(share, week, calculator, player); + player.addMoney(calculator.calculateTotal()); + return new Sale(share, week, calculator); } public void advance() {