ProjectLighthouse/ProjectLighthouse.Tests.GameApiTests/Unit/Controllers/StatisticsControllerTests.cs

using System.Collections.Generic;
using System.Threading.Tasks;
using LBPUnion.ProjectLighthouse.Database;
using LBPUnion.ProjectLighthouse.Servers.GameServer.Controllers;
using LBPUnion.ProjectLighthouse.Tests.Helpers;
using LBPUnion.ProjectLighthouse.Types.Entities.Level;
using LBPUnion.ProjectLighthouse.Types.Serialization;
using LBPUnion.ProjectLighthouse.Types.Users;
using Microsoft.AspNetCore.Mvc;
using Xunit;

namespace ProjectLighthouse.Tests.GameApiTests.Unit.Controllers;

[Trait("Category", "Unit")]
public class StatisticsControllerTests
{
    [Fact]
    public async Task PlanetStats_ShouldReturnCorrectCounts_WhenEmpty()
    {
        await using DatabaseContext db = await MockHelper.GetTestDatabase();

        StatisticsController statsController = new(db);
        statsController.SetupTestController();

        const int expectedSlots = 0;
        const int expectedTeamPicks = 0;

        IActionResult result = await statsController.PlanetStats();

        PlanetStatsResponse statsResponse = result.CastTo<OkObjectResult, PlanetStatsResponse>();
        Assert.Equal(expectedSlots, statsResponse.TotalSlotCount);
        Assert.Equal(expectedTeamPicks, statsResponse.TeamPickCount);
    }

    [Fact]
    public async Task PlanetStats_ShouldReturnCorrectCounts_WhenNotEmpty()
    {
        List<SlotEntity> slots = new()
        {
            new SlotEntity
            {
                SlotId = 1,
                CreatorId = 1,
            },
            new SlotEntity
            {
                SlotId = 2,
                CreatorId = 1,
            },
            new SlotEntity
            {
                SlotId = 3,
                CreatorId = 1,
                TeamPick = true,
            },
        };
        await using DatabaseContext db = await MockHelper.GetTestDatabase(slots);

        StatisticsController statsController = new(db);
        statsController.SetupTestController();

        const int expectedSlots = 3;
        const int expectedTeamPicks = 1;

        IActionResult result = await statsController.PlanetStats();

        PlanetStatsResponse statsResponse = result.CastTo<OkObjectResult, PlanetStatsResponse>();
        Assert.Equal(expectedSlots, statsResponse.TotalSlotCount);
        Assert.Equal(expectedTeamPicks, statsResponse.TeamPickCount);
    }

    [Fact]
    public async Task PlanetStats_ShouldReturnCorrectCounts_WhenSlotsAreIncompatibleGameVersion()
    {
        List<SlotEntity> slots = new()
        {
            new SlotEntity
            {
                SlotId = 1,
                CreatorId = 1,
                GameVersion = GameVersion.LittleBigPlanet2,
            },
            new SlotEntity
            {
                SlotId = 2,
                CreatorId = 1,
                GameVersion = GameVersion.LittleBigPlanet2,
            },
            new SlotEntity
            {
                SlotId = 3,
                CreatorId = 1,
                TeamPick = true,
                GameVersion = GameVersion.LittleBigPlanet2,
            },
        };
        await using DatabaseContext dbMock = await MockHelper.GetTestDatabase(slots);

        StatisticsController statsController = new(dbMock);
        statsController.SetupTestController();

        const int expectedSlots = 0;
        const int expectedTeamPicks = 0;

        IActionResult result = await statsController.PlanetStats();

        PlanetStatsResponse statsResponse = result.CastTo<OkObjectResult, PlanetStatsResponse>();
        Assert.Equal(expectedSlots, statsResponse.TotalSlotCount);
        Assert.Equal(expectedTeamPicks, statsResponse.TeamPickCount);
    }

    [Fact]
    public async Task TotalLevelCount_ShouldReturnCorrectCount_WhenSlotsAreCompatible()
    {
        List<SlotEntity> slots = new()
        {
            new SlotEntity
            {
                SlotId = 1,
                CreatorId = 1,
                GameVersion = GameVersion.LittleBigPlanet1,
            },
            new SlotEntity
            {
                SlotId = 2,
                CreatorId = 1,
                GameVersion = GameVersion.LittleBigPlanet1,
            },
            new SlotEntity
            {
                SlotId = 3,
                CreatorId = 1,
                TeamPick = true,
                GameVersion = GameVersion.LittleBigPlanet1,
            },
        };
        await using DatabaseContext dbMock = await MockHelper.GetTestDatabase(slots);

        StatisticsController statsController = new(dbMock);
        statsController.SetupTestController();

        const string expectedTotal = "3";

        IActionResult result = await statsController.TotalLevelCount();

        string totalSlotsResponse = result.CastTo<OkObjectResult, string>();
        Assert.Equal(expectedTotal, totalSlotsResponse);
    }

    [Fact]
    public async Task TotalLevelCount_ShouldReturnCorrectCount_WhenSlotsAreNotCompatible()
    {
        List<SlotEntity> slots = new()
        {
            new SlotEntity
            {
                SlotId = 1,
                CreatorId = 1,
                GameVersion = GameVersion.LittleBigPlanet2,
            },
            new SlotEntity
            {
                SlotId = 2,
                CreatorId = 1,
                GameVersion = GameVersion.LittleBigPlanet2,
            },
            new SlotEntity
            {
                SlotId = 3,
                CreatorId = 1,
                TeamPick = true,
                GameVersion = GameVersion.LittleBigPlanet2,
            },
        }; 
        await using DatabaseContext dbMock = await MockHelper.GetTestDatabase(slots);

        StatisticsController statsController = new(dbMock);
        statsController.SetupTestController();

        const string expectedTotal = "0";

        IActionResult result = await statsController.TotalLevelCount();

        string totalSlots = result.CastTo<OkObjectResult, string>();
        Assert.Equal(expectedTotal, totalSlots);
    }
}