tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make cargo arbitrage test ignore gambling crate (#22943)

Browse Source
This commit is contained in:
Leon Friedrich
2023-12-25 02:15:33 -05:00
committed by GitHub
parent e10a314ec2
commit 9eaa4c3a5d
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
Content.IntegrationTests/Tests/CargoTest.cs
View File

@@ -1,3 +1,4 @@
using System.Collections.Generic;
using System.Linq; using System.Linq;
using System.Numerics; using System.Numerics;
using Content.Server.Cargo.Components; using Content.Server.Cargo.Components;
@@ -6,7 +7,6 @@ using Content.Shared.Cargo.Prototypes;
using Content.Shared.Stacks; using Content.Shared.Stacks;
using Robust.Shared.GameObjects; using Robust.Shared.GameObjects;
using Robust.Shared.Map; using Robust.Shared.Map;
using Robust.Shared.Maths;
using Robust.Shared.Prototypes; using Robust.Shared.Prototypes;
namespace Content.IntegrationTests.Tests; namespace Content.IntegrationTests.Tests;
@@ -14,6 +14,12 @@ namespace Content.IntegrationTests.Tests;
[TestFixture] [TestFixture]
public sealed class CargoTest public sealed class CargoTest
{ {
public static HashSet<ProtoId<CargoProductPrototype>> Ignored = new ()
{
// This is ignored because it is explicitly intended to be able to sell for more than it costs.
new("FunCrateGambling")
};
[Test] [Test]
public async Task NoCargoOrderArbitrage() public async Task NoCargoOrderArbitrage()
{ {
@@ -23,27 +29,25 @@ public sealed class CargoTest
var testMap = await pair.CreateTestMap(); var testMap = await pair.CreateTestMap();
var entManager = server.ResolveDependency<IEntityManager>(); var entManager = server.ResolveDependency<IEntityManager>();
var mapManager = server.ResolveDependency<IMapManager>();
var protoManager = server.ResolveDependency<IPrototypeManager>(); var protoManager = server.ResolveDependency<IPrototypeManager>();
var pricing = server.ResolveDependency<IEntitySystemManager>().GetEntitySystem<PricingSystem>(); var pricing = server.ResolveDependency<IEntitySystemManager>().GetEntitySystem<PricingSystem>();
await server.WaitAssertion(() => await server.WaitAssertion(() =>
{ {
var mapId = testMap.MapId;
Assert.Multiple(() => Assert.Multiple(() =>
{ {
foreach (var proto in protoManager.EnumeratePrototypes<CargoProductPrototype>()) foreach (var proto in protoManager.EnumeratePrototypes<CargoProductPrototype>())
{ {
var ent = entManager.SpawnEntity(proto.Product, new MapCoordinates(Vector2.Zero, mapId)); if (Ignored.Contains(proto.ID))
continue;
var ent = entManager.SpawnEntity(proto.Product, testMap.MapCoords);
var price = pricing.GetPrice(ent); var price = pricing.GetPrice(ent);
Assert.That(price, Is.AtMost(proto.PointCost), $"Found arbitrage on {proto.ID} cargo product! Cost is {proto.PointCost} but sell is {price}!"); Assert.That(price, Is.AtMost(proto.PointCost), $"Found arbitrage on {proto.ID} cargo product! Cost is {proto.PointCost} but sell is {price}!");
entManager.DeleteEntity(ent); entManager.DeleteEntity(ent);
} }
}); });
mapManager.DeleteMap(mapId);
}); });
await pair.CleanReturnAsync(); await pair.CleanReturnAsync();