24 lines
522 B
Java
24 lines
522 B
Java
|
package units;
|
||
|
|
|
||
|
|
import abstractextensions.SoldierExtension;
|
||
|
|
import org.junit.Test;
|
||
|
|
|
||
|
|
import static org.junit.Assert.*;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Created by Srdjan on 03-May-17.
|
||
|
|
*/
|
||
|
|
public class SoldierUnitTest {
|
||
|
|
@Test
|
||
|
|
public void getUnitExtension() throws Exception {
|
||
|
|
|
||
|
|
final Unit unit = new SoldierUnit("SoldierUnitName");
|
||
|
|
|
||
|
|
assertNotNull((SoldierExtension)unit.getUnitExtension("SoldierExtension"));
|
||
|
|
assertNull(unit.getUnitExtension("SergeantExtension"));
|
||
|
|
assertNull(unit.getUnitExtension("CommanderExtension"));
|
||
|
|
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|