RobustToolbox/Robust.UnitTesting/Client/GameObjects/Components/TransformComponentTests.cs

using NUnit.Framework;
using Robust.Shared.Containers;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Map;
using Robust.Shared.Maths;
using Robust.Shared.Physics;
using Robust.UnitTesting.Server;

namespace Robust.UnitTesting.Client.GameObjects.Components
{
    [TestFixture]
    [TestOf(typeof(TransformComponent))]
    public sealed class TransformComponentTests
    {
        private static readonly MapId TestMapId = new(1);
        private static readonly GridId TestGridAId = new(1);
        private static readonly GridId TestGridBId = new(2);

        private static ISimulation SimulationFactory()
        {
            var sim = RobustServerSimulation
                .NewSimulation()
                .InitializeInstance();

            var mapManager = sim.Resolve<IMapManager>();

            // Adds the map with id 1, and spawns entity 1 as the map entity.
            mapManager.CreateMap(TestMapId);

            // Adds two grids to use in tests.
            mapManager.CreateGrid(TestMapId, TestGridAId);
            mapManager.CreateGrid(TestMapId, TestGridBId);

            return sim;
        }

        /// <summary>
        ///     Make sure that component state locations are RELATIVE.
        /// </summary>
        [Test]
        public void ComponentStatePositionTest()
        {
            var sim = SimulationFactory();
            var entMan = sim.Resolve<IEntityManager>();
            var mapMan = sim.Resolve<IMapManager>();

            var gridA = mapMan.GetGrid(TestGridAId);
            var gridB = mapMan.GetGrid(TestGridBId);

            // Arrange
            var initialPos = new EntityCoordinates(gridA.GridEntityId, (0, 0));
            var parent = entMan.SpawnEntity(null, initialPos);
            var child = entMan.SpawnEntity(null, initialPos);
            var parentTrans = IoCManager.Resolve<IEntityManager>().GetComponent<TransformComponent>(parent);
            var childTrans = IoCManager.Resolve<IEntityManager>().GetComponent<TransformComponent>(child);

            var compState = new TransformComponent.TransformComponentState(new Vector2(5, 5), new Angle(0), gridB.GridEntityId, false, false);
            parentTrans.HandleComponentState(compState, null);

            compState = new TransformComponent.TransformComponentState(new Vector2(6, 6), new Angle(0), gridB.GridEntityId, false, false);
            childTrans.HandleComponentState(compState, null);
            // World pos should be 6, 6 now.

            // Act
            var oldWpos = childTrans.WorldPosition;
            compState = new TransformComponent.TransformComponentState(new Vector2(1, 1), new Angle(0), parent, false, false);
            childTrans.HandleComponentState(compState, null);
            var newWpos = childTrans.WorldPosition;

            // Assert
            Assert.That(newWpos, Is.EqualTo(oldWpos));
        }

        /// <summary>
        ///     Tests that world rotation is built properly
        /// </summary>
        [Test]
        public void WorldRotationTest()
        {
            var sim = SimulationFactory();
            var entMan = sim.Resolve<IEntityManager>();
            var mapMan = sim.Resolve<IMapManager>();

            var gridA = mapMan.GetGrid(TestGridAId);
            var gridB = mapMan.GetGrid(TestGridBId);

            // Arrange
            var initalPos = new EntityCoordinates(gridA.GridEntityId, (0, 0));
            var node1 = entMan.SpawnEntity(null, initalPos);
            var node2 = entMan.SpawnEntity(null, initalPos);
            var node3 = entMan.SpawnEntity(null, initalPos);

            IoCManager.Resolve<IEntityManager>().GetComponent<MetaDataComponent>(node1).EntityName = "node1_dummy";
            IoCManager.Resolve<IEntityManager>().GetComponent<MetaDataComponent>(node2).EntityName = "node2_dummy";
            IoCManager.Resolve<IEntityManager>().GetComponent<MetaDataComponent>(node3).EntityName = "node3_dummy";

            var node1Trans = IoCManager.Resolve<IEntityManager>().GetComponent<TransformComponent>(node1);
            var node2Trans = IoCManager.Resolve<IEntityManager>().GetComponent<TransformComponent>(node2);
            var node3Trans = IoCManager.Resolve<IEntityManager>().GetComponent<TransformComponent>(node3);

            var compState = new TransformComponent.TransformComponentState(new Vector2(6, 6), Angle.FromDegrees(135), gridB.GridEntityId, false, false);
            node1Trans.HandleComponentState(compState, null);
            compState = new TransformComponent.TransformComponentState(new Vector2(1, 1), Angle.FromDegrees(45), node1, false, false);
            node2Trans.HandleComponentState(compState, null);
            compState = new TransformComponent.TransformComponentState(new Vector2(0, 0), Angle.FromDegrees(45), node2, false, false);
            node3Trans.HandleComponentState(compState, null);

            // Act
            var result = node3Trans.WorldRotation;

            // Assert (135 + 45 + 45 = 225)
            Assert.That(result, new ApproxEqualityConstraint(Angle.FromDegrees(225)));
        }
    }
}