package hu.bme.mit.spaceship;

import static org.junit.Assert.*;
import static org.mockito.Mockito.*;

import org.junit.Before;
import org.junit.Test;

public class GT4500Test {

  private GT4500 ship;

  @Before
  public void init(){
    mock1 = mock(TorpedoStore.class);
    mock2 = mock(TorpedoStore.class);
    this.ship = new GT4500(mock1, mock2);
  }

  @Test
  public void fireTorpedos_Single_Success(){
    // Arrange

    // Act
    boolean result = ship.fireTorpedos(FiringMode.SINGLE);

    // Assert
    assertEquals(true, result);
  }

  @Test
  public void fireTorpedos_All_Success(){
    // Arrange

    // Act
    boolean result = ship.fireTorpedos(FiringMode.ALL);

    // Assert
    assertEquals(true, result);
  }

}

@@ -236,16 +236,36 @@
     
-mock1 = 
+%0A    this.ship = new GT4500(
 mock
@@ -288,21 +288,9 @@
 ass)
-;%0A    mock2 =
+,
  moc
@@ -314,49 +314,8 @@
 ass)
-;%0A    this.ship = new GT4500(mock1, mock2
 );%0A 

Update GT4500Test.java