136 lines
5.2 KiB
C#
136 lines
5.2 KiB
C#
#nullable enable
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Content.Server.GameObjects.Components;
|
|
using Content.Server.GameObjects.Components.Disposal;
|
|
using Content.Server.GameObjects.Components.Power.ApcNetComponents;
|
|
using NUnit.Framework;
|
|
using Robust.Shared.Interfaces.GameObjects;
|
|
using Robust.Shared.Interfaces.Map;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Map;
|
|
|
|
namespace Content.IntegrationTests.Tests.Disposal
|
|
{
|
|
[TestFixture]
|
|
[TestOf(typeof(DisposalHolderComponent))]
|
|
[TestOf(typeof(DisposalEntryComponent))]
|
|
[TestOf(typeof(DisposalUnitComponent))]
|
|
public class DisposalUnitTest : ContentIntegrationTest
|
|
{
|
|
private void UnitInsert(DisposalUnitComponent unit, bool result, params IEntity[] entities)
|
|
{
|
|
foreach (var entity in entities)
|
|
{
|
|
Assert.That(unit.CanInsert(entity), Is.EqualTo(result));
|
|
Assert.That(unit.TryInsert(entity), Is.EqualTo(result));
|
|
|
|
if (result)
|
|
{
|
|
// Not in a tube yet
|
|
Assert.That(entity.Transform.Parent == unit.Owner.Transform);
|
|
}
|
|
}
|
|
}
|
|
|
|
private void UnitContains(DisposalUnitComponent unit, bool result, params IEntity[] entities)
|
|
{
|
|
foreach (var entity in entities)
|
|
{
|
|
Assert.That(unit.ContainedEntities.Contains(entity), Is.EqualTo(result));
|
|
}
|
|
}
|
|
|
|
private void UnitInsertContains(DisposalUnitComponent unit, bool result, params IEntity[] entities)
|
|
{
|
|
UnitInsert(unit, result, entities);
|
|
UnitContains(unit, result, entities);
|
|
}
|
|
|
|
private void Flush(DisposalUnitComponent unit, bool result, DisposalEntryComponent? entry = null, params IEntity[] entities)
|
|
{
|
|
Assert.That(unit.ContainedEntities, Is.SupersetOf(entities));
|
|
Assert.That(entities.Length, Is.EqualTo(unit.ContainedEntities.Count));
|
|
|
|
Assert.That(result, Is.EqualTo(unit.TryFlush()));
|
|
Assert.That(result || entities.Length == 0, Is.EqualTo(unit.ContainedEntities.Count == 0));
|
|
}
|
|
|
|
[Test]
|
|
public async Task Test()
|
|
{
|
|
var server = StartServerDummyTicker();
|
|
|
|
IEntity human;
|
|
IEntity wrench;
|
|
DisposalUnitComponent unit;
|
|
DisposalEntryComponent entry;
|
|
|
|
server.Assert(async () =>
|
|
{
|
|
var mapManager = IoCManager.Resolve<IMapManager>();
|
|
|
|
mapManager.CreateNewMapEntity(MapId.Nullspace);
|
|
|
|
var entityManager = IoCManager.Resolve<IEntityManager>();
|
|
|
|
// Spawn the entities
|
|
human = entityManager.SpawnEntity("HumanMob_Content", MapCoordinates.Nullspace);
|
|
wrench = entityManager.SpawnEntity("Wrench", MapCoordinates.Nullspace);
|
|
var disposalUnit = entityManager.SpawnEntity("DisposalUnit", MapCoordinates.Nullspace);
|
|
var disposalTrunk = entityManager.SpawnEntity("DisposalTrunk", disposalUnit.Transform.MapPosition);
|
|
|
|
// Test for components existing
|
|
Assert.True(disposalUnit.TryGetComponent(out unit!));
|
|
Assert.True(disposalTrunk.TryGetComponent(out entry!));
|
|
|
|
// Can't insert, unanchored and unpowered
|
|
var disposalUnitAnchorable = disposalUnit.GetComponent<AnchorableComponent>();
|
|
await disposalUnitAnchorable.TryUnAnchor(human, null, true);
|
|
Assert.False(unit.Anchored);
|
|
UnitInsertContains(unit, false, human, wrench, disposalUnit, disposalTrunk);
|
|
|
|
// Anchor the disposal unit
|
|
await disposalUnitAnchorable.TryAnchor(human, null, true);
|
|
Assert.True(disposalUnit.TryGetComponent(out AnchorableComponent? anchorableUnit));
|
|
Assert.True(await anchorableUnit!.TryAnchor(human, wrench));
|
|
Assert.True(unit.Anchored);
|
|
|
|
// No power
|
|
Assert.False(unit.Powered);
|
|
|
|
// Can't insert the trunk or the unit into itself
|
|
UnitInsertContains(unit, false, disposalUnit, disposalTrunk);
|
|
|
|
// Can insert mobs and items
|
|
UnitInsertContains(unit, true, human, wrench);
|
|
|
|
// Move the disposal trunk away
|
|
disposalTrunk.Transform.WorldPosition += (1, 0);
|
|
|
|
// Fail to flush with a mob and an item
|
|
Flush(unit, false, null, human, wrench);
|
|
|
|
// Move the disposal trunk back
|
|
disposalTrunk.Transform.WorldPosition -= (1, 0);
|
|
|
|
// Fail to flush with a mob and an item, no power
|
|
Flush(unit, false, entry, human, wrench);
|
|
|
|
// Remove power need
|
|
Assert.True(disposalUnit.TryGetComponent(out PowerReceiverComponent? power));
|
|
power!.NeedsPower = false;
|
|
Assert.True(unit.Powered);
|
|
|
|
// Flush with a mob and an item
|
|
Flush(unit, true, entry, human, wrench);
|
|
|
|
// Re-pressurizing
|
|
Flush(unit, false, entry);
|
|
});
|
|
|
|
await server.WaitIdleAsync();
|
|
}
|
|
}
|
|
}
|