151 lines
4.9 KiB
C#
151 lines
4.9 KiB
C#
using Content.Shared.Inventory;
|
|
using Robust.Shared.GameObjects;
|
|
|
|
namespace Content.IntegrationTests.Tests
|
|
{
|
|
// Tests the behavior of InventoryComponent.
|
|
// i.e. the interaction between uniforms and the pocket/ID slots.
|
|
// and also how big items don't fit in pockets.
|
|
[TestFixture]
|
|
public sealed class HumanInventoryUniformSlotsTest
|
|
{
|
|
private const string Prototypes = @"
|
|
- type: entity
|
|
name: HumanDummy
|
|
id: HumanDummy
|
|
components:
|
|
- type: Inventory
|
|
- type: ContainerContainer
|
|
|
|
- type: entity
|
|
name: UniformDummy
|
|
id: UniformDummy
|
|
components:
|
|
- type: Clothing
|
|
slots: [innerclothing]
|
|
- type: Item
|
|
size: 5
|
|
|
|
- type: entity
|
|
name: IDCardDummy
|
|
id: IDCardDummy
|
|
components:
|
|
- type: Clothing
|
|
slots:
|
|
- idcard
|
|
- type: Item
|
|
size: 5
|
|
- type: IdCard
|
|
|
|
- type: entity
|
|
name: FlashlightDummy
|
|
id: FlashlightDummy
|
|
components:
|
|
- type: Item
|
|
size: 5
|
|
|
|
- type: entity
|
|
name: ToolboxDummy
|
|
id: ToolboxDummy
|
|
components:
|
|
- type: Item
|
|
size: 9999
|
|
";
|
|
[Test]
|
|
public async Task Test()
|
|
{
|
|
await using var pairTracker = await PoolManager.GetServerClient(new PoolSettings { NoClient = true, ExtraPrototypes = Prototypes });
|
|
var server = pairTracker.Pair.Server;
|
|
var testMap = await PoolManager.CreateTestMap(pairTracker);
|
|
var coordinates = testMap.GridCoords;
|
|
|
|
EntityUid human = default;
|
|
EntityUid uniform = default;
|
|
EntityUid idCard = default;
|
|
EntityUid pocketItem = default;
|
|
|
|
InventorySystem invSystem = default!;
|
|
var entityMan = server.ResolveDependency<IEntityManager>();
|
|
|
|
await server.WaitAssertion(() =>
|
|
{
|
|
invSystem = entityMan.System<InventorySystem>();
|
|
|
|
human = entityMan.SpawnEntity("HumanDummy", coordinates);
|
|
uniform = entityMan.SpawnEntity("UniformDummy", coordinates);
|
|
idCard = entityMan.SpawnEntity("IDCardDummy", coordinates);
|
|
pocketItem = entityMan.SpawnEntity("FlashlightDummy", coordinates);
|
|
var tooBigItem = entityMan.SpawnEntity("ToolboxDummy", coordinates);
|
|
|
|
|
|
Assert.Multiple(() =>
|
|
{
|
|
Assert.That(invSystem.CanEquip(human, uniform, "jumpsuit", out _));
|
|
|
|
// Can't equip any of these since no uniform!
|
|
Assert.That(invSystem.CanEquip(human, idCard, "id", out _), Is.False);
|
|
Assert.That(invSystem.CanEquip(human, pocketItem, "pocket1", out _), Is.False);
|
|
Assert.That(invSystem.CanEquip(human, tooBigItem, "pocket2", out _), Is.False); // This one fails either way.
|
|
});
|
|
|
|
Assert.Multiple(() =>
|
|
{
|
|
Assert.That(invSystem.TryEquip(human, uniform, "jumpsuit"));
|
|
Assert.That(invSystem.TryEquip(human, idCard, "id"));
|
|
});
|
|
|
|
#pragma warning disable NUnit2045
|
|
Assert.That(invSystem.CanEquip(human, tooBigItem, "pocket1", out _), Is.False); // Still failing!
|
|
Assert.That(invSystem.TryEquip(human, pocketItem, "pocket1"));
|
|
#pragma warning restore NUnit2045
|
|
|
|
Assert.Multiple(() =>
|
|
{
|
|
Assert.That(IsDescendant(idCard, human, entityMan));
|
|
Assert.That(IsDescendant(pocketItem, human, entityMan));
|
|
});
|
|
|
|
// Now drop the jumpsuit.
|
|
Assert.That(invSystem.TryUnequip(human, "jumpsuit"));
|
|
});
|
|
|
|
await server.WaitRunTicks(2);
|
|
|
|
await server.WaitAssertion(() =>
|
|
{
|
|
Assert.Multiple(() =>
|
|
{
|
|
// Items have been dropped!
|
|
Assert.That(IsDescendant(uniform, human, entityMan), Is.False);
|
|
Assert.That(IsDescendant(idCard, human, entityMan), Is.False);
|
|
Assert.That(IsDescendant(pocketItem, human, entityMan), Is.False);
|
|
|
|
// Ensure everything null here.
|
|
Assert.That(!invSystem.TryGetSlotEntity(human, "jumpsuit", out _));
|
|
Assert.That(!invSystem.TryGetSlotEntity(human, "id", out _));
|
|
Assert.That(!invSystem.TryGetSlotEntity(human, "pocket1", out _));
|
|
});
|
|
});
|
|
|
|
await pairTracker.CleanReturnAsync();
|
|
}
|
|
|
|
private static bool IsDescendant(EntityUid descendant, EntityUid parent, IEntityManager entManager)
|
|
{
|
|
var xforms = entManager.GetEntityQuery<TransformComponent>();
|
|
var tmpParent = xforms.GetComponent(descendant).ParentUid;
|
|
while (tmpParent.IsValid())
|
|
{
|
|
if (tmpParent == parent)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
tmpParent = xforms.GetComponent(tmpParent).ParentUid;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|
|
}
|