Gravity (#841)

Content.IntegrationTests/Tests/GravityGridTest.cs

@@ -0,0 +1,64 @@
+using System.Threading.Tasks;
+using Content.Client.GameObjects.Components.Gravity;
+using Content.Server.GameObjects.Components.Gravity;
+using Content.Server.GameObjects.Components.Power;
+using NUnit.Framework;
+using Robust.Shared.Interfaces.GameObjects;
+using Robust.Shared.Interfaces.Map;
+using Robust.Shared.IoC;
+using Robust.Shared.Map;
+using Robust.Shared.Maths;
+
+namespace Content.IntegrationTests.Tests
+{
+    /// Tests the behavior of GravityGeneratorComponent,
+    /// making sure that gravity is applied to the correct grids.
+    [TestFixture]
+    [TestOf(typeof(GravityGeneratorComponent))]
+    public class GravityGridTest : ContentIntegrationTest
+    {
+        [Test]
+        public async Task Test()
+        {
+            var server = StartServerDummyTicker();
+
+            IEntity generator = null;
+
+            IMapGrid grid1 = null;
+            IMapGrid grid2 = null;
+
+            // Create grids
+            server.Assert(() =>
+            {
+                var mapMan = IoCManager.Resolve<IMapManager>();
+
+                mapMan.CreateMap(new MapId(1));
+                grid1 = mapMan.CreateGrid(new MapId(1));
+                grid2 = mapMan.CreateGrid(new MapId(1));
+
+                var entityMan = IoCManager.Resolve<IEntityManager>();
+
+                generator = entityMan.SpawnEntity("GravityGenerator", new GridCoordinates(new Vector2(0, 0), grid2.Index));
+                Assert.That(generator.HasComponent<GravityGeneratorComponent>());
+                Assert.That(generator.HasComponent<PowerDeviceComponent>());
+                var generatorComponent = generator.GetComponent<GravityGeneratorComponent>();
+                var powerComponent = generator.GetComponent<PowerDeviceComponent>();
+                Assert.AreEqual(generatorComponent.Status, GravityGeneratorStatus.Unpowered);
+                powerComponent.ExternalPowered = true;
+            });
+            server.RunTicks(1);
+
+            server.Assert(() =>
+            {
+                var generatorComponent = generator.GetComponent<GravityGeneratorComponent>();
+
+                Assert.AreEqual(generatorComponent.Status, GravityGeneratorStatus.On);
+
+                Assert.That(!grid1.HasGravity);
+                Assert.That(grid2.HasGravity);
+            });
+
+            await server.WaitIdleAsync();
+        }
+    }
+}