public void Weapon_UpgradeWeapon_Support()
{
Weapon w = new WeaponKnife();
w = w.Upgrade(CharacterClass.Support);
Assert.IsTrue(w is WeaponShotgun);
w = w.Upgrade(CharacterClass.Support);
Assert.IsTrue(w is WeaponAdvancedShotgun);
w = w.Upgrade(CharacterClass.Support);
Assert.IsTrue(w is WeaponAdvancedShotgun);
}
public void Weapon_UpgradeWeapon_Tank()
{
Weapon w = new WeaponKnife();
w = w.Upgrade(CharacterClass.Tank);
Assert.IsTrue(w is WeaponPistol);
w = w.Upgrade(CharacterClass.Tank);
Assert.IsTrue(w is WeaponPistolShield);
w = w.Upgrade(CharacterClass.Tank);
Assert.IsTrue(w is WeaponPistolShield);
}
public void Weapon_UpgradeWeapon_Gunner()
{
Weapon w = new WeaponKnife();
w = w.Upgrade(CharacterClass.Gunner);
Assert.IsTrue(w is WeaponAR);
w = w.Upgrade(CharacterClass.Gunner);
Assert.IsTrue(w is WeaponAdvancedAR);
w = w.Upgrade(CharacterClass.Gunner);
Assert.IsTrue(w is WeaponAdvancedAR);
}