phosit/0ad
1
0
Fork 0
forked from 0ad/0ad
Code Pull Requests Activity

Add a unit test to EntityLimits component. Refs 1f10d2a711.

Browse Source 
Differential Revision: https://code.wildfiregames.com/D987
This was SVN commit r20373.
This commit is contained in:
fatherbushido 2017-10-29 08:06:45 +00:00
parent 1f10d2a711
commit c4cf9efad0
1 changed files with 46 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
binaries/data/mods/public/simulation/components/tests/test_EntityLimits.js
View File

@ -146,6 +146,52 @@ cmpEntityLimits.OnGlobalOwnershipChanged({ "entity": 92, "from": 1, "to": -1 });
TS_ASSERT_UNEVAL_EQUALS(cmpEntityLimits.GetLimits(), { "Tower": 5, "Wonder": 1, "Hero": 2, "Champion": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpEntityLimits.GetCounts(), { "Tower": 0, "Wonder": 0, "Hero": 0, "Champion": 0 });
// Test AllowedToReplace
AddMock(SYSTEM_ENTITY, IID_TemplateManager, {
"GetTemplate": name => {
switch (name)
{
case "templateA":
return { "TrainingRestrictions": { "Category": "Champion" } };
case "templateB":
return { "TrainingRestrictions": { "Category": "Hero" } };
case "templateC":
return { "BuildRestrictions": { "Category": "Wonder" } };
case "templateD":
return { "BuildRestrictions": { "Category": "Tower" } };
default:
return null;
}
},
"GetCurrentTemplateName": id => {
switch (id)
{
case 100:
return "templateA";
case 101:
return "templateB";
case 102:
return "templateC";
case 103:
return "templateD";
default:
return null;
}
}
});
cmpEntityLimits.ChangeCount( "Champion", 1)
TS_ASSERT(cmpEntityLimits.AllowedToReplace(100, "templateA"))
TS_ASSERT(!cmpEntityLimits.AllowedToReplace(101, "templateA"))
cmpEntityLimits.ChangeCount( "Champion", -1)
cmpEntityLimits.ChangeCount( "Tower", 5)
TS_ASSERT(!cmpEntityLimits.AllowedToReplace(102, "templateD"))
TS_ASSERT(cmpEntityLimits.AllowedToReplace(103, "templateD"))
cmpEntityLimits.ChangeCount( "Tower", -5)
TS_ASSERT_UNEVAL_EQUALS(cmpEntityLimits.GetCounts(), { "Tower": 0, "Wonder": 0, "Hero": 0, "Champion": 0 });
// Test limit removers by tech
cmpEntityLimits.UpdateLimitsFromTech("TechB");
TS_ASSERT_UNEVAL_EQUALS(cmpEntityLimits.GetLimits(), { "Tower": 5, "Wonder": 1, "Hero": 2, "Champion": 1 });