machine deconstruct test (#11779)
This commit is contained in:
59
Content.IntegrationTests/Tests/MachineBoardTest.cs
Normal file
59
Content.IntegrationTests/Tests/MachineBoardTest.cs
Normal file
@@ -0,0 +1,59 @@
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Threading.Tasks;
|
||||
using Content.Server.Construction.Components;
|
||||
using NUnit.Framework;
|
||||
using Robust.Shared.Prototypes;
|
||||
|
||||
namespace Content.IntegrationTests.Tests;
|
||||
|
||||
public sealed class MachineBoardTest
|
||||
{
|
||||
/// <summary>
|
||||
/// A list of machine boards that can be ignored by this test.
|
||||
/// </summary>
|
||||
private readonly HashSet<string> _ignoredPrototypes = new()
|
||||
{
|
||||
//These have their own construction thing going on here
|
||||
"MachineParticleAcceleratorEndCapCircuitboard",
|
||||
"MachineParticleAcceleratorFuelChamberCircuitboard",
|
||||
"MachineParticleAcceleratorFuelChamberCircuitboard",
|
||||
"MachineParticleAcceleratorPowerBoxCircuitboard",
|
||||
"MachineParticleAcceleratorEmitterLeftCircuitboard",
|
||||
"MachineParticleAcceleratorEmitterCenterCircuitboard",
|
||||
"MachineParticleAcceleratorEmitterRightCircuitboard"
|
||||
};
|
||||
|
||||
/// <summary>
|
||||
/// Ensures that every single machine board's corresponding entity
|
||||
/// is a machine and can be properly deconstructed.
|
||||
/// </summary>
|
||||
[Test]
|
||||
public async Task TestMachineBoardHasValidMachine()
|
||||
{
|
||||
await using var pairTracker = await PoolManager.GetServerClient(new PoolSettings{NoClient = true});
|
||||
var server = pairTracker.Pair.Server;
|
||||
|
||||
var protoMan = server.ResolveDependency<IPrototypeManager>();
|
||||
|
||||
await server.WaitAssertion(() =>
|
||||
{
|
||||
foreach (var p in protoMan.EnumeratePrototypes<EntityPrototype>().Where(p => !p.Abstract && !_ignoredPrototypes.Contains(p.ID)))
|
||||
{
|
||||
if (!p.TryGetComponent<MachineBoardComponent>(out var mbc))
|
||||
continue;
|
||||
var mId = mbc.Prototype;
|
||||
|
||||
Assert.That(mId, Is.Not.Null, $"Machine board {p.ID} does not have a corresponding machine.");
|
||||
Assert.That(protoMan.TryIndex<EntityPrototype>(mId, out var mProto),
|
||||
$"Machine board {p.ID}'s corresponding machine has an invalid prototype.");
|
||||
Assert.That(mProto.TryGetComponent<MachineComponent>(out var mComp),
|
||||
$"Machine board {p.ID}'s corresponding machine {mId} does not have MachineComponent");
|
||||
Assert.That(mComp.BoardPrototype, Is.EqualTo(p.ID),
|
||||
$"Machine {mId}'s BoardPrototype is not equal to it's corresponding machine board, {p.ID}");
|
||||
}
|
||||
});
|
||||
|
||||
await pairTracker.CleanReturnAsync();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user