PnvPanel/backend/tests/PnvPanel.Application.Tests/Telegram/Bot/LinkTelegramCommandHandlerTests.cs

using NSubstitute;
using PnvPanel.Application.Common.Interfaces;
using PnvPanel.Application.Common.Models;
using PnvPanel.Application.Telegram;
using PnvPanel.Application.Telegram.Bot;
using PnvPanel.Application.Tests.TestSupport;
using PnvPanel.Domain.Telegram;
using Xunit;

namespace PnvPanel.Application.Tests.Telegram.Bot;

public class LinkTelegramCommandHandlerTests
{
    private readonly IIdentityService _identityService = Substitute.For<IIdentityService>();

    [Fact]
    public async Task Handle_WhenTokenValid_LinksUserAndConsumesToken()
    {
        using var dbContext = InMemoryDbContextFactory.Create();
        var userId = Guid.NewGuid();
        var token = TelegramLinkToken.Create(userId, TimeSpan.FromMinutes(5));
        dbContext.TelegramLinkTokens.Add(token);
        await dbContext.SaveChangesAsync(CancellationToken.None);

        _identityService
            .LinkTelegramAsync(userId, 123456L, "alice_tg", Arg.Any<CancellationToken>())
            .Returns(Result.Success());

        var handler = new LinkTelegramCommandHandler(dbContext, _identityService);

        var result = await handler.Handle(
            new LinkTelegramCommand(token.Token, 123456L, "alice_tg"),
            CancellationToken.None
        );

        Assert.True(result.IsSuccess);
        Assert.Equal(userId, result.Value);
        Assert.False(token.IsValid);
    }

    [Fact]
    public async Task Handle_WhenTokenNotFound_ReturnsNotFound()
    {
        using var dbContext = InMemoryDbContextFactory.Create();

        var handler = new LinkTelegramCommandHandler(dbContext, _identityService);

        var result = await handler.Handle(
            new LinkTelegramCommand("missing-token", 123456L, null),
            CancellationToken.None
        );

        Assert.False(result.IsSuccess);
        Assert.Equal(TelegramErrors.LinkTokenNotFound, result.Error);
    }

    [Fact]
    public async Task Handle_WhenTokenAlreadyConsumed_ReturnsNotFound()
    {
        using var dbContext = InMemoryDbContextFactory.Create();
        var userId = Guid.NewGuid();
        var token = TelegramLinkToken.Create(userId, TimeSpan.FromMinutes(5));
        token.Consume();
        dbContext.TelegramLinkTokens.Add(token);
        await dbContext.SaveChangesAsync(CancellationToken.None);

        var handler = new LinkTelegramCommandHandler(dbContext, _identityService);

        var result = await handler.Handle(
            new LinkTelegramCommand(token.Token, 123456L, null),
            CancellationToken.None
        );

        Assert.False(result.IsSuccess);
        Assert.Equal(TelegramErrors.LinkTokenNotFound, result.Error);
        await _identityService
            .DidNotReceive()
            .LinkTelegramAsync(
                Arg.Any<Guid>(),
                Arg.Any<long>(),
                Arg.Any<string?>(),
                Arg.Any<CancellationToken>()
            );
    }

    [Fact]
    public async Task Handle_WhenTokenExpired_ReturnsNotFound()
    {
        using var dbContext = InMemoryDbContextFactory.Create();
        var userId = Guid.NewGuid();
        var token = TelegramLinkToken.Create(userId, TimeSpan.FromMinutes(-5));
        dbContext.TelegramLinkTokens.Add(token);
        await dbContext.SaveChangesAsync(CancellationToken.None);

        var handler = new LinkTelegramCommandHandler(dbContext, _identityService);

        var result = await handler.Handle(
            new LinkTelegramCommand(token.Token, 123456L, null),
            CancellationToken.None
        );

        Assert.False(result.IsSuccess);
        Assert.Equal(TelegramErrors.LinkTokenNotFound, result.Error);
    }

    [Fact]
    public async Task Handle_WhenIdentityServiceFails_ReturnsFailureWithoutConsumingToken()
    {
        using var dbContext = InMemoryDbContextFactory.Create();
        var userId = Guid.NewGuid();
        var token = TelegramLinkToken.Create(userId, TimeSpan.FromMinutes(5));
        dbContext.TelegramLinkTokens.Add(token);
        await dbContext.SaveChangesAsync(CancellationToken.None);

        var error = Error.Conflict(
            "Telegram.AlreadyLinked",
            "Этот Telegram уже привязан к другому аккаунту."
        );
        _identityService
            .LinkTelegramAsync(userId, 123456L, null, Arg.Any<CancellationToken>())
            .Returns(Result.Failure(error));

        var handler = new LinkTelegramCommandHandler(dbContext, _identityService);

        var result = await handler.Handle(
            new LinkTelegramCommand(token.Token, 123456L, null),
            CancellationToken.None
        );

        Assert.False(result.IsSuccess);
        Assert.Equal(error, result.Error);
        Assert.True(token.IsValid);
    }
}