RobustToolbox/Robust.UnitTesting/Shared/Physics/PhysicsComponent_Test.cs

using System.Collections.Generic;
using System.Threading.Tasks;
using NUnit.Framework;
using Robust.Shared.GameObjects;
using Robust.Shared.Map;
using Robust.Shared.Maths;
using Robust.Shared.Physics;
using Robust.Shared.Physics.Collision.Shapes;
using Robust.Shared.Physics.Dynamics;

namespace Robust.UnitTesting.Shared.Physics
{
    [TestFixture]
    [TestOf(typeof(PhysicsComponent))]
    public class PhysicsComponent_Test : RobustIntegrationTest
    {
        private ServerIntegrationInstance _server = default!;

        [OneTimeSetUp]
        public async Task Setup()
        {
            _server = StartServer();
            await _server.WaitIdleAsync();
            var mapManager = _server.ResolveDependency<IMapManager>();
            mapManager.CreateMap();
        }

        [Test]
        public async Task TestPointLinearImpulse()
        {
            await _server.WaitIdleAsync();
            var entManager = _server.ResolveDependency<IEntityManager>();
            var fixtureSystem = _server.ResolveDependency<IEntitySystemManager>()
                .GetEntitySystem<FixtureSystem>();

            await _server.WaitAssertion(() =>
            {
                var boxEnt = entManager.SpawnEntity(null, new MapCoordinates(Vector2.Zero, new MapId(1)));
                var box = boxEnt.AddComponent<PhysicsComponent>();
                var poly = new PolygonShape();
                poly.SetAsBox(0.5f, 0.5f);
                var fixture = fixtureSystem.CreateFixture(box, poly);
                fixture.Mass = 1f;
                box.FixedRotation = false;
                box.BodyType = BodyType.Dynamic;
                Assert.That(box.InvI, Is.GreaterThan(0f));

                // Check regular impulse works
                box.ApplyLinearImpulse(new Vector2(0f, 1f));
                Assert.That(box.LinearVelocity.Length, Is.GreaterThan(0f));

                // Reset the box
                box.LinearVelocity = Vector2.Zero;
                Assert.That(box.LinearVelocity.Length, Is.EqualTo(0f));
                Assert.That(box.AngularVelocity, Is.EqualTo(0f));

                // Check the angular impulse is applied from the point
                box.ApplyLinearImpulse(new Vector2(0f, 1f), new Vector2(0.5f, 0f));
                Assert.That(box.LinearVelocity.Length, Is.GreaterThan(0f));
                Assert.That(box.AngularVelocity, Is.Not.EqualTo(0f));
            });
        }
    }
}