Differential D270 Diff 12246 binaries/data/mods/public/simulation/components/tests/test_ProductionQueue.js

Changeset View

Standalone View

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

Resources = {
"BuildSchema": (a, b) => {}
};

Engine.LoadHelperScript("Player.js");		Engine.LoadHelperScript("Player.js");
		Engine.LoadHelperScript("Sound.js");
Engine.LoadComponentScript("interfaces/TechnologyManager.js");		Engine.LoadComponentScript("interfaces/TechnologyManager.js");
Engine.LoadComponentScript("interfaces/ProductionQueue.js");		Engine.LoadComponentScript("interfaces/ProductionQueue.js");
		Engine.LoadComponentScript("interfaces/BuildRestrictions.js");
		Engine.LoadComponentScript("interfaces/EntityLimits.js");
		Engine.LoadComponentScript("interfaces/Foundation.js");
		Engine.LoadComponentScript("interfaces/StatisticsTracker.js");
		Engine.LoadComponentScript("interfaces/Timer.js");
		Engine.LoadComponentScript("interfaces/TrainingRestrictions.js");
		Engine.LoadComponentScript("interfaces/Trigger.js");
		Engine.LoadComponentScript("EntityLimits.js");

		Engine.RegisterGlobal("Resources", {
		"BuildSchema": (a, b) => {}
		});
Engine.LoadComponentScript("ProductionQueue.js");		Engine.LoadComponentScript("ProductionQueue.js");
		Engine.LoadComponentScript("TrainingRestrictions.js");

		Engine.RegisterGlobal("ApplyValueModificationsToEntity", (_, value) => value);
		Engine.RegisterGlobal("ApplyValueModificationsToTemplate", (_, value) => value);

