Wino-Mail/Wino.Core.Tests/Services/WinoAccountProfileServiceTests.cs

using System;
using System.Threading.Tasks;
using FluentAssertions;
using Moq;
using Wino.Core.Domain.Entities.Shared;
using Wino.Core.Domain.Enums;
using Wino.Core.Domain.Interfaces;
using Wino.Core.Domain.Models.Accounts;
using Wino.Mail.Api.Contracts.Auth;
using Wino.Mail.Api.Contracts.Common;
using Wino.Services;
using Wino.Core.Tests.Helpers;
using Xunit;

namespace Wino.Core.Tests.Services;

public class WinoAccountProfileServiceTests : IAsyncLifetime
{
    private readonly Mock<IWinoAccountApiClient> _apiClient = new();
    private readonly Mock<IStoreManagementService> _storeManagementService = new();
    private InMemoryDatabaseService _databaseService = null!;
    private WinoAccountProfileService _service = null!;

    public async Task InitializeAsync()
    {
        _databaseService = new InMemoryDatabaseService();
        await _databaseService.InitializeAsync();
        _service = new WinoAccountProfileService(_databaseService, _apiClient.Object, _storeManagementService.Object);
    }

    public async Task DisposeAsync()
    {
        await _databaseService.DisposeAsync();
    }

    [Fact]
    public async Task LoginAsync_ShouldPersistSingleActiveAccount()
    {
        var authResult = CreateAuthResult("first@example.com");

        _apiClient
            .Setup(x => x.LoginAsync("first@example.com", "pw", default))
            .ReturnsAsync(WinoAccountApiResult<AuthResultDto>.Success(authResult));

        var result = await _service.LoginAsync("first@example.com", "pw");

        result.IsSuccess.Should().BeTrue();
        result.Account.Should().NotBeNull();

        var persisted = await _databaseService.Connection.Table<WinoAccount>().ToListAsync();
        persisted.Should().ContainSingle();
        persisted[0].Email.Should().Be("first@example.com");
        persisted[0].AccessToken.Should().Be(authResult.AccessToken);
        persisted[0].RefreshToken.Should().Be(authResult.RefreshToken);
    }

    [Fact]
    public async Task LoginAsync_ShouldReplaceExistingActiveAccount()
    {
        _apiClient
            .Setup(x => x.LoginAsync("first@example.com", "pw", default))
            .ReturnsAsync(WinoAccountApiResult<AuthResultDto>.Success(CreateAuthResult("first@example.com")));

        _apiClient
            .Setup(x => x.LoginAsync("second@example.com", "pw", default))
            .ReturnsAsync(WinoAccountApiResult<AuthResultDto>.Success(CreateAuthResult("second@example.com")));

        await _service.LoginAsync("first@example.com", "pw");
        await _service.LoginAsync("second@example.com", "pw");

        var persisted = await _databaseService.Connection.Table<WinoAccount>().ToListAsync();
        persisted.Should().ContainSingle();
        persisted[0].Email.Should().Be("second@example.com");
    }

    [Fact]
    public async Task SignOutAsync_ShouldDeletePersistedAccount()
    {
        var authResult = CreateAuthResult("signout@example.com");

        _apiClient
            .Setup(x => x.LoginAsync("signout@example.com", "pw", default))
            .ReturnsAsync(WinoAccountApiResult<AuthResultDto>.Success(authResult));

        _apiClient
            .Setup(x => x.LogoutAsync(authResult.RefreshToken, default))
            .ReturnsAsync(ApiEnvelope<System.Text.Json.JsonElement>.Success(default));

        await _service.LoginAsync("signout@example.com", "pw");
        await _service.SignOutAsync();

        var persisted = await _databaseService.Connection.Table<WinoAccount>().ToListAsync();
        persisted.Should().BeEmpty();
    }

