36 lines
1.2 KiB
C#
36 lines
1.2 KiB
C#
using System.Threading.Tasks;
|
|
using Content.Server.Disease;
|
|
using Content.Shared.Disease;
|
|
using NUnit.Framework;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.IntegrationTests.Tests.Disease;
|
|
|
|
[TestFixture]
|
|
[TestOf(typeof(DiseaseSystem))]
|
|
public sealed class DeviceNetworkTest
|
|
{
|
|
[Test]
|
|
public async Task AddAllDiseases()
|
|
{
|
|
await using var pairTracker = await PoolManager.GetServerClient(new PoolSettings{NoClient = true});
|
|
var server = pairTracker.Pair.Server;
|
|
var testMap = await PoolManager.CreateTestMap(pairTracker);
|
|
await server.WaitPost(() =>
|
|
{
|
|
var protoManager = IoCManager.Resolve<IPrototypeManager>();
|
|
var entManager = IoCManager.Resolve<IEntityManager>();
|
|
var entSysManager = IoCManager.Resolve<IEntitySystemManager>();
|
|
var diseaseSystem = entSysManager.GetEntitySystem<DiseaseSystem>();
|
|
|
|
var sickEntity = entManager.SpawnEntity("MobHuman", testMap.GridCoords);
|
|
foreach (var diseaseProto in protoManager.EnumeratePrototypes<DiseasePrototype>())
|
|
{
|
|
diseaseSystem.TryAddDisease(sickEntity, diseaseProto);
|
|
}
|
|
});
|
|
}
|
|
}
|