79 lines
2.6 KiB
C#
79 lines
2.6 KiB
C#
using System.Numerics;
|
|
using Content.Server.Body.Systems;
|
|
using Content.Shared.Body.Components;
|
|
using Content.Shared.Body.Part;
|
|
using Content.Shared.Rotation;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.Map;
|
|
using Robust.Shared.Maths;
|
|
|
|
namespace Content.IntegrationTests.Tests.Body
|
|
{
|
|
[TestFixture]
|
|
[TestOf(typeof(BodyPartComponent))]
|
|
[TestOf(typeof(BodyComponent))]
|
|
public sealed class LegTest
|
|
{
|
|
[TestPrototypes]
|
|
private const string Prototypes = @"
|
|
- type: entity
|
|
name: HumanBodyAndAppearanceDummy
|
|
id: HumanBodyAndAppearanceDummy
|
|
components:
|
|
- type: Appearance
|
|
- type: Body
|
|
prototype: Human
|
|
- type: StandingState
|
|
";
|
|
|
|
[Test]
|
|
public async Task RemoveLegsFallTest()
|
|
{
|
|
await using var pair = await PoolManager.GetServerClient();
|
|
var server = pair.Server;
|
|
|
|
EntityUid human = default!;
|
|
AppearanceComponent appearance = null;
|
|
|
|
var entityManager = server.ResolveDependency<IEntityManager>();
|
|
var mapManager = server.ResolveDependency<IMapManager>();
|
|
var appearanceSystem = entityManager.System<SharedAppearanceSystem>();
|
|
|
|
await server.WaitAssertion(() =>
|
|
{
|
|
var mapId = mapManager.CreateMap();
|
|
BodyComponent body = null;
|
|
|
|
human = entityManager.SpawnEntity("HumanBodyAndAppearanceDummy",
|
|
new MapCoordinates(Vector2.Zero, mapId));
|
|
|
|
Assert.Multiple(() =>
|
|
{
|
|
Assert.That(entityManager.TryGetComponent(human, out body));
|
|
Assert.That(entityManager.TryGetComponent(human, out appearance));
|
|
});
|
|
|
|
Assert.That(!appearanceSystem.TryGetData(human, RotationVisuals.RotationState, out RotationState _, appearance));
|
|
|
|
var bodySystem = entityManager.System<BodySystem>();
|
|
var legs = bodySystem.GetBodyChildrenOfType(human, BodyPartType.Leg, body);
|
|
|
|
foreach (var leg in legs)
|
|
{
|
|
bodySystem.DropPart(leg.Id, leg.Component);
|
|
}
|
|
});
|
|
|
|
await server.WaitAssertion(() =>
|
|
{
|
|
#pragma warning disable NUnit2045
|
|
// Interdependent assertions.
|
|
Assert.That(appearanceSystem.TryGetData(human, RotationVisuals.RotationState, out RotationState state, appearance));
|
|
Assert.That(state, Is.EqualTo(RotationState.Horizontal));
|
|
#pragma warning restore NUnit2045
|
|
});
|
|
await pair.CleanReturnAsync();
|
|
}
|
|
}
|
|
}
|