import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Spy;
import static org.mockito.Mockito.*;
import org.testng.annotations.Test;
public class TestStubForReturnMethod extends BaseTeser{
@Mock
private Weapon weapon1;
@Spy
private Weapon weapon2;
@Mock
private Weapon weapon;
@InjectMocks
private Soldier soldier;
//@Test
public void test1(){
when(weapon1.getWeaponName()).thenReturn("Gun");
p(weapon1.getWeaponName());
}
//@Test
public void test2(){
when(weapon2.getWeaponName()).thenReturn("Gun");
p("===");
p(weapon2.getWeaponName());
}
//@Test
public void test3(){
doReturn("Gun").when(weapon2).getWeaponName();
p("===");
p(weapon2.getWeaponName());
}
@Test
public void test4(){
doReturn("Gun").when(weapon).getWeaponName();
p("===");
soldier.showWeapon();
}
}