    [Fact]
    public async Task HasAddOnAsync_ShouldUseLegacyStoreForUnlimitedAccounts()
    {
        _storeManagementService
            .Setup(x => x.HasProductAsync(WinoAddOnProductType.UNLIMITED_ACCOUNTS))
            .ReturnsAsync(true);

        var hasAddOn = await _service.HasAddOnAsync(WinoAddOnProductType.UNLIMITED_ACCOUNTS);

        hasAddOn.Should().BeTrue();
    }

    [Fact]
    public async Task LoginAsync_ShouldPreserveEnvelopeErrorMessage()
    {
        _apiClient
            .Setup(x => x.LoginAsync("first@example.com", "pw", default))
            .ReturnsAsync(WinoAccountApiResult<AuthResultDto>.Failure(ApiErrorCodes.InvalidCredentials, "Password does not match this account."));

        var result = await _service.LoginAsync("first@example.com", "pw");

        result.IsSuccess.Should().BeFalse();
        result.ErrorCode.Should().Be(ApiErrorCodes.InvalidCredentials);
        result.ErrorMessage.Should().Be("Password does not match this account.");
    }

    [Fact]
    public async Task RefreshProfileAsync_ShouldPersistLatestProfileData()
    {
        var authResult = CreateAuthResult("first@example.com");

        _apiClient
            .Setup(x => x.LoginAsync("first@example.com", "pw", default))
            .ReturnsAsync(WinoAccountApiResult<AuthResultDto>.Success(authResult));

        _apiClient
            .Setup(x => x.GetCurrentUserAsync(default))
            .ReturnsAsync(ApiEnvelope<AuthUserDto>.Success(new AuthUserDto(
                authResult.User.UserId,
                "updated@example.com",
                "Premium",
                authResult.User.HasPassword,
                authResult.User.HasGoogleLogin,
                authResult.User.HasFacebookLogin)));

        await _service.LoginAsync("first@example.com", "pw");

        var result = await _service.RefreshProfileAsync();

        result.IsSuccess.Should().BeTrue();
        result.Account.Should().NotBeNull();
        result.Account!.Email.Should().Be("updated@example.com");
        result.Account.AccountStatus.Should().Be("Premium");

        var persisted = await _databaseService.Connection.Table<WinoAccount>().FirstOrDefaultAsync();
        persisted.Should().NotBeNull();
        persisted!.Email.Should().Be("updated@example.com");
        persisted.AccountStatus.Should().Be("Premium");
        persisted.AccessToken.Should().Be(authResult.AccessToken);
        persisted.RefreshToken.Should().Be(authResult.RefreshToken);
    }

    [Fact]
    public async Task ProcessBillingCallbackAsync_ShouldConfirmPurchasedAddOn()
    {
        var authResult = CreateAuthResult("first@example.com");
        var callbackUri = new Uri("wino://billing/success?productCode=UNLIMITED_ACCOUNTS");

        _apiClient
            .Setup(x => x.LoginAsync("first@example.com", "pw", default))
            .ReturnsAsync(WinoAccountApiResult<AuthResultDto>.Success(authResult));

        _apiClient
            .Setup(x => x.GetCurrentUserAsync(default))
            .ReturnsAsync(ApiEnvelope<AuthUserDto>.Success(authResult.User));

        _storeManagementService
            .Setup(x => x.HasProductAsync(WinoAddOnProductType.UNLIMITED_ACCOUNTS))
            .ReturnsAsync(true);

        await _service.LoginAsync("first@example.com", "pw");

        var processed = await _service.ProcessBillingCallbackAsync(callbackUri);

        processed.Should().BeTrue();
        _apiClient.Verify(x => x.GetCurrentUserAsync(default), Times.AtLeastOnce);
    }

    private static AuthResultDto CreateAuthResult(string email)
    {
        return new AuthResultDto(
            new AuthUserDto(Guid.NewGuid(), email, "Active", true, false, false),
            "access-token",
            DateTimeOffset.UtcNow.AddMinutes(30),
            "refresh-token",
            DateTimeOffset.UtcNow.AddDays(30));
    }
}