Page MenuHomeWildfire Games
Differential D2662 Diff 13775 binaries/data/mods/public/simulation/components/tests/test_ResourceGatherer.js

Changeset View

Standalone View

View Options

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

  • This file was added.
Resources = {
"BuildSchema": () => {
let schema = "";
for (let res of ["food", "wood"])
{
for (let subtype in ["meat", "grain"])
schema += "<value>" + res + "." + subtype + "</value>";
schema += "<value> treasure." + res + "</value>";
}
return "<choice>" + schema + "</choice>";
}
};
Engine.LoadComponentScript("interfaces/ResourceGatherer.js");
Engine.LoadComponentScript("interfaces/ResourceSupply.js");
Engine.LoadComponentScript("interfaces/StatisticsTracker.js");
Engine.LoadComponentScript("interfaces/Timer.js");
Engine.LoadComponentScript("interfaces/Trigger.js");
Engine.LoadComponentScript("interfaces/UnitAI.js");
Engine.LoadComponentScript("ResourceGatherer.js");
Engine.LoadComponentScript("Timer.js");
Engine.LoadHelperScript("Player.js");
AddMock(SYSTEM_ENTITY, IID_ObstructionManager, {
"IsInTargetRange": () => true
});
let ApplyValueModificationsToEntity = (valueName, value, entity) => value;
Engine.RegisterGlobal("ApplyValueModificationsToEntity", ApplyValueModificationsToEntity);
let QueryOwnerInterface = () => {};
Engine.RegisterGlobal("QueryOwnerInterface", QueryOwnerInterface);
let cmpTimer;
let gathererID = 101;
let supplyID = 102;
let template = {
"MaxDistance": "10",
"BaseSpeed": "1",
"Rates": {
"food": "1",
"wood": "2"
},
"Capacities": {
"food": "10",
"wood": "20"
}
};
let cmpResourceGatherer = ConstructComponent(gathererID, "ResourceGatherer", template);
function reset()
{
cmpResourceGatherer = ConstructComponent(gathererID, "ResourceGatherer", template);
cmpResourceGatherer.RecalculateGatherRatesAndCapacities(); // Force updating values.
cmpTimer = ConstructComponent(SYSTEM_ENTITY, "Timer", {});
}
// General getters.
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(), []);
TS_ASSERT_EQUALS(cmpResourceGatherer.GetMainCarryingType(), undefined);
TS_ASSERT_EQUALS(cmpResourceGatherer.GetLastCarriedType(), undefined);
reset();
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetGatherRates(), {
"food": 1,
"wood": 2
});
TS_ASSERT_EQUALS(cmpResourceGatherer.GetGatherRate("food"), 1);
TS_ASSERT_EQUALS(cmpResourceGatherer.GetGatherRate("bogus"), 0);
TS_ASSERT_EQUALS(cmpResourceGatherer.GetCapacity("wood"), 20);
TS_ASSERT_EQUALS(cmpResourceGatherer.GetCapacity("bogus"), 0);
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetRange(), {
"max": 10,
"min": 0
});
TS_ASSERT(cmpResourceGatherer.CanCarryMore("food"));
TS_ASSERT(!cmpResourceGatherer.CanCarryMore("bogus"));
TS_ASSERT(!cmpResourceGatherer.IsCarrying("food"));
TS_ASSERT(!cmpResourceGatherer.IsCarrying("bogus"));
bbUnsubmitted
Done
Inline Actions

"GetType": () => ({ "generic": "bogus" })

