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

using System.Linq;
using NUnit.Framework;
using Robust.Shared;
using Robust.Shared.Configuration;
using Robust.Shared.GameObjects;
using Robust.Shared.Map;
using Robust.Shared.Maths;
using Robust.UnitTesting.Server;

namespace Robust.UnitTesting.Shared.Map;

[TestFixture]
public sealed class GridSplit_Tests
{
    private ISimulation GetSim()
    {
        var sim = RobustServerSimulation.NewSimulation().InitializeInstance();

        var config = sim.Resolve<IConfigurationManager>();
        config.SetCVar(CVars.GridSplitting, true);

        return sim;
    }

    [Test]
    public void SimpleSplit()
    {
        var sim =GetSim();
        var mapManager = sim.Resolve<IMapManager>();
        var mapId = mapManager.CreateMap();
        var grid = mapManager.CreateGrid(mapId);

        for (var x = 0; x < 3; x++)
        {
            grid.SetTile(new Vector2i(x, 0), new Tile(1));
        }

        Assert.That(mapManager.GetAllMapGrids(mapId).Count(), Is.EqualTo(1));

        grid.SetTile(new Vector2i(1, 0), Tile.Empty);
        Assert.That(mapManager.GetAllMapGrids(mapId).Count(), Is.EqualTo(2));

        mapManager.DeleteMap(mapId);
    }

    [Test]
    public void DonutSplit()
    {
        var sim =GetSim();
        var mapManager = sim.Resolve<IMapManager>();
        var mapId = mapManager.CreateMap();
        var grid = mapManager.CreateGrid(mapId);

        for (var x = 0; x < 3; x++)
        {
            for (var y = 0; y < 3; y++)
            {
                grid.SetTile(new Vector2i(x, y), new Tile(1));
            }
        }

        Assert.That(mapManager.GetAllMapGrids(mapId).Count(), Is.EqualTo(1));

        grid.SetTile(Vector2i.One, Tile.Empty);
        Assert.That(mapManager.GetAllMapGrids(mapId).Count(), Is.EqualTo(1));

        grid.SetTile(new Vector2i(1, 2), Tile.Empty);
        Assert.That(mapManager.GetAllMapGrids(mapId).Count(), Is.EqualTo(1));

        grid.SetTile(new Vector2i(1, 0), Tile.Empty);
        Assert.That(mapManager.GetAllMapGrids(mapId).Count(), Is.EqualTo(2));

        mapManager.DeleteMap(mapId);
    }

    [Test]
    public void TriSplit()
    {
        var sim =GetSim();
        var mapManager = sim.Resolve<IMapManager>();
        var mapId = mapManager.CreateMap();
        var grid = mapManager.CreateGrid(mapId);

        for (var x = 0; x < 3; x++)
        {
            grid.SetTile(new Vector2i(x, 0), new Tile(1));
        }

        grid.SetTile(Vector2i.One, new Tile(1));

        Assert.That(mapManager.GetAllMapGrids(mapId).Count(), Is.EqualTo(1));

        grid.SetTile(new Vector2i(1, 0), Tile.Empty);
        Assert.That(mapManager.GetAllMapGrids(mapId).Count(), Is.EqualTo(3));

        mapManager.DeleteMap(mapId);
    }

    /// <summary>
    /// Checks GridId and Parents update correctly for re-parented entities.
    /// </summary>
    [Test]
    public void ReparentSplit()
    {
        var sim =GetSim();
        var entManager = sim.Resolve<IEntityManager>();
        var mapManager = sim.Resolve<IMapManager>();
        var mapId = mapManager.CreateMap();
        var grid = mapManager.CreateGrid(mapId);

        for (var x = 0; x < 4; x++)
        {
            grid.SetTile(new Vector2i(x, 0), new Tile(1));
        }

        Assert.That(mapManager.GetAllMapGrids(mapId).Count(), Is.EqualTo(1));

        var dummy = entManager.SpawnEntity(null, new EntityCoordinates(grid.GridEntityId, new Vector2(3.5f, 0.5f)));
        var dummyXform = entManager.GetComponent<TransformComponent>(dummy);
        var anchored = entManager.SpawnEntity(null, new EntityCoordinates(grid.GridEntityId, new Vector2(3.5f, 0.5f)));
        var anchoredXform = entManager.GetComponent<TransformComponent>(anchored);
        anchoredXform.Anchored = true;
        Assert.That(anchoredXform.Anchored);

        grid.SetTile(new Vector2i(2, 0), Tile.Empty);
        Assert.That(mapManager.GetAllMapGrids(mapId).Count(), Is.EqualTo(2));

        var newGrid = mapManager.GetAllMapGrids(mapId).Last();
        var newGridXform = entManager.GetComponent<TransformComponent>(newGrid.GridEntityId);

        Assert.Multiple(() =>
        {
            // Assertions baby
            Assert.That(anchoredXform.Anchored);
            Assert.That(anchoredXform.ParentUid, Is.EqualTo(newGrid.GridEntityId));
            Assert.That(anchoredXform.GridID, Is.EqualTo(newGrid.Index));
            Assert.That(newGridXform._children, Does.Contain(anchored));

            Assert.That(dummyXform.ParentUid, Is.EqualTo(newGrid.GridEntityId));
            Assert.That(dummyXform.GridID, Is.EqualTo(newGrid.Index));
            Assert.That(newGridXform._children, Does.Contain(dummy));
        });
        mapManager.DeleteMap(mapId);
    }
}