cfc0247e5c
* Cleanup PuddleTest * Cleanup GravityGridTest * Cleanup PowerTest * Cleanup SaveLoadMapTest * Cleanup Body tests * Cleanup ContainerOcclusionTest * Cleanup AirlockTest * Cleanup DamageableTest * Cleanup EntityTest * Cleanup FluidSpillTest * Cleanup FollowerSystemTest * Cleanup HandCuffTest * Cleanup InteractionSystemTests * Cleanup InRangeUnobstructed * Cleanup SimplePredictReconcileTest * Cleanup PostMapInitTest * Cleanup SalvageTest * Cleanup SaveLoadSaveTest * Cleanup ShuttleTest * Cleanup MaterialArbitrageTest * Cleanup PrototypeSaveTest * Fix ShuttleTest * Bunch of small ones * Move JobTests to Station directory * More small fixes * Cleanup InteractionTest.Helpers Had to change a method signature, so some callers were modified too. * Missed one
102 lines
3.6 KiB
C#
102 lines
3.6 KiB
C#
using Content.Server.Gravity;
|
|
using Content.Server.Power.Components;
|
|
using Content.Shared.Gravity;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.Map;
|
|
using Robust.Shared.Map.Components;
|
|
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 sealed class GravityGridTest
|
|
{
|
|
[TestPrototypes]
|
|
private const string Prototypes = @"
|
|
- type: entity
|
|
name: GridGravityGeneratorDummy
|
|
id: GridGravityGeneratorDummy
|
|
components:
|
|
- type: GravityGenerator
|
|
chargeRate: 1000000000 # Set this really high so it discharges in a single tick.
|
|
activePower: 500
|
|
- type: ApcPowerReceiver
|
|
- type: UserInterface
|
|
";
|
|
[Test]
|
|
public async Task Test()
|
|
{
|
|
await using var pair = await PoolManager.GetServerClient();
|
|
var server = pair.Server;
|
|
|
|
var testMap = await pair.CreateTestMap();
|
|
|
|
var entityMan = server.EntMan;
|
|
var mapMan = server.MapMan;
|
|
var mapSys = entityMan.System<SharedMapSystem>();
|
|
|
|
EntityUid generator = default;
|
|
Entity<MapGridComponent> grid1 = default;
|
|
Entity<MapGridComponent> grid2 = default;
|
|
|
|
// Create grids
|
|
await server.WaitAssertion(() =>
|
|
{
|
|
var mapId = testMap.MapId;
|
|
grid1 = mapMan.CreateGridEntity(mapId);
|
|
grid2 = mapMan.CreateGridEntity(mapId);
|
|
|
|
mapSys.SetTile(grid1, grid1, Vector2i.Zero, new Tile(1));
|
|
mapSys.SetTile(grid2, grid2, Vector2i.Zero, new Tile(1));
|
|
|
|
generator = entityMan.SpawnEntity("GridGravityGeneratorDummy", new EntityCoordinates(grid1, 0.5f, 0.5f));
|
|
Assert.Multiple(() =>
|
|
{
|
|
Assert.That(entityMan.HasComponent<GravityGeneratorComponent>(generator));
|
|
Assert.That(entityMan.HasComponent<ApcPowerReceiverComponent>(generator));
|
|
});
|
|
|
|
var powerComponent = entityMan.GetComponent<ApcPowerReceiverComponent>(generator);
|
|
powerComponent.NeedsPower = false;
|
|
});
|
|
|
|
await server.WaitRunTicks(5);
|
|
|
|
await server.WaitAssertion(() =>
|
|
{
|
|
var generatorComponent = entityMan.GetComponent<GravityGeneratorComponent>(generator);
|
|
var powerComponent = entityMan.GetComponent<ApcPowerReceiverComponent>(generator);
|
|
|
|
Assert.Multiple(() =>
|
|
{
|
|
Assert.That(generatorComponent.GravityActive, Is.True);
|
|
Assert.That(!entityMan.GetComponent<GravityComponent>(grid1).EnabledVV);
|
|
Assert.That(entityMan.GetComponent<GravityComponent>(grid2).EnabledVV);
|
|
});
|
|
|
|
// Re-enable needs power so it turns off again.
|
|
// Charge rate is ridiculously high so it finishes in one tick.
|
|
powerComponent.NeedsPower = true;
|
|
});
|
|
|
|
await server.WaitRunTicks(5);
|
|
|
|
await server.WaitAssertion(() =>
|
|
{
|
|
var generatorComponent = entityMan.GetComponent<GravityGeneratorComponent>(generator);
|
|
|
|
Assert.Multiple(() =>
|
|
{
|
|
Assert.That(generatorComponent.GravityActive, Is.False);
|
|
Assert.That(entityMan.GetComponent<GravityComponent>(grid2).EnabledVV, Is.False);
|
|
});
|
|
});
|
|
|
|
await pair.CleanReturnAsync();
|
|
}
|
|
}
|
|
}
|