bb: `"GetType": () => ({ "generic": "bogus" })`
// Test normal gathering.
reset();
// Resource supply is full (or something else).
AddMock(supplyID, IID_ResourceSupply, {
"AddGatherer": () => false
});
TS_ASSERT(!cmpResourceGatherer.StartGathering(supplyID));
// Supply is empty.
AddMock(supplyID, IID_ResourceSupply, {
"AddGatherer": () => true,
"GetCurrentAmount": () => 0
});
TS_ASSERT(!cmpResourceGatherer.StartGathering(supplyID));
// Supply is wrong type.
AddMock(supplyID, IID_ResourceSupply, {
"AddGatherer": () => true,
"GetCurrentAmount": () => 1,
"GetType": () => ({ "generic": "bogus" }),
"GetDiminishingReturns": () => 1
});
TS_ASSERT(!cmpResourceGatherer.StartGathering(supplyID));
// Supply is non-empty and right type.
AddMock(supplyID, IID_ResourceSupply, {
"AddGatherer": () => true,
"GetCurrentAmount": () => 2,
"GetType": () => ({ "generic": "food" }),
"GetDiminishingReturns": () => 1,
"TakeResources": (amount) => {
return {
"amount": amount,
"exhausted": false
};
},
"RemoveGatherer": () => {}
});
TS_ASSERT(cmpResourceGatherer.StartGathering(supplyID));
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 1, "max": 10 }]);
TS_ASSERT_EQUALS(cmpResourceGatherer.GetMainCarryingType(), "food");
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetLastCarriedType(), { "generic": "food" });
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 2, "max": 10 }]);
cmpResourceGatherer.StopGathering();
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 2, "max": 10 }]);
TS_ASSERT(cmpResourceGatherer.IsCarrying("food"));
TS_ASSERT(cmpResourceGatherer.CanCarryMore("food"));
// Test that when gathering a second type the first gathered type is ditched.
reset();
AddMock(supplyID, IID_ResourceSupply, {
"AddGatherer": () => true,
"GetCurrentAmount": () => 2,
"GetType": () => ({ "generic": "food" }),
"GetDiminishingReturns": () => 1,
"TakeResources": (amount) => {
return {
"amount": amount,
"exhausted": false
};
},
"RemoveGatherer": () => {}
});
TS_ASSERT(cmpResourceGatherer.StartGathering(supplyID));
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 1, "max": 10 }]);
cmpResourceGatherer.StopGathering();
AddMock(supplyID, IID_ResourceSupply, {
"AddGatherer": () => true,
"GetCurrentAmount": () => 3,
"GetType": () => ({ "generic": "wood" }),
"GetDiminishingReturns": () => 1,
"TakeResources": (amount) => {
return {
"amount": amount,
"exhausted": false
};
},
"RemoveGatherer": () => {}
});
TS_ASSERT(cmpResourceGatherer.StartGathering(supplyID));
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "wood", "amount": 2, "max": 20 }]);
TS_ASSERT(!cmpResourceGatherer.IsCarrying("food"));
TS_ASSERT(cmpResourceGatherer.CanCarryMore("food"));
TS_ASSERT(cmpResourceGatherer.IsCarrying("wood"));
TS_ASSERT(cmpResourceGatherer.CanCarryMore("wood"));
// Test that we stop gathering when the target is exhausted.
reset();
AddMock(supplyID, IID_ResourceSupply, {
"AddGatherer": () => true,
"GetCurrentAmount": () => 1,
"GetType": () => ({ "generic": "food" }),
"GetDiminishingReturns": () => 1,
"TakeResources": (amount) => {
return {
"amount": amount,
"exhausted": true
};
},
"RemoveGatherer": () => {}
});
TS_ASSERT(cmpResourceGatherer.StartGathering(supplyID));
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 1, "max": 10 }]);
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 1, "max": 10 }]);
// Test that we stop gathering when we are filled.
reset();
AddMock(supplyID, IID_ResourceSupply, {
"AddGatherer": () => true,
"GetCurrentAmount": () => 11,
"GetType": () => ({ "generic": "food" }),
"GetDiminishingReturns": () => 1,
"TakeResources": (amount) => {
return {
"amount": amount,
"exhausted": false
};
},
"RemoveGatherer": () => {}
});
TS_ASSERT(cmpResourceGatherer.StartGathering(supplyID));
cmpTimer.OnUpdate({ "turnLength": 10 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 10, "max": 10 }]);
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 10, "max": 10 }]);
// Test that starting to gather twice does not add resources at twice the speed.
reset();
AddMock(supplyID, IID_ResourceSupply, {
"AddGatherer": () => true,
"GetCurrentAmount": () => 3,
"GetType": () => ({ "generic": "food" }),
"GetDiminishingReturns": () => 1,
"TakeResources": (amount) => {
return {
"amount": amount,
"exhausted": false
};
},
"RemoveGatherer": () => {}
});
TS_ASSERT(cmpResourceGatherer.StartGathering(supplyID));
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 1, "max": 10 }]);
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 2, "max": 10 }]);
TS_ASSERT(cmpResourceGatherer.StartGathering(supplyID));
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 3, "max": 10 }]);
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(),
[{ "type": "food", "amount": 4, "max": 10 }]);
// Test treasure gathering.
reset();
AddMock(supplyID, IID_ResourceSupply, {
"AddGatherer": () => true,
"GetType": () => ({ "generic": "treasure" }),
"GetCurrentAmount": () => 10,
"TakeResources": (value) => {
return {
"amount": value,
"exhausted": true
};
},
"RemoveGatherer": () => {}
});
TS_ASSERT(cmpResourceGatherer.StartGathering(supplyID));
cmpTimer.OnUpdate({ "turnLength": 1 });
TS_ASSERT_UNEVAL_EQUALS(cmpResourceGatherer.GetCarryingStatus(), []);
Wildfire Games · Phabricator