CatStation/Content.IntegrationTests/Tests/Storage/StorageInteractionTest.cs

using Content.Client.UserInterface.Systems.Hotbar.Widgets;
using Content.Client.UserInterface.Systems.Storage.Controls;
using Content.IntegrationTests.Tests.Interaction;
using Content.Shared.Input;
using Content.Shared.PDA;
using Content.Shared.Storage;
using Content.Shared.Timing;
using Robust.Client.UserInterface;
using Robust.Shared.Containers;
using Robust.Shared.GameObjects;

namespace Content.IntegrationTests.Tests.Storage;

public sealed class StorageInteractionTest : InteractionTest
{
    /// <summary>
    /// Check that players can interact with items in storage if the storage UI is open
    /// </summary>
    [Test]
    public async Task UiInteractTest()
    {
        var sys = Server.System<SharedContainerSystem>();

        await SpawnTarget("ClothingBackpack");
        var backpack = ToServer(Target);

        // Initially no BUI is open.
        Assert.That(IsUiOpen(StorageComponent.StorageUiKey.Key), Is.False);
        Assert.That(IsUiOpen(PdaUiKey.Key), Is.False);

        await Server.WaitPost(() => SEntMan.RemoveComponent<UseDelayComponent>(STarget!.Value));
        await RunTicks(5);

        // Activating the backpack opens the UI
        await Activate();
        Assert.That(IsUiOpen(StorageComponent.StorageUiKey.Key), Is.True);
        Assert.That(IsUiOpen(PdaUiKey.Key), Is.False);

        // Activating it again closes the UI
        await Activate();
        Assert.That(IsUiOpen(StorageComponent.StorageUiKey.Key), Is.False);

        // Open it again
        await Activate();
        Assert.That(IsUiOpen(StorageComponent.StorageUiKey.Key), Is.True);

        // Pick up a PDA
        var pda = await PlaceInHands("PassengerPDA");
        var sPda = ToServer(pda);
        Assert.That(sys.IsEntityInContainer(sPda), Is.True);
        Assert.That(sys.TryGetContainingContainer((sPda, null), out var container));
        Assert.That(container!.Owner, Is.EqualTo(SPlayer));

        // Insert the PDA into the backpack
        await Interact();
        Assert.That(sys.TryGetContainingContainer((sPda, null), out container));
        Assert.That(container!.Owner, Is.EqualTo(backpack));

        // Use "e" / ActivateInWorld to open the PDA UI while it is still in the backpack.
        var ctrl = GetStorageControl(pda);
        await ClickControl(ctrl, ContentKeyFunctions.ActivateItemInWorld);
        await RunTicks(10);
        Assert.That(IsUiOpen(StorageComponent.StorageUiKey.Key), Is.True);
        Assert.That(IsUiOpen(PdaUiKey.Key), Is.True);

        // Click on the pda to pick it up and remove it from the backpack.
        await ClickControl(ctrl, ContentKeyFunctions.MoveStoredItem);
        await RunTicks(10);
        Assert.That(sys.TryGetContainingContainer((sPda, null), out container));
        Assert.That(container!.Owner, Is.EqualTo(SPlayer));

        // UIs should still be open
        Assert.That(IsUiOpen(StorageComponent.StorageUiKey.Key), Is.True);
        Assert.That(IsUiOpen(PdaUiKey.Key), Is.True);
    }

    /// <summary>
    /// Retrieve the control that corresponds to the given entity in the currently open storage UI.
    /// </summary>
    private ItemGridPiece GetStorageControl(NetEntity target)
    {
        var uid = ToClient(target);
        var hotbar = GetWidget<HotbarGui>();
        var storageContainer  = GetControlFromField<Control>(nameof(HotbarGui.SingleStorageContainer), hotbar);
        return GetControlFromChildren<ItemGridPiece>(c => c.Entity == uid, storageContainer);
    }
}