99 lines
3.9 KiB
TypeScript
99 lines
3.9 KiB
TypeScript
module TK.SpaceTac {
|
|
// unit testing utilities
|
|
export class TestTools {
|
|
|
|
// Create a battle between two fleets, with a fixed play order (owned ships, then enemy ships)
|
|
static createBattle(own_ships: number, enemy_ships: number): Battle {
|
|
var fleet1 = new Fleet();
|
|
var fleet2 = new Fleet();
|
|
|
|
while (own_ships--) {
|
|
fleet1.addShip(new Ship());
|
|
}
|
|
while (enemy_ships--) {
|
|
fleet2.addShip(new Ship());
|
|
}
|
|
|
|
var battle = new Battle(fleet1, fleet2);
|
|
battle.play_order = fleet1.ships.concat(fleet2.ships);
|
|
battle.advanceToNextShip();
|
|
return battle;
|
|
}
|
|
|
|
// Get or add an equipment of a given slot type
|
|
static getOrGenEquipment(ship: Ship, slot: SlotType, template: LootTemplate, force_generate = false): Equipment {
|
|
var equipped = ship.listEquipment(slot);
|
|
var equipment: Equipment;
|
|
if (force_generate || equipped.length === 0) {
|
|
equipment = template.generate(1);
|
|
equipment.requirements = {};
|
|
ship.addSlot(slot).attach(equipment);
|
|
} else {
|
|
equipment = equipped[0];
|
|
}
|
|
|
|
return equipment;
|
|
}
|
|
|
|
/**
|
|
* Add an engine, allowing a ship to move *distance*, for each action points
|
|
*/
|
|
static addEngine(ship: Ship, distance: number): Equipment {
|
|
let equipment = ship.addSlot(SlotType.Engine).attach(new Equipment(SlotType.Engine));
|
|
equipment.action = new MoveAction(equipment, distance);
|
|
return equipment;
|
|
}
|
|
|
|
/**
|
|
* Add a weapon to a ship
|
|
*/
|
|
static addWeapon(ship: Ship, damage = 100, power_usage = 1, max_distance = 100, blast = 0, angle = 0): Equipment {
|
|
var equipment = ship.addSlot(SlotType.Weapon).attach(new Equipment(SlotType.Weapon));
|
|
equipment.action = new TriggerAction(equipment, [new DamageEffect(damage)], power_usage, max_distance, blast, angle, "Test Weapon");
|
|
return equipment;
|
|
}
|
|
|
|
// Set a ship action points, adding/updating an equipment if needed
|
|
static setShipAP(ship: Ship, points: number, recovery: number = 0): void {
|
|
var equipment = this.getOrGenEquipment(ship, SlotType.Power, new Equipments.NuclearReactor());
|
|
|
|
equipment.effects.forEach(effect => {
|
|
if (effect instanceof AttributeEffect) {
|
|
if (effect.attrcode === "power_capacity") {
|
|
effect.value = points;
|
|
} else if (effect.attrcode === "power_generation") {
|
|
effect.value = recovery;
|
|
}
|
|
}
|
|
});
|
|
|
|
ship.updateAttributes();
|
|
ship.setValue("power", points);
|
|
}
|
|
|
|
// Set a ship hull and shield points, adding/updating an equipment if needed
|
|
static setShipHP(ship: Ship, hull_points: number, shield_points: number): void {
|
|
var armor = TestTools.getOrGenEquipment(ship, SlotType.Hull, new Equipments.IronHull());
|
|
var shield = TestTools.getOrGenEquipment(ship, SlotType.Shield, new Equipments.ForceField());
|
|
|
|
armor.effects.forEach(effect => {
|
|
if (effect instanceof AttributeEffect) {
|
|
if (effect.attrcode === "hull_capacity") {
|
|
effect.value = hull_points;
|
|
}
|
|
}
|
|
});
|
|
shield.effects.forEach((effect: BaseEffect) => {
|
|
if (effect instanceof AttributeEffect) {
|
|
if (effect.attrcode === "shield_capacity") {
|
|
effect.value = shield_points;
|
|
}
|
|
}
|
|
});
|
|
|
|
ship.updateAttributes();
|
|
ship.restoreHealth();
|
|
}
|
|
}
|
|
}
|