+ laser tests

ftsrg-edu · Apr 21, 2017 · 56da240 · 56da240
1 parent c7aea5c
commit 56da240
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 2 deletions.
diff --git a/src/main/java/hu/bme/mit/spaceship/GT4500.java b/src/main/java/hu/bme/mit/spaceship/GT4500.java
@@ -96,12 +96,12 @@ public boolean fireLasers(FiringMode firingMode) {
     if(firingMode == FiringMode.SINGLE)
     {
       //no lasers
-      firingSuccess = false;
+      firingSuccess = true;
     }
     else if(firingMode == FiringMode.ALL)
     {
       //no lasrs
-      firingSuccess = false;
+      firingSuccess = true;
     }
 
 

diff --git a/src/test/java/hu/bme/mit/spaceship/GT4500Test.java b/src/test/java/hu/bme/mit/spaceship/GT4500Test.java
@@ -52,4 +52,33 @@ public void fireTorpedos_All_Success(){
     assertEquals(true, result);
   }
 
+  public void fireLasers_Single_Success(){
+    // Arrange
+
+    when(mts.fire(1)).thenReturn(true);
+    when(mts.isEmpty()).thenReturn(false);
+    when(mts.getNumberOfTorpedos()).thenReturn(1);
+
+
+    // Assert
+    assertEquals(true, ship.fireLasers(FiringMode.SINGLE));
+  }
+
+  @Test
+  public void fireLasers_All_Success(){
+    // Arrange
+    when(mts.fire(1)).thenReturn(true);
+    when(mts.isEmpty()).thenReturn(false);
+    when(mts.getNumberOfTorpedos()).thenReturn(1);
+    when(mts2.fire(1)).thenReturn(true);
+    when(mts2.isEmpty()).thenReturn(false);
+    when(mts2.getNumberOfTorpedos()).thenReturn(1);
+
+    // Act
+    boolean result = ship.fireLasers(FiringMode.ALL);
+
+    // Assert
+    assertEquals(true, result);
+  }
+
 }