Differential D3628 Diff 16312 binaries/data/mods/public/simulation/components/tests/test_Garrisoning.js

Changeset View

Standalone View

binaries/data/mods/public/simulation/components/tests/test_Garrisoning.js

	Engine.LoadHelperScript("ValueModification.js");			Engine.LoadHelperScript("ValueModification.js");
	Engine.LoadHelperScript("Player.js");			Engine.LoadHelperScript("Player.js");
	Engine.LoadComponentScript("interfaces/Auras.js");			Engine.LoadComponentScript("interfaces/Auras.js");
	Engine.LoadComponentScript("interfaces/Garrisonable.js");			Engine.LoadComponentScript("interfaces/Garrisonable.js");
	Engine.LoadComponentScript("interfaces/GarrisonHolder.js");			Engine.LoadComponentScript("interfaces/GarrisonHolder.js");
	Engine.LoadComponentScript("interfaces/Health.js");			Engine.LoadComponentScript("interfaces/Health.js");
	Engine.LoadComponentScript("interfaces/ModifiersManager.js");			Engine.LoadComponentScript("interfaces/ModifiersManager.js");
	Engine.LoadComponentScript("interfaces/ProductionQueue.js");			Engine.LoadComponentScript("interfaces/ProductionQueue.js");
	Engine.LoadComponentScript("interfaces/Timer.js");			Engine.LoadComponentScript("interfaces/Timer.js");
	Engine.LoadComponentScript("interfaces/TurretHolder.js");			Engine.LoadComponentScript("interfaces/TurretHolder.js");
	Engine.LoadComponentScript("interfaces/UnitAI.js");			Engine.LoadComponentScript("interfaces/UnitAI.js");
	Engine.LoadComponentScript("Garrisonable.js");			Engine.LoadComponentScript("Garrisonable.js");
	Engine.LoadComponentScript("GarrisonHolder.js");			Engine.LoadComponentScript("GarrisonHolder.js");
				Engine.LoadComponentScript("TurretHolder.js");

	const player = 1;			const player = 1;
	const enemyPlayer = 2;			const enemyPlayer = 2;
	const friendlyPlayer = 3;			const friendlyPlayer = 3;
	const garrison = 10;			const garrison = 10;
	const holder = 11;			const holder = 11;

				let createGarrisonCmp = entity => {
				AddMock(entity, IID_Identity, {
				"GetClassesList": () => ["Ranged"],
				"GetSelectionGroupName": () => "mace_infantry_archer_a"
				});

				AddMock(entity, IID_Ownership, {
				"GetOwner": () => player
				});

				AddMock(entity, IID_Position, {
				"GetHeightOffset": () => 0,
				"GetPosition": () => new Vector3D(4, 3, 25),
				"GetRotation": () => new Vector3D(4, 0, 6),
				"JumpTo": (posX, posZ) => {},
				"MoveOutOfWorld": () => {},
				"SetHeightOffset": height => {},
				"SetTurretParent": ent => {},
				"SetYRotation": angle => {}
				});

				return ConstructComponent(entity, "Garrisonable", {
				"Size": "1"
				});
				};

	AddMock(holder, IID_Footprint, {			AddMock(holder, IID_Footprint, {
	"PickSpawnPointBothPass": entity => new Vector3D(4, 3, 30),			"PickSpawnPointBothPass": entity => new Vector3D(4, 3, 30),
	"PickSpawnPoint": entity => new Vector3D(4, 3, 30)			"PickSpawnPoint": entity => new Vector3D(4, 3, 30)
	});			});

	AddMock(holder, IID_Ownership, {			AddMock(holder, IID_Ownership, {
	"GetOwner": () => player			"GetOwner": () => player
	});			});
	Show All 28 Lines
	});			});

	AddMock(garrison, IID_Position, {			AddMock(garrison, IID_Position, {
	"GetHeightOffset": () => 0,			"GetHeightOffset": () => 0,
	"GetPosition": () => new Vector3D(4, 3, 25),			"GetPosition": () => new Vector3D(4, 3, 25),
	"GetRotation": () => new Vector3D(4, 0, 6),			"GetRotation": () => new Vector3D(4, 0, 6),
	"JumpTo": (posX, posZ) => {},			"JumpTo": (posX, posZ) => {},
	"MoveOutOfWorld": () => {},			"MoveOutOfWorld": () => {},
	"SetHeightOffset": height => {}			"SetHeightOffset": height => {},
				"SetTurretParent": entity => {},
				"SetYRotation": angle => {}
	});			});

	let cmpGarrisonable = ConstructComponent(garrison, "Garrisonable", {			let cmpGarrisonable = ConstructComponent(garrison, "Garrisonable", {
	"Size": "1"			"Size": "1"
	});			});

	let cmpGarrisonHolder = ConstructComponent(holder, "GarrisonHolder", {			let cmpGarrisonHolder = ConstructComponent(holder, "GarrisonHolder", {
	"Max": "10",			"Max": "10",
	Show All 11 Lines
	cmpGarrisonable.OnEntityRenamed({			cmpGarrisonable.OnEntityRenamed({
	"entity": garrison,			"entity": garrison,
	"newentity": -1			"newentity": -1
	});			});
	TS_ASSERT_EQUALS(cmpGarrisonHolder.GetGarrisonedEntitiesCount(), 0);			TS_ASSERT_EQUALS(cmpGarrisonHolder.GetGarrisonedEntitiesCount(), 0);

	TS_ASSERT(cmpGarrisonable.Garrison(holder));			TS_ASSERT(cmpGarrisonable.Garrison(holder));
	TS_ASSERT_UNEVAL_EQUALS(cmpGarrisonHolder.GetEntities(), [garrison]);			TS_ASSERT_UNEVAL_EQUALS(cmpGarrisonHolder.GetEntities(), [garrison]);

				// Can't garrison twice.
				TS_ASSERT(!cmpGarrisonable.Garrison(holder));
				TS_ASSERT_UNEVAL_EQUALS(cmpGarrisonHolder.GetEntities(), [garrison]);

				TS_ASSERT(cmpGarrisonHolder.Unload(garrison));
				TS_ASSERT_EQUALS(cmpGarrisonHolder.GetGarrisonedEntitiesCount(), 0);

				// Test initGarrison.
				let entities = [21, 22, 23, 24];
				for (let entity of entities)
				createGarrisonCmp(entity);
				cmpGarrisonHolder.SetInitGarrison(entities);
				cmpGarrisonHolder.OnGlobalInitGame();
				TS_ASSERT_UNEVAL_EQUALS(cmpGarrisonHolder.GetEntities(), entities);

				// They turned against us!
				AddMock(entities[0], IID_Ownership, {
				"GetOwner": () => enemyPlayer
				});
				cmpGarrisonHolder.OnDiplomacyChanged();
				TS_ASSERT_UNEVAL_EQUALS(cmpGarrisonHolder.GetGarrisonedEntitiesCount(), entities.length - 1);

				TS_ASSERT(cmpGarrisonHolder.UnloadAll());
				TS_ASSERT_UNEVAL_EQUALS(cmpGarrisonHolder.GetEntities(), []);

				// Turrets!
				AddMock(holder, IID_Position, {
				"GetPosition": () => new Vector3D(4, 3, 25),
				"GetRotation": () => new Vector3D(4, 0, 6)
				});

				let cmpTurretHolder = ConstructComponent(holder, "TurretHolder", {
				"TurretPoints": {
				"archer1": {
				"X": "12.0",
				"Y": "5.",
				"Z": "6.0"
				},
				"archer2": {
				"X": "15.0",
				"Y": "5.0",
				"Z": "6.0"
				}
				}
				});

				TS_ASSERT(cmpGarrisonable.Garrison(holder));
				TS_ASSERT_UNEVAL_EQUALS(cmpGarrisonHolder.GetEntities(), [garrison]);
				TS_ASSERT(cmpTurretHolder.OccupiesTurret(garrison));
				TS_ASSERT(cmpGarrisonable.UnGarrison());
				TS_ASSERT_UNEVAL_EQUALS(cmpGarrisonHolder.GetEntities(), []);
				TS_ASSERT_UNEVAL_EQUALS(cmpTurretHolder.GetEntities(), []);

				// Test renaming on a turret.
				// Ensure we test renaming from the second spot, not the first.
				const newGarrison = 31;
				let cmpGarrisonableNew = createGarrisonCmp(newGarrison);
				TS_ASSERT(cmpGarrisonableNew.Garrison(holder));
				TS_ASSERT(cmpGarrisonable.Garrison(holder));
				TS_ASSERT(cmpGarrisonableNew.UnGarrison());
				let previousTurret = cmpTurretHolder.GetOccupiedTurretName(garrison);
				cmpGarrisonable.OnEntityRenamed({
				"entity": garrison,
				"newentity": newGarrison
				});
				let newTurret = cmpTurretHolder.GetOccupiedTurretName(newGarrison);
				TS_ASSERT_UNEVAL_EQUALS(newTurret, previousTurret);
				TS_ASSERT(cmpGarrisonableNew.UnGarrison());

				// Test initTurrets.
				cmpTurretHolder.SetInitEntity("archer1", garrison);
				cmpTurretHolder.SetInitEntity("archer2", newGarrison);
				cmpTurretHolder.OnGlobalInitGame();
				TS_ASSERT_UNEVAL_EQUALS(cmpTurretHolder.GetEntities(), [garrison, newGarrison]);