global.TechnologyTemplates = {		function testEntitiesList()
		FreagarachUnsubmitted Not Done Inline Actions One can seperate logical parts in the test using comments instead of creating a function and calling it once (agreeing with @bb here). Freagarach: One can seperate logical parts in the test using comments instead of creating a function and…
		wraitiiAuthorUnsubmitted Done Inline Actions Mh, I would disagree on the grounds that this creates some weird global state. I don't really care what "testEntitiesList" does in my own test below. wraitii: Mh, I would disagree on the grounds that this creates some weird global state. I don't really…
		wraitiiAuthorUnsubmitted Done Inline Actions er -> I say that but I keep the global state since things don't get magically cleaned up -__- Still... wraitii: er -> I say that but I keep the global state since things don't get magically cleaned up -__…
		FreagarachUnsubmitted Not Done Inline Actions Still... Yes? :P Freagarach: > Still... Yes? :P
		wraitiiAuthorUnsubmitted Done Inline Actions Point is I don't think this is much worse or much better :p I'm leaving as is wraitii: Point is I don't think this is much worse or much better :p I'm leaving as is
		{
		Engine.RegisterGlobal("TechnologyTemplates", {
"Has": name => name == "phase_town_athen" \|\| name == "phase_city_athen",		"Has": name => name == "phase_town_athen" \|\| name == "phase_city_athen",
"Get": () => ({})		"Get": () => ({})
};		});

const productionQueueId = 6;		const productionQueueId = 6;
const playerId = 1;		const playerId = 1;
const playerEntityID = 2;		const playerEntityID = 2;

AddMock(SYSTEM_ENTITY, IID_TemplateManager, {		AddMock(SYSTEM_ENTITY, IID_TemplateManager, {
"TemplateExists": () => true,		"TemplateExists": () => true,
"GetTemplate": name => ({})		"GetTemplate": name => ({})
});		});

let cmpProductionQueue = ConstructComponent(productionQueueId, "ProductionQueue", {		let cmpProductionQueue = ConstructComponent(productionQueueId, "ProductionQueue", {
"Entities": { "_string": "units/{civ}_cavalry_javelinist_b " +		"Entities": { "_string": "units/{civ}_cavalry_javelinist_b " +
"units/{civ}_infantry_swordsman_b " +		"units/{civ}_infantry_swordsman_b " +
"units/{native}_support_female_citizen" },		"units/{native}_support_female_citizen" },
"Technologies": { "_string": "gather_fishing_net " +		"Technologies": { "_string": "gather_fishing_net " +
"phase_town_{civ} " +		"phase_town_{civ} " +
"phase_city_{civ}" }		"phase_city_{civ}" }
});		});
		cmpProductionQueue.GetUpgradedTemplate = (template) => template;
cmpProductionQueue.CalculateEntitiesList();
TS_ASSERT_UNEVAL_EQUALS(cmpProductionQueue.GetEntitiesList(), []);

AddMock(SYSTEM_ENTITY, IID_PlayerManager, {		AddMock(SYSTEM_ENTITY, IID_PlayerManager, {
"GetPlayerByID": id => playerEntityID		"GetPlayerByID": id => playerEntityID
});		});

AddMock(playerEntityID, IID_Player, {		AddMock(playerEntityID, IID_Player, {
"GetCiv": () => "iber",		"GetCiv": () => "iber",
"GetDisabledTechnologies": () => ({}),		"GetDisabledTechnologies": () => ({}),
"GetDisabledTemplates": () => ({}),		"GetDisabledTemplates": () => ({}),
"GetPlayerID": () => playerId		"GetPlayerID": () => playerId
});		});

AddMock(playerEntityID, IID_TechnologyManager, {		AddMock(playerEntityID, IID_TechnologyManager, {
"CheckTechnologyRequirements": () => true,		"CheckTechnologyRequirements": () => true,
"IsInProgress": () => false,		"IsInProgress": () => false,
"IsTechnologyResearched": () => false		"IsTechnologyResearched": () => false
});		});

AddMock(productionQueueId, IID_Ownership, {		AddMock(productionQueueId, IID_Ownership, {
"GetOwner": () => playerId		"GetOwner": () => playerId
});		});

AddMock(productionQueueId, IID_Identity, {		AddMock(productionQueueId, IID_Identity, {
"GetCiv": () => "iber"		"GetCiv": () => "iber"
});		});

cmpProductionQueue.CalculateEntitiesList();		cmpProductionQueue.CalculateEntitiesMap();
TS_ASSERT_UNEVAL_EQUALS(		TS_ASSERT_UNEVAL_EQUALS(
cmpProductionQueue.GetEntitiesList(),		cmpProductionQueue.GetEntitiesList(),
["units/iber_cavalry_javelinist_b", "units/iber_infantry_swordsman_b", "units/iber_support_female_citizen"]		["units/iber_cavalry_javelinist_b", "units/iber_infantry_swordsman_b", "units/iber_support_female_citizen"]
);		);
TS_ASSERT_UNEVAL_EQUALS(		TS_ASSERT_UNEVAL_EQUALS(
cmpProductionQueue.GetTechnologiesList(),		cmpProductionQueue.GetTechnologiesList(),
["gather_fishing_net", "phase_town_generic", "phase_city_generic"]		["gather_fishing_net", "phase_town_generic", "phase_city_generic"]
);		);

AddMock(SYSTEM_ENTITY, IID_TemplateManager, {		AddMock(SYSTEM_ENTITY, IID_TemplateManager, {
"TemplateExists": name => name == "units/iber_support_female_citizen",		"TemplateExists": name => name == "units/iber_support_female_citizen",
"GetTemplate": name => ({})		"GetTemplate": name => ({})
});		});

cmpProductionQueue.CalculateEntitiesList();		cmpProductionQueue.CalculateEntitiesMap();
TS_ASSERT_UNEVAL_EQUALS(cmpProductionQueue.GetEntitiesList(), ["units/iber_support_female_citizen"]);		TS_ASSERT_UNEVAL_EQUALS(cmpProductionQueue.GetEntitiesList(), ["units/iber_support_female_citizen"]);

AddMock(SYSTEM_ENTITY, IID_TemplateManager, {		AddMock(SYSTEM_ENTITY, IID_TemplateManager, {
"TemplateExists": () => true,		"TemplateExists": () => true,
"GetTemplate": name => ({})		"GetTemplate": name => ({})
});		});

AddMock(playerEntityID, IID_Player, {		AddMock(playerEntityID, IID_Player, {
"GetCiv": () => "iber",		"GetCiv": () => "iber",
"GetDisabledTechnologies": () => ({}),		"GetDisabledTechnologies": () => ({}),
"GetDisabledTemplates": () => ({ "units/athen_infantry_swordsman_b": true }),		"GetDisabledTemplates": () => ({ "units/athen_infantry_swordsman_b": true }),
"GetPlayerID": () => playerId		"GetPlayerID": () => playerId
});		});

cmpProductionQueue.CalculateEntitiesList();		cmpProductionQueue.CalculateEntitiesMap();
TS_ASSERT_UNEVAL_EQUALS(		TS_ASSERT_UNEVAL_EQUALS(
cmpProductionQueue.GetEntitiesList(),		cmpProductionQueue.GetEntitiesList(),
["units/iber_cavalry_javelinist_b", "units/iber_infantry_swordsman_b", "units/iber_support_female_citizen"]		["units/iber_cavalry_javelinist_b", "units/iber_infantry_swordsman_b", "units/iber_support_female_citizen"]
);		);

AddMock(playerEntityID, IID_Player, {		AddMock(playerEntityID, IID_Player, {
"GetCiv": () => "iber",		"GetCiv": () => "iber",
"GetDisabledTechnologies": () => ({}),		"GetDisabledTechnologies": () => ({}),
"GetDisabledTemplates": () => ({ "units/iber_infantry_swordsman_b": true }),		"GetDisabledTemplates": () => ({ "units/iber_infantry_swordsman_b": true }),
"GetPlayerID": () => playerId		"GetPlayerID": () => playerId
});		});

cmpProductionQueue.CalculateEntitiesList();		cmpProductionQueue.CalculateEntitiesMap();
TS_ASSERT_UNEVAL_EQUALS(		TS_ASSERT_UNEVAL_EQUALS(
cmpProductionQueue.GetEntitiesList(),		cmpProductionQueue.GetEntitiesList(),
["units/iber_cavalry_javelinist_b", "units/iber_support_female_citizen"]		["units/iber_cavalry_javelinist_b", "units/iber_support_female_citizen"]
);		);

AddMock(playerEntityID, IID_Player, {		AddMock(playerEntityID, IID_Player, {
"GetCiv": () => "athen",		"GetCiv": () => "athen",
"GetDisabledTechnologies": () => ({ "gather_fishing_net": true }),		"GetDisabledTechnologies": () => ({ "gather_fishing_net": true }),
"GetDisabledTemplates": () => ({ "units/athen_infantry_swordsman_b": true }),		"GetDisabledTemplates": () => ({ "units/athen_infantry_swordsman_b": true }),
"GetPlayerID": () => playerId		"GetPlayerID": () => playerId
});		});

cmpProductionQueue.CalculateEntitiesList();		cmpProductionQueue.CalculateEntitiesMap();
TS_ASSERT_UNEVAL_EQUALS(		TS_ASSERT_UNEVAL_EQUALS(
cmpProductionQueue.GetEntitiesList(),		cmpProductionQueue.GetEntitiesList(),
["units/athen_cavalry_javelinist_b", "units/iber_support_female_citizen"]		["units/athen_cavalry_javelinist_b", "units/iber_support_female_citizen"]
);		);
TS_ASSERT_UNEVAL_EQUALS(cmpProductionQueue.GetTechnologiesList(), ["phase_town_athen",		TS_ASSERT_UNEVAL_EQUALS(cmpProductionQueue.GetTechnologiesList(), ["phase_town_athen",
"phase_city_athen"]		"phase_city_athen"]
);		);

AddMock(playerEntityID, IID_TechnologyManager, {		AddMock(playerEntityID, IID_TechnologyManager, {
"CheckTechnologyRequirements": () => true,		"CheckTechnologyRequirements": () => true,
"IsInProgress": () => false,		"IsInProgress": () => false,
"IsTechnologyResearched": tech => tech == "phase_town_athen"		"IsTechnologyResearched": tech => tech == "phase_town_athen"
});		});
TS_ASSERT_UNEVAL_EQUALS(cmpProductionQueue.GetTechnologiesList(), [undefined, "phase_city_athen"]);		TS_ASSERT_UNEVAL_EQUALS(cmpProductionQueue.GetTechnologiesList(), [undefined, "phase_city_athen"]);

AddMock(playerEntityID, IID_Player, {		AddMock(playerEntityID, IID_Player, {
"GetCiv": () => "iber",		"GetCiv": () => "iber",
"GetDisabledTechnologies": () => ({}),		"GetDisabledTechnologies": () => ({}),
"GetPlayerID": () => playerId		"GetPlayerID": () => playerId
});		});
TS_ASSERT_UNEVAL_EQUALS(		TS_ASSERT_UNEVAL_EQUALS(
cmpProductionQueue.GetTechnologiesList(),		cmpProductionQueue.GetTechnologiesList(),
["gather_fishing_net", "phase_town_generic", "phase_city_generic"]		["gather_fishing_net", "phase_town_generic", "phase_city_generic"]
);		);
		}
