tbd-station-14/Content.IntegrationTests/Tests/Interaction/InRangeUnobstructed.cs

using System.Threading.Tasks;
using Content.Client.Utility;
using Content.Shared.GameObjects.EntitySystems;
using Content.Shared.Utility;
using NUnit.Framework;
using Robust.Server.GameObjects.Components.Container;
using Robust.Shared.Interfaces.GameObjects;
using Robust.Shared.Interfaces.GameObjects.Components;
using Robust.Shared.Interfaces.Map;
using Robust.Shared.Map;

namespace Content.IntegrationTests.Tests.Interaction
{
    [TestFixture]
    [TestOf(typeof(SharedInteractionSystem))]
    [TestOf(typeof(SharedRangeExtensions))]
    [TestOf(typeof(RangeExtensions))]
    public class InRangeUnobstructed : ContentIntegrationTest
    {
        private const string HumanId = "BaseHumanMob_Content";

        private const float InteractionRange = SharedInteractionSystem.InteractionRange;

        private const float InteractionRangeDivided15 = InteractionRange / 1.5f;

        private readonly (float, float) _interactionRangeDivided15X = (InteractionRangeDivided15, 0f);

        private const float InteractionRangeDivided15Times3 = InteractionRangeDivided15 * 3;

        [Test]
        public async Task EntityEntityTest()
        {
            var server = StartServerDummyTicker();

            await server.WaitIdleAsync();

            var entityManager = server.ResolveDependency<IEntityManager>();
            var mapManager = server.ResolveDependency<IMapManager>();

            IEntity origin = null;
            IEntity other = null;
            IContainer container = null;
            IComponent component = null;
            GridCoordinates gridCoordinates = default;
            MapCoordinates mapCoordinates = default;

            server.Assert(() =>
            {
                mapManager.CreateNewMapEntity(MapId.Nullspace);
                var coordinates = MapCoordinates.Nullspace;

                origin = entityManager.SpawnEntity(HumanId, coordinates);
                other = entityManager.SpawnEntity(HumanId, coordinates);
                container = ContainerManagerComponent.Ensure<Container>("InRangeUnobstructedTestOtherContainer", other);
                component = other.Transform;
                gridCoordinates = other.Transform.GridPosition;
                mapCoordinates = other.Transform.MapPosition;
            });

            await server.WaitIdleAsync();

            server.Assert(() =>
            {
                // Entity <-> Entity
                Assert.True(origin.InRangeUnobstructed(other));
                Assert.True(other.InRangeUnobstructed(origin));

                // Entity <-> Component
                Assert.True(origin.InRangeUnobstructed(component));
                Assert.True(component.InRangeUnobstructed(origin));

                // Entity <-> Container
                Assert.True(origin.InRangeUnobstructed(container));
                Assert.True(container.InRangeUnobstructed(origin));

                // Entity <-> GridCoordinates
                Assert.True(origin.InRangeUnobstructed(gridCoordinates));
                Assert.True(gridCoordinates.InRangeUnobstructed(origin));

                // Entity <-> MapCoordinates
                Assert.True(origin.InRangeUnobstructed(mapCoordinates));
                Assert.True(mapCoordinates.InRangeUnobstructed(origin));


                // Move them slightly apart
                origin.Transform.LocalPosition += _interactionRangeDivided15X;

                // Entity <-> Entity
                Assert.True(origin.InRangeUnobstructed(other));
                Assert.True(other.InRangeUnobstructed(origin));

                // Entity <-> Component
                Assert.True(origin.InRangeUnobstructed(component));
                Assert.True(component.InRangeUnobstructed(origin));

                // Entity <-> Container
                Assert.True(origin.InRangeUnobstructed(container));
                Assert.True(container.InRangeUnobstructed(origin));

                // Entity <-> GridCoordinates
                Assert.True(origin.InRangeUnobstructed(gridCoordinates));
                Assert.True(gridCoordinates.InRangeUnobstructed(origin));

                // Entity <-> MapCoordinates
                Assert.True(origin.InRangeUnobstructed(mapCoordinates));
                Assert.True(mapCoordinates.InRangeUnobstructed(origin));


                // Move them out of range
                origin.Transform.LocalPosition += _interactionRangeDivided15X;

                // Entity <-> Entity
                Assert.False(origin.InRangeUnobstructed(other));
                Assert.False(other.InRangeUnobstructed(origin));

                // Entity <-> Component
                Assert.False(origin.InRangeUnobstructed(component));
                Assert.False(component.InRangeUnobstructed(origin));

                // Entity <-> Container
                Assert.False(origin.InRangeUnobstructed(container));
                Assert.False(container.InRangeUnobstructed(origin));

                // Entity <-> GridCoordinates
                Assert.False(origin.InRangeUnobstructed(gridCoordinates));
                Assert.False(gridCoordinates.InRangeUnobstructed(origin));

                // Entity <-> MapCoordinates
                Assert.False(origin.InRangeUnobstructed(mapCoordinates));
                Assert.False(mapCoordinates.InRangeUnobstructed(origin));


                // Checks with increased range

                // Entity <-> Entity
                Assert.True(origin.InRangeUnobstructed(other, InteractionRangeDivided15Times3));
                Assert.True(other.InRangeUnobstructed(origin, InteractionRangeDivided15Times3));

                // Entity <-> Component
                Assert.True(origin.InRangeUnobstructed(component, InteractionRangeDivided15Times3));
                Assert.True(component.InRangeUnobstructed(origin, InteractionRangeDivided15Times3));

                // Entity <-> Container
                Assert.True(origin.InRangeUnobstructed(container, InteractionRangeDivided15Times3));
                Assert.True(container.InRangeUnobstructed(origin, InteractionRangeDivided15Times3));

                // Entity <-> GridCoordinates
                Assert.True(origin.InRangeUnobstructed(gridCoordinates, InteractionRangeDivided15Times3));
                Assert.True(gridCoordinates.InRangeUnobstructed(origin, InteractionRangeDivided15Times3));

                // Entity <-> MapCoordinates
                Assert.True(origin.InRangeUnobstructed(mapCoordinates, InteractionRangeDivided15Times3));
                Assert.True(mapCoordinates.InRangeUnobstructed(origin, InteractionRangeDivided15Times3));
            });

            await server.WaitIdleAsync();
        }
    }
}