mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
bd60760f9d
Fixes https://github.com/Khopa/dcs_liberation/issues/473. Air defenses for bases, strike locations, and fixed IADS will now all downgrade to lower tier systems as needed. Strike locations will still be spawned as an equally weighted random generator from either the medium or long range groups, but will use a short range system if none are available to the faction. I've made the change in a way that leaves factions compatible, but will follow up to clean up our built-in factions.
101 lines
4.2 KiB
Python
101 lines
4.2 KiB
Python
import json
|
|
from pathlib import Path
|
|
import unittest
|
|
|
|
from dcs.helicopters import UH_1H, AH_64A
|
|
from dcs.planes import F_15C, F_15E, F_14B, FA_18C_hornet, F_16C_50, A_10A, AV8BNA, B_52H, B_1B, F_117A, MQ_9_Reaper, \
|
|
E_3A, KC130, KC_135, A_10C, A_10C_2
|
|
from dcs.ships import CVN_74_John_C__Stennis, LHA_1_Tarawa, Oliver_Hazzard_Perry_class, USS_Arleigh_Burke_IIa, \
|
|
Ticonderoga_class
|
|
from dcs.vehicles import Armor, Unarmed, Infantry, Artillery
|
|
|
|
from game.factions.faction import Faction
|
|
|
|
|
|
THIS_DIR = Path(__file__).parent
|
|
RESOURCES_DIR = THIS_DIR / "resources"
|
|
|
|
|
|
class TestFactionLoader(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
pass
|
|
|
|
def test_load_valid_faction(self):
|
|
with (RESOURCES_DIR / "valid_faction.json").open('r') as data:
|
|
faction = Faction.from_json(json.load(data))
|
|
|
|
self.assertEqual(faction.country, "USA")
|
|
self.assertEqual(faction.name, "USA 2005")
|
|
self.assertEqual(faction.authors, "Khopa")
|
|
self.assertEqual(faction.description, "This is a test description")
|
|
|
|
self.assertIn(F_15C, faction.aircrafts)
|
|
self.assertIn(F_15E, faction.aircrafts)
|
|
self.assertIn(F_14B, faction.aircrafts)
|
|
self.assertIn(FA_18C_hornet, faction.aircrafts)
|
|
self.assertIn(F_16C_50, faction.aircrafts)
|
|
self.assertIn(A_10A, faction.aircrafts)
|
|
self.assertIn(AV8BNA, faction.aircrafts)
|
|
self.assertIn(UH_1H, faction.aircrafts)
|
|
self.assertIn(AH_64A, faction.aircrafts)
|
|
self.assertIn(B_52H, faction.aircrafts)
|
|
self.assertIn(B_1B, faction.aircrafts)
|
|
self.assertIn(F_117A, faction.aircrafts)
|
|
self.assertIn(A_10C, faction.aircrafts)
|
|
self.assertIn(A_10C_2, faction.aircrafts)
|
|
|
|
self.assertEqual(len(faction.awacs), 1)
|
|
self.assertIn(E_3A, faction.awacs)
|
|
|
|
self.assertEqual(len(faction.tankers), 2)
|
|
self.assertIn(KC_135, faction.tankers)
|
|
self.assertIn(KC130, faction.tankers)
|
|
|
|
self.assertTrue(faction.has_jtac)
|
|
self.assertEqual(faction.jtac_unit, MQ_9_Reaper)
|
|
|
|
self.assertIn(Armor.MBT_M1A2_Abrams, faction.frontline_units)
|
|
self.assertIn(Armor.ATGM_M1134_Stryker, faction.frontline_units)
|
|
self.assertIn(Armor.APC_M1126_Stryker_ICV, faction.frontline_units)
|
|
self.assertIn(Armor.IFV_M2A2_Bradley, faction.frontline_units)
|
|
self.assertIn(Armor.IFV_LAV_25, faction.frontline_units)
|
|
self.assertIn(Armor.APC_M1043_HMMWV_Armament, faction.frontline_units)
|
|
self.assertIn(Armor.ATGM_M1045_HMMWV_TOW, faction.frontline_units)
|
|
|
|
self.assertIn(Artillery.MLRS_M270, faction.artillery_units)
|
|
self.assertIn(Artillery.SPH_M109_Paladin, faction.artillery_units)
|
|
|
|
self.assertIn(Unarmed.Transport_M818, faction.logistics_units)
|
|
|
|
self.assertIn(Infantry.Infantry_M4, faction.infantry_units)
|
|
self.assertIn(Infantry.Soldier_M249, faction.infantry_units)
|
|
|
|
self.assertIn("AvengerGenerator", faction.air_defenses)
|
|
|
|
self.assertIn("HawkGenerator", faction.air_defenses)
|
|
|
|
self.assertIn(CVN_74_John_C__Stennis, faction.aircraft_carrier)
|
|
self.assertIn(LHA_1_Tarawa, faction.helicopter_carrier)
|
|
self.assertIn(Oliver_Hazzard_Perry_class, faction.destroyers)
|
|
self.assertIn(USS_Arleigh_Burke_IIa, faction.destroyers)
|
|
self.assertIn(Ticonderoga_class, faction.cruisers)
|
|
|
|
self.assertIn("mod", faction.requirements.keys())
|
|
self.assertIn("Some mod is required", faction.requirements.values())
|
|
|
|
self.assertEqual(4, len(faction.carrier_names))
|
|
self.assertEqual(5, len(faction.helicopter_carrier_names))
|
|
|
|
self.assertIn("OliverHazardPerryGroupGenerator", faction.navy_generators)
|
|
self.assertIn("ArleighBurkeGroupGenerator", faction.navy_generators)
|
|
|
|
def test_load_valid_faction_with_invalid_country(self):
|
|
|
|
with (RESOURCES_DIR / "invalid_faction_country.json").open("r") as data:
|
|
try:
|
|
Faction.from_json(json.load(data))
|
|
self.fail("Should have thrown assertion error")
|
|
except AssertionError as e:
|
|
pass
|