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