DoneWithNaiveTicket

TicketMachine.java

@@ -39,6 +39,14 @@ public class TicketMachine
     }
 
     /**
+     * set the price
+     */
+    public void setPrice(int ticketCost){
+
+        price = ticketCost;
+    }
+
+    /**
      * Return the amount of money already inserted for the
      * next ticket.
      */
@@ -48,11 +56,54 @@ public class TicketMachine
     }
 
     /**
+     * return the total
+     */
+    public int getTotal(){
+
+        return total;
+    }
+
+    /**
+     * reduce the price by amount 
+     */
+    public void discount(int amount){
+        price -= amount;
+    }
+
+    /**
      * Receive an amount of money in cents from a customer.
      */
     public void insertMoney(int amount)
     {
-        balance = balance + amount;
+        if (amount > 0){
+            balance = balance + amount;
+        }
+        else{
+            System.out.println("Enter a positive amount: " + amount);
+        }
+    }
+
+    /**
+     * refund the balance
+     */
+    public int refundBalance(){
+        int amountToReturn = balance;
+        balance = 0;
+        return amountToReturn;
+    }
+
+    /**
+     * print a prompt
+     */
+    public void prompt(){
+        System.out.println("Please insert the correct amount of money");
+    }
+
+    /**
+     * show the price of the ticket
+     */
+    public void showPrice(){
+        System.out.println("The price of the ticket " + price + " cents"); 
     }
 
     /**
@@ -62,17 +113,32 @@ public class TicketMachine
      */
     public void printTicket()
     {
-        // Simulate the printing of a ticket.
-        System.out.println("##################");
-        System.out.println("# The BlueJ Line");
-        System.out.println("# Ticket");
-        System.out.println("# " + price + " cents.");
-        System.out.println("##################");
-        System.out.println();
+        int amountLeftToPay = price - balance;
+        if (balance >= price){
+            // Simulate the printing of a ticket.
+            System.out.println("##################");
+            System.out.println("# The BlueJ Line");
+            System.out.println("# Ticket");
+            System.out.println("# " + price + " cents.");
+            System.out.println("##################");
+            System.out.println();
 
-        // Update the total collected with the balance.
-        total = total + balance;
-        // Clear the balance.
-        balance = 0;
+            // Update the total collected with the balance.
+            total = total + price;
+            // Clear the balance.
+            balance = balance - price;
+        }
+        else{
+            System.out.println("You must enter atleast: " + (amountLeftToPay) + " cents");
+        }
+    }
+
+    /**
+     * emptying machine
+     */
+    public int emptyMachine(){
+        int moneyInMachine = total;
+        total = 0;
+        return moneyInMachine;
     }
 }