Files
tbd-station-14/Content.IntegrationTests/Tests/ResearchTest.cs
2022-12-19 21:37:40 -06:00

60 lines
1.8 KiB
C#

using System.Collections.Generic;
using System.Threading.Tasks;
using Content.Shared.Lathe;
using Content.Shared.Research.Prototypes;
using NUnit.Framework;
using Robust.Shared.Prototypes;
namespace Content.IntegrationTests.Tests;
[TestFixture]
public sealed class ResearchTest
{
[Test]
public async Task AllTechPrintableTest()
{
await using var pairTracker = await PoolManager.GetServerClient(new PoolSettings {NoClient = true});
var server = pairTracker.Pair.Server;
var protoManager = server.ResolveDependency<IPrototypeManager>();
await server.WaitAssertion(() =>
{
var allEnts = protoManager.EnumeratePrototypes<EntityPrototype>();
var allLathes = new HashSet<LatheComponent>();
foreach (var proto in allEnts)
{
if (proto.Abstract)
continue;
if (!proto.TryGetComponent<LatheComponent>(out var lathe))
continue;
allLathes.Add(lathe);
}
var latheTechs = new HashSet<string>();
foreach (var lathe in allLathes)
{
if (lathe.DynamicRecipes == null)
continue;
foreach (var recipe in lathe.DynamicRecipes)
{
if (!latheTechs.Contains(recipe))
latheTechs.Add(recipe);
}
}
foreach (var tech in protoManager.EnumeratePrototypes<TechnologyPrototype>())
{
foreach (var recipe in tech.UnlockedRecipes)
{
Assert.That(latheTechs, Does.Contain(recipe), $"Recipe \"{recipe}\" cannot be unlocked on any lathes.");
}
}
});
await pairTracker.CleanReturnAsync();
}
}