57 lines
2.1 KiB
C#
57 lines
2.1 KiB
C#
using System.Threading.Tasks;
|
|
using Content.Server.Clothing.Components;
|
|
using Content.Server.Inventory;
|
|
using Content.Shared.Clothing.EntitySystems;
|
|
using Content.Shared.Inventory;
|
|
using Content.Shared.Item;
|
|
using NUnit.Framework;
|
|
using Robust.Shared.Containers;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Map;
|
|
|
|
namespace Content.IntegrationTests.Tests
|
|
{
|
|
[TestFixture]
|
|
public sealed class DeleteInventoryTest
|
|
{
|
|
// Test that when deleting an entity with an InventoryComponent,
|
|
// any equipped items also get deleted.
|
|
[Test]
|
|
public async Task Test()
|
|
{
|
|
await using var pairTracker = await PoolManager.GetServerClient(new PoolSettings{NoClient = true});
|
|
var server = pairTracker.Pair.Server;
|
|
|
|
await server.WaitAssertion(() =>
|
|
{
|
|
// Spawn everything.
|
|
var mapMan = IoCManager.Resolve<IMapManager>();
|
|
var invSystem = IoCManager.Resolve<IEntitySystemManager>().GetEntitySystem<InventorySystem>();
|
|
|
|
mapMan.CreateNewMapEntity(MapId.Nullspace);
|
|
|
|
var entMgr = IoCManager.Resolve<IEntityManager>();
|
|
var container = entMgr.SpawnEntity(null, MapCoordinates.Nullspace);
|
|
entMgr.AddComponent<ServerInventoryComponent>(container);
|
|
entMgr.AddComponent<ContainerManagerComponent>(container);
|
|
|
|
var child = entMgr.SpawnEntity(null, MapCoordinates.Nullspace);
|
|
var item = entMgr.AddComponent<ClothingComponent>(child);
|
|
|
|
IoCManager.Resolve<IEntitySystemManager>().GetEntitySystem<ClothingSystem>().SetSlots(item.Owner, SlotFlags.HEAD, item);
|
|
|
|
// Equip item.
|
|
Assert.That(invSystem.TryEquip(container, child, "head"), Is.True);
|
|
|
|
// Delete parent.
|
|
entMgr.DeleteEntity(container);
|
|
|
|
// Assert that child item was also deleted.
|
|
Assert.That(item.Deleted, Is.True);
|
|
});
|
|
await pairTracker.CleanReturnAsync();
|
|
}
|
|
}
|
|
}
|