Fix random fails on NoCargoOrderArbitrage Test (#13273)

This commit is contained in:
Nemanja
2023-01-01 16:58:18 -05:00
committed by GitHub
parent ac6bb3e21b
commit 204e727993

View File

@@ -20,6 +20,8 @@ public sealed class CargoTest
await using var pairTracker = await PoolManager.GetServerClient(new PoolSettings() {NoClient = true}); await using var pairTracker = await PoolManager.GetServerClient(new PoolSettings() {NoClient = true});
var server = pairTracker.Pair.Server; var server = pairTracker.Pair.Server;
var testMap = await PoolManager.CreateTestMap(pairTracker);
var entManager = server.ResolveDependency<IEntityManager>(); var entManager = server.ResolveDependency<IEntityManager>();
var mapManager = server.ResolveDependency<IMapManager>(); var mapManager = server.ResolveDependency<IMapManager>();
var protoManager = server.ResolveDependency<IPrototypeManager>(); var protoManager = server.ResolveDependency<IPrototypeManager>();
@@ -27,7 +29,7 @@ public sealed class CargoTest
await server.WaitAssertion(() => await server.WaitAssertion(() =>
{ {
var mapId = mapManager.CreateMap(); var mapId = testMap.MapId;
foreach (var proto in protoManager.EnumeratePrototypes<CargoProductPrototype>()) foreach (var proto in protoManager.EnumeratePrototypes<CargoProductPrototype>())
{ {
@@ -50,13 +52,15 @@ public sealed class CargoTest
await using var pairTracker = await PoolManager.GetServerClient(new PoolSettings{NoClient = true}); await using var pairTracker = await PoolManager.GetServerClient(new PoolSettings{NoClient = true});
var server = pairTracker.Pair.Server; var server = pairTracker.Pair.Server;
var testMap = await PoolManager.CreateTestMap(pairTracker);
var entManager = server.ResolveDependency<IEntityManager>(); var entManager = server.ResolveDependency<IEntityManager>();
var mapManager = server.ResolveDependency<IMapManager>(); var mapManager = server.ResolveDependency<IMapManager>();
var protoManager = server.ResolveDependency<IPrototypeManager>(); var protoManager = server.ResolveDependency<IPrototypeManager>();
await server.WaitAssertion(() => await server.WaitAssertion(() =>
{ {
var mapId = mapManager.CreateMap(); var mapId = testMap.MapId;
var grid = mapManager.CreateGrid(mapId); var grid = mapManager.CreateGrid(mapId);
var coord = new EntityCoordinates(grid.Owner, 0, 0); var coord = new EntityCoordinates(grid.Owner, 0, 0);
@@ -73,8 +77,10 @@ public sealed class CargoTest
&& stackpricecomp.Price > 0) && stackpricecomp.Price > 0)
{ {
if (entManager.TryGetComponent<StaticPriceComponent>(ent, out var staticpricecomp)) if (entManager.TryGetComponent<StaticPriceComponent>(ent, out var staticpricecomp))
{
Assert.That(staticpricecomp.Price, Is.EqualTo(0), Assert.That(staticpricecomp.Price, Is.EqualTo(0),
$"The prototype {proto} have a StackPriceComponent and StaticPriceComponent whose values are not compatible with each other."); $"The prototype {proto} have a StackPriceComponent and StaticPriceComponent whose values are not compatible with each other.");
}
} }
entManager.DeleteEntity(ent); entManager.DeleteEntity(ent);
} }