RobustToolbox/Robust.UnitTesting/Shared/Map/MapGridMap_Tests.cs

using System.Collections.Generic;
using System.Linq;
using NUnit.Framework;
using Robust.Shared.GameObjects;
using Robust.Shared.Map;
using Robust.Shared.Map.Components;
using Robust.Shared.Maths;
using Robust.UnitTesting.Server;

namespace Robust.UnitTesting.Shared.Map;

[TestFixture]
public sealed class MapGridMap_Tests
{
    /// <summary>
    /// Asserts FindGrids only returns the map once.
    /// </summary>
    [Test]
    public void FindGrids()
    {
        var sim = RobustServerSimulation.NewSimulation().InitializeInstance();

        var entManager = sim.Resolve<IEntityManager>();
        var mapManager = sim.Resolve<IMapManager>();
        var mapSystem = entManager.System<SharedMapSystem>();

        var mapId = sim.CreateMap().MapId;
        List<Entity<MapGridComponent>> grids = [];
        mapManager.FindGridsIntersecting(mapId, Box2.UnitCentered, ref grids);
        Assert.That(grids, Is.Empty);

        entManager.AddComponent<MapGridComponent>(mapSystem.GetMapOrInvalid(mapId));
        mapManager.FindGridsIntersecting(mapId, Box2.UnitCentered, ref grids);
        Assert.That(grids, Has.Count.EqualTo(1));
    }

    /// <summary>
    /// Asserts that adding <see cref="MapGridComponent"/> to an existing map with a grid on it doesn't explode.
    /// </summary>
    [Test]
    public void AddGridCompToMap()
    {
        var sim = RobustServerSimulation.NewSimulation().InitializeInstance();

        var entManager = sim.Resolve<IEntityManager>();
        var mapManager = sim.Resolve<IMapManager>();
        var mapSystem = entManager.System<SharedMapSystem>();

        var mapId = sim.CreateMap().MapId;
        mapManager.CreateGridEntity(mapId);

        Assert.DoesNotThrow(() =>
        {
            entManager.AddComponent<MapGridComponent>(mapSystem.GetMapOrInvalid(mapId));
            entManager.TickUpdate(0.016f, false);
        });

        mapSystem.DeleteMap(mapId);
    }
}