Updated slot machine and tests

src/main/java/io/zipcoder/casino/Casino.java

@@ -72,7 +72,7 @@ public class Casino {
                         Printer.printMessage("Invalid bet amount entered, min bet for slot is $1 and max $100 ");
                         break;
                     }
-                    Game slot= new SlotMachine(slotBet1);
+                    Game slot= new SlotMachine(slotBet1, player);
                     game = slot;
                     slot.startGame();
                     ((SlotMachine) slot).slotResult();

src/main/java/io/zipcoder/casino/SlotMachine.java

@@ -12,18 +12,23 @@ public class SlotMachine implements Game, Gamble {
     String word2="";
     String word3="";
     double totalBet=0;
+    Player player;
+
+    public SlotMachine(int betAmount, Player player) {
 
-    public SlotMachine(int betAmount) {
         this.betAmount = betAmount;
+        this.player = player;
     }
 
     @Override
     public void bet(int betAmount) {
-        this.betAmount= betAmount;
+        player.changeBalance(-betAmount);
     }
 
     public void payout(){
+        getPlayer().changeBalance(payoutAmt);
         Printer.printMessage("Your payout amount for slot machine is: $" + payoutAmt + "\n");
+
     }
 
     @Override
@@ -42,9 +47,11 @@ public class SlotMachine implements Game, Gamble {
             e.printStackTrace();
         }
 
+        generateWords();
 
-        outputword = "";
+    }
 
+    public void generateWords() {
         Random rand = new Random();
 
         for (int i = 1; i <= 3; i++) {
@@ -79,13 +86,11 @@ public class SlotMachine implements Game, Gamble {
                 word3 = word;
             }
         }
-
+        outputword= "[ " + word1+ " ]" + "   " + "[ " + word2 + " ]" + "   "+ "[ " + word3 + " ]" + "\n" ;
     }
 
     public void slotResult()
     {
-            outputword= "[ " + word1+ " ]" + "   " + "[ " + word2 + " ]" + "   "+ "[ " + word3 + " ]" + "\n" ;
-
             if(((!word1.equals(word2)) )&& ((!word1.equals(word3))) && ((!word2.equals(word3)))){
 
                 outputword= outputword + "\n"+" You have won $0";
@@ -98,7 +103,7 @@ public class SlotMachine implements Game, Gamble {
             }
 
 
-            else if(word1.equals(word2) && word1.equals(word3)  /*&& ((word2.equals(word1)) && (word2.equals(word3))) && ( (word3.equals(word1)) && (word3.equals(word2)))*/){
+            else if(word1.equals(word2) && word1.equals(word3)) {
                 outputword= outputword + "\n" + "You have won $" + (betAmount*3);
                 payoutAmt=betAmount*3;
             }
@@ -130,5 +135,14 @@ public class SlotMachine implements Game, Gamble {
     public void setWord3(String word3) {
         this.word3 = word3;
     }
+
+    public String getOutputword() {
+        return outputword;
+    }
+
+    public Player getPlayer() {
+        return player;
+    }
+
 }
 

src/test/java/io/zipcoder/casino/SlotTest.java

@@ -7,6 +7,8 @@ import org.junit.Assert;
 
 public class SlotTest {
 
+    Player player = new Player("Bob", 400);
+
     @Test
     public void testSlotResult1(){
         int betAmount=10;
@@ -14,7 +16,7 @@ public class SlotTest {
         String word2="MOUSE";
         String word3="MOUSE";
         //given
-        SlotMachine slotmachine = new SlotMachine(betAmount);
+        SlotMachine slotmachine = new SlotMachine(betAmount, player);
         slotmachine.setWord1(word1);
         slotmachine.setWord2(word2);
         slotmachine.setWord3(word3);
@@ -34,7 +36,7 @@ public class SlotTest {
 
 
         //given
-        SlotMachine slotmachine = new SlotMachine(betAmount);
+        SlotMachine slotmachine = new SlotMachine(betAmount, player);
 
         slotmachine.setWord1(word1);
         slotmachine.setWord2(word2);
@@ -54,7 +56,7 @@ public class SlotTest {
         String word3="CAT";
 
         //given
-        SlotMachine slotmachine = new SlotMachine(betAmount);
+        SlotMachine slotmachine = new SlotMachine(betAmount, player);
 
         slotmachine.setWord1(word1);
         slotmachine.setWord2(word2);
@@ -64,4 +66,36 @@ public class SlotTest {
         int payout=slotmachine.getPayoutAmt();
         Assert.assertEquals(0,payout);
     }
+
+    @Test
+    public void testGenerateWords() {
+        //Given
+        SlotMachine game = new SlotMachine(100, player);
+        String before = game.getOutputword();
+
+        //When
+        game.generateWords();
+        String after = game.getOutputword();
+
+        //Then
+        Assert.assertNotEquals(before, after);
+
+
+    }
+
+    @Test
+    public void testBet() {
+        //Given
+        SlotMachine game = new SlotMachine(100, player);
+        int expected = 300;
+
+        //When
+        game.bet(100);
+        int actual = game.getPlayer().getCurrentBalance();
+
+        //Then
+        Assert.assertEquals(expected, actual);
+
+
+    }
 }