RobustToolbox/Robust.UnitTesting/Shared/GameObjects/Systems/TransformSystemTests.cs

using System.Numerics;
using NUnit.Framework;
using Robust.Server.GameObjects;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Map;
using Robust.UnitTesting.Server;

namespace Robust.UnitTesting.Shared.GameObjects.Systems
{
    [TestFixture, Parallelizable]
    sealed class TransformSystemTests
    {
        private static ISimulation SimulationFactory()
        {
            var sim = RobustServerSimulation
                .NewSimulation()
                .RegisterEntitySystems(f => f.LoadExtraSystemType<AnchoredSystemTests.MoveEventTestSystem>())
                .InitializeInstance();

            return sim;
        }

        /// <summary>
        /// When the local position of the transform changes, a MoveEvent is raised.
        /// </summary>
        [Test]
        public void OnMove_LocalPosChanged_RaiseMoveEvent()
        {
            var sim = SimulationFactory();
            var entMan = sim.Resolve<IEntityManager>();
            var map = sim.CreateMap().MapId;
            var ent1 = entMan.SpawnEntity(null, new MapCoordinates(Vector2.Zero, map));

            entMan.System<AnchoredSystemTests.MoveEventTestSystem>().ResetCounters();
            entMan.System<TransformSystem>().SetLocalPosition(ent1, Vector2.One);
            entMan.System<AnchoredSystemTests.MoveEventTestSystem>().AssertMoved(false);
        }

        /// <summary>
        /// Checks that the MoverCoordinates between parent and children is correct.
        /// </summary>
        [Test]
        public void MoverCoordinatesCorrect()
        {
            var sim = SimulationFactory();
            var entManager = sim.Resolve<IEntityManager>();
            var xformSystem = sim.Resolve<IEntitySystemManager>().GetEntitySystem<SharedTransformSystem>();
            var mapId = sim.CreateMap().MapId;

            var parent = entManager.SpawnEntity(null, new MapCoordinates(Vector2.One, mapId));
            var parentXform = entManager.GetComponent<TransformComponent>(parent);
            Assert.That(parentXform.LocalPosition, Is.EqualTo(Vector2.One));

            var child1 = entManager.SpawnEntity(null, new MapCoordinates(Vector2.One, mapId));
            var child2 = entManager.SpawnEntity(null, new MapCoordinates(new Vector2(10f, 10f), mapId));

            var child1Xform = entManager.GetComponent<TransformComponent>(child1);
            var child2Xform = entManager.GetComponent<TransformComponent>(child2);
            xformSystem.SetParent(child1, child1Xform, parent, parentXform: parentXform);
            xformSystem.SetParent(child2, child2Xform, parent, parentXform: parentXform);

            var mover1 = xformSystem.GetMoverCoordinates(child1, child1Xform);
            var mover2 = xformSystem.GetMoverCoordinates(child2, child2Xform);

            Assert.That(mover1.Position, Is.EqualTo(Vector2.One));
            Assert.That(mover2.Position, Is.EqualTo(new Vector2(10f, 10f)));

            var child3 = entManager.SpawnEntity(null, new MapCoordinates(Vector2.One, mapId));
            var child3Xform = entManager.GetComponent<TransformComponent>(child3);
            xformSystem.SetParent(child3, child3Xform, child2, parentXform: child2Xform);

            Assert.That(xformSystem.GetMoverCoordinates(child3, child3Xform).Position, Is.EqualTo(Vector2.One));
        }

        /// <summary>
        /// Asserts that when a transformcomponent is detached to null all of its children update their mapids.
        /// </summary>
        [Test]
        public void DetachMapRecursive()
        {
            var sim = SimulationFactory();
            var entManager = sim.Resolve<IEntityManager>();
            var xformSystem = sim.Resolve<IEntitySystemManager>().GetEntitySystem<SharedTransformSystem>();
            var mapId = sim.CreateMap().MapId;

            var parent = entManager.SpawnEntity(null, new MapCoordinates(Vector2.One, mapId));
            var parentXform = entManager.GetComponent<TransformComponent>(parent);

            var child = entManager.SpawnEntity(null, new EntityCoordinates(parent, Vector2.Zero));
            var childXform = entManager.GetComponent<TransformComponent>(child);

            Assert.That(parentXform.MapID, Is.EqualTo(mapId));
            Assert.That(childXform.MapID, Is.EqualTo(mapId));

            xformSystem.DetachEntity(parent, parentXform);
            Assert.That(parentXform.MapID, Is.EqualTo(MapId.Nullspace));
            Assert.That(childXform.MapID, Is.EqualTo(MapId.Nullspace));
        }

        private sealed class Subscriber : IEntityEventSubscriber { }
    }
}