AdamuSw/tests/MUnique.OpenMU.Tests/Offline/ItemPickupHandlerTests.cs

// <copyright file="ItemPickupHandlerTests.cs" company="MUnique">
// Licensed under the MIT License. See LICENSE file in the project root for full license information.
// </copyright>

namespace MUnique.OpenMU.Tests.Offline;

using Moq;
using MUnique.OpenMU.DataModel.Configuration.Items;
using MUnique.OpenMU.DataModel.Entities;
using MUnique.OpenMU.GameLogic;
using MUnique.OpenMU.GameLogic.MuHelper;
using MUnique.OpenMU.GameLogic.Offline;
using MUnique.OpenMU.GameServer.RemoteView.MuHelper;

/// <summary>
/// Tests for <see cref="ItemPickupHandler"/>.
/// </summary>
[TestFixture]
public class ItemPickupHandlerTests
{
    private IGameContext _gameContext = null!;

    /// <summary>
    /// Sets up a fresh game context before each test.
    /// </summary>
    [SetUp]
    public void SetUp()
    {
        this._gameContext = GameContextTestHelper.CreateGameContext();
    }

    /// <summary>
    /// Tests that pickup does nothing when all pickup options are disabled.
    /// </summary>
    [Test]
    public async ValueTask DoesNothingWhenAllDisabledAsync()
    {
        // Arrange
        var player = await this.CreateOfflinePlayerAsync().ConfigureAwait(false);
        var config = new MuHelperSettings
        {
            PickAllItems = false,
            PickJewel = false,
            PickAncient = false,
            PickZen = false,
            PickExcellent = false,
        };

        var handler = new ItemPickupHandler(player, config);

        // Act
        await handler.PickupItemsAsync().ConfigureAwait(false);
    }

    /// <summary>
    /// Tests that pickup does nothing when config is null.
    /// </summary>
    [Test]
    public async ValueTask DoesNothingWhenConfigNullAsync()
    {
        // Arrange
        var player = await this.CreateOfflinePlayerAsync().ConfigureAwait(false);
        var handler = new ItemPickupHandler(player, null);

        // Act
        await handler.PickupItemsAsync().ConfigureAwait(false);
    }

    private async ValueTask<OfflinePlayer> CreateOfflinePlayerAsync()
    {
        return await PlayerTestHelper.CreateOfflineLevelingPlayerAsync(this._gameContext).ConfigureAwait(false);
    }

    private Item CreateItem(byte group, short number)
    {
        var definition = new Mock<ItemDefinition>();
        definition.SetupAllProperties();
        definition.Object.Group = group;
        definition.Object.Number = number;
        definition.Object.Width = 1;
        definition.Object.Height = 1;

        var item = new Mock<Item>();
        item.SetupAllProperties();
        item.Setup(i => i.Definition).Returns(definition.Object);
        item.Setup(i => i.ItemOptions).Returns(new List<ItemOptionLink>());
        item.Setup(i => i.ItemSetGroups).Returns(new List<ItemOfItemSet>());
        return item.Object;
    }

    /// <summary>
    /// Tests that a jewel pickup option only picks up actual jewels.
    /// </summary>
    [Test]
    public async ValueTask PickJewel_PicksUpJewelsOnlyAsync()
    {
        // Arrange
        var player = await this.CreateOfflinePlayerAsync().ConfigureAwait(false);
        Assert.That(player.CurrentMap, Is.Not.Null);

        var config = new MuHelperSettings
        {
            PickSelectItems = true,
            PickJewel = true,
            ObtainRange = 5,
        };

        var handler = new ItemPickupHandler(player, config);

        var bless = this.CreateItem(14, 13);
        var chaos = this.CreateItem(12, 15);
        var potion = this.CreateItem(14, 0);
        var eye = this.CreateItem(14, 17);

        var blessDrop = new DroppedItem(bless, player.Position, player.CurrentMap!, player);
        var chaosDrop = new DroppedItem(chaos, player.Position, player.CurrentMap!, player);
        var potionDrop = new DroppedItem(potion, player.Position, player.CurrentMap!, player);
        var eyeDrop = new DroppedItem(eye, player.Position, player.CurrentMap!, player);

        await player.CurrentMap!.AddAsync(blessDrop).ConfigureAwait(false);
        await player.CurrentMap.AddAsync(chaosDrop).ConfigureAwait(false);
        await player.CurrentMap.AddAsync(potionDrop).ConfigureAwait(false);
        await player.CurrentMap.AddAsync(eyeDrop).ConfigureAwait(false);

        // Act
        await handler.PickupItemsAsync().ConfigureAwait(false);

        // Assert
        Assert.That(player.CurrentMap.GetObject(blessDrop.Id), Is.Null, "Bless should be picked up");
        Assert.That(player.CurrentMap.GetObject(chaosDrop.Id), Is.Null, "Chaos should be picked up");
        Assert.That(player.CurrentMap.GetObject(potionDrop.Id), Is.Not.Null, "Potion should NOT be picked up");
        Assert.That(player.CurrentMap.GetObject(eyeDrop.Id), Is.Not.Null, "Devil's Eye should NOT be picked up");
    }
}