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

using System.Threading.Tasks;
using Content.Client.Interactable;
using Content.Shared.Interaction;
using Content.Shared.Interaction.Helpers;
using NUnit.Framework;
using Robust.Shared.Containers;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Map;
using Robust.Shared.Maths;

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

        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 = StartServer();

            await server.WaitIdleAsync();

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

            EntityUid origin = default;
            EntityUid other = default;
            IContainer container = null;
            IComponent component = null;
            EntityCoordinates entityCoordinates = default;
            MapCoordinates mapCoordinates = default;

            server.Assert(() =>
            {
                var mapId = mapManager.CreateMap();
                var coordinates = new MapCoordinates(Vector2.Zero, mapId);

                origin = sEntities.SpawnEntity(HumanId, coordinates);
                other = sEntities.SpawnEntity(HumanId, coordinates);
                container = other.EnsureContainer<Container>("InRangeUnobstructedTestOtherContainer");
                component = sEntities.GetComponent<TransformComponent>(other);
                entityCoordinates = sEntities.GetComponent<TransformComponent>(other).Coordinates;
                mapCoordinates = sEntities.GetComponent<TransformComponent>(other).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 <-> EntityCoordinates
                Assert.True(origin.InRangeUnobstructed(entityCoordinates));
                Assert.True(entityCoordinates.InRangeUnobstructed(origin));

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


                // Move them slightly apart
                IoCManager.Resolve<IEntityManager>().GetComponent<TransformComponent>(origin).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 <-> EntityCoordinates
                Assert.True(origin.InRangeUnobstructed(entityCoordinates));
                Assert.True(entityCoordinates.InRangeUnobstructed(origin));

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


                // Move them out of range
                IoCManager.Resolve<IEntityManager>().GetComponent<TransformComponent>(origin).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 <-> EntityCoordinates
                Assert.False(origin.InRangeUnobstructed(entityCoordinates));
                Assert.False(entityCoordinates.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 <-> EntityCoordinates
                Assert.True(origin.InRangeUnobstructed(entityCoordinates, InteractionRangeDivided15Times3));
                Assert.True(entityCoordinates.InRangeUnobstructed(origin, InteractionRangeDivided15Times3));

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

            await server.WaitIdleAsync();
        }
    }
}