test methods covered

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

@@ -2,7 +2,6 @@ package io.zipcoder.casino;
 
 
 import java.util.ArrayList;
-import java.util.InputMismatchException;
 
 public class Casino {
     public boolean running = true;

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

@@ -3,7 +3,7 @@ package io.zipcoder.casino;
 import java.util.Random;
 
 public class SlotMachine implements Game, Gamble {
-    private int betAmount;
+    protected int betAmount;
     private int payoutAmt=0;
     String word="";
     String outputword="";
@@ -19,6 +19,7 @@ public class SlotMachine implements Game, Gamble {
 
     @Override
     public void bet(int betAmount) {
+
         this.betAmount= betAmount;
     }
 
@@ -47,6 +48,11 @@ public class SlotMachine implements Game, Gamble {
 
         Random rand = new Random();
 
+        generateWords(rand);
+
+    }
+
+    protected void generateWords(Random rand) {
         for (int i = 1; i <= 3; i++) {
             int randnum = rand.nextInt(6);
 
@@ -79,7 +85,6 @@ public class SlotMachine implements Game, Gamble {
                 word3 = word;
             }
         }
-
     }
 
     public void slotResult()
@@ -117,9 +122,11 @@ public class SlotMachine implements Game, Gamble {
 
     }
     public int getPayoutAmt() {
+
         return payoutAmt;
     }
     public void setWord1(String word1) {
+
         this.word1 = word1;
     }
 

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

@@ -1,20 +1,23 @@
 package io.zipcoder.casino;
 
 
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.Assert;
 
+import java.util.Random;
+
 
 public class SlotTest {
+    private int betAmount = 10;
+    private SlotMachine slotmachine = new SlotMachine(betAmount);
 
     @Test
     public void testSlotResult1(){
-        int betAmount=10;
         String word1="MOUSE";
         String word2="MOUSE";
         String word3="MOUSE";
         //given
-        SlotMachine slotmachine = new SlotMachine(betAmount);
         slotmachine.setWord1(word1);
         slotmachine.setWord2(word2);
         slotmachine.setWord3(word3);
@@ -27,15 +30,12 @@ public class SlotTest {
 
     @Test
     public void testSlotResult2(){
-        int betAmount=10;
         String word1="MOUSE";
         String word2="MOUSE";
         String word3="CAT";
 
 
         //given
-        SlotMachine slotmachine = new SlotMachine(betAmount);
-
         slotmachine.setWord1(word1);
         slotmachine.setWord2(word2);
         slotmachine.setWord3(word3);
@@ -65,4 +65,35 @@ public class SlotTest {
         Assert.assertEquals(0,payout);
 
     }
+
+    @Test
+    public void testSetBetAmount(){
+        int betAmount=10;
+        SlotMachine slotmachine = new SlotMachine(betAmount);
+
+        int newBet = 40;
+
+        slotmachine.bet(newBet);
+
+        int actual = slotmachine.betAmount;
+
+        Assert.assertEquals(newBet, actual);
+
+    }
+
+    @Test
+    public void testGenerateWords(){
+        Random random = new Random(1);
+
+
+        String expectedWord1 = "SQUIRREL";
+        String expectedWord2 = "FISH";
+        String expectedWord3 = "CAT";
+
+        slotmachine.generateWords(random);
+
+        Assert.assertEquals(expectedWord1, slotmachine.word1);
+
+    }
+
 }