machine deconstruct test (#11779)

This commit is contained in:
Nemanja
2022-10-09 14:38:35 -04:00
committed by GitHub
parent f7c8c5feac
commit 75fa4fedf6
5 changed files with 75 additions and 4 deletions

View 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();
}
}