66 lines
2.3 KiB
C#
66 lines
2.3 KiB
C#
using System.Threading.Tasks;
|
|
using Content.Client.GameObjects.Components.Gravity;
|
|
using Content.Server.GameObjects.Components.Gravity;
|
|
using Content.Server.GameObjects.Components.Power.ApcNetComponents;
|
|
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<PowerReceiverComponent>());
|
|
var generatorComponent = generator.GetComponent<GravityGeneratorComponent>();
|
|
var powerComponent = generator.GetComponent<PowerReceiverComponent>();
|
|
Assert.AreEqual(generatorComponent.Status, GravityGeneratorStatus.Unpowered);
|
|
powerComponent.NeedsPower = false;
|
|
});
|
|
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();
|
|
}
|
|
}
|
|
}
|