Engine.LoadComponentScript("interfaces/BuildRestrictions.js");
Engine.LoadComponentScript("interfaces/EntityLimits.js");
Engine.LoadComponentScript("interfaces/Foundation.js");
Engine.LoadComponentScript("interfaces/StatisticsTracker.js");
Engine.LoadComponentScript("interfaces/Timer.js");
Engine.LoadComponentScript("interfaces/TrainingRestrictions.js");
Engine.LoadComponentScript("interfaces/Trigger.js");
Engine.LoadComponentScript("EntityLimits.js");
Engine.LoadComponentScript("TrainingRestrictions.js");
Engine.LoadHelperScript("Sound.js");

Engine.RegisterGlobal("ApplyValueModificationsToEntity", (_, value) => value);
Engine.RegisterGlobal("ApplyValueModificationsToTemplate", (_, value) => value);

function regression_test_d1879()		function regression_test_d1879()
{		{
// Setup		// Setup
let playerEnt = 2;		let playerEnt = 2;
let playerID = 1;		let playerID = 1;
let testEntity = 3;		let testEntity = 3;
let spawedEntityIDs = [4, 5, 6, 7, 8];		let spawedEntityIDs = [4, 5, 6, 7, 8];
▲ Show 20 Lines • Show All 111 Lines • ▼ Show 20 Lines	function regression_test_d1879()

cmpProdQueue.AddBatch("some_template", "unit", 3);		cmpProdQueue.AddBatch("some_template", "unit", 3);
Engine.QueryInterface(testEntity, IID_ProductionQueue).ProgressTimeout();		Engine.QueryInterface(testEntity, IID_ProductionQueue).ProgressTimeout();

TS_ASSERT_EQUALS(cmpProdQueue.GetQueue().length, 1);		TS_ASSERT_EQUALS(cmpProdQueue.GetQueue().length, 1);
TS_ASSERT_EQUALS(cmpEntLimits.GetCounts().some_limit, 6);		TS_ASSERT_EQUALS(cmpEntLimits.GetCounts().some_limit, 6);
}		}

		function test_token_changes()
		{
		const ent = 10;
		let cmpProductionQueue = ConstructComponent(10, "ProductionQueue", {
		"Entities": { "_string": "units/{civ}_a " +
		"units/{civ}_b" },
		"Technologies": { "_string": "a " +
		"b_{civ} " +
		"c_{civ}" },
		"BatchTimeModifier": 1
		});
		cmpProductionQueue.GetUpgradedTemplate = (template) => template;

		// Merges interface of multiple components because it's enough here.
		Engine.RegisterGlobal("QueryOwnerInterface", () => ({
		// player
		"GetCiv": () => "test",
		"GetDisabledTemplates": () => [],
		"GetDisabledTechnologies": () => [],
		"TrySubtractResources": () => true,
		"AddResources": () => {},
		"GetPlayerID": () => 1,
		// entitylimits
		"ChangeCount": () => {},
		// techmanager
		"CheckTechnologyRequirements": () => true,
		"IsTechnologyResearched": () => false,
		"IsInProgress": () => false
		}));
		Engine.RegisterGlobal("QueryPlayerIDInterface", QueryOwnerInterface);

		AddMock(SYSTEM_ENTITY, IID_GuiInterface, {
		"SetSelectionDirty": () => {}
		});

		// Test Setup
		cmpProductionQueue.CalculateEntitiesMap();
		TS_ASSERT_UNEVAL_EQUALS(
		cmpProductionQueue.GetEntitiesList(), ["units/test_a", "units/test_b"]
		);
		TS_ASSERT_UNEVAL_EQUALS(
		cmpProductionQueue.GetTechnologiesList(), ["a", "b_generic", "c_generic"]
		);
		// Add a unit of each type to our queue, validate.
		cmpProductionQueue.AddBatch("units/test_a", "unit", 1, {});
		cmpProductionQueue.AddBatch("units/test_b", "unit", 1, {});
		TS_ASSERT_EQUALS(cmpProductionQueue.GetQueue()[0].unitTemplate, "units/test_a");
		TS_ASSERT_EQUALS(cmpProductionQueue.GetQueue()[1].unitTemplate, "units/test_b");

		// Add a modifier that replaces unit A with unit C,
		// adds a unit D and removes unit B from the roster.
		Engine.RegisterGlobal("ApplyValueModificationsToEntity", (_, val) => {
		return HandleTokens(val, "units/{civ}_a>units/{civ}_c units/{civ}_d -units/{civ}_b");
		});

		cmpProductionQueue.OnValueModification({
		"component": "ProductionQueue",
		"valueNames": ["ProductionQueue/Entities/_string"],
		"entities": [ent]
		});

		TS_ASSERT_UNEVAL_EQUALS(
		cmpProductionQueue.GetEntitiesList(), ["units/test_c", "units/test_d"]
		);
		TS_ASSERT_EQUALS(cmpProductionQueue.GetQueue()[0].unitTemplate, "units/test_c");
		TS_ASSERT_EQUALS(cmpProductionQueue.GetQueue().length, 1);
		}

		testEntitiesList();
regression_test_d1879();		regression_test_d1879();
		test_token_changes();