Add Telegram bot integration and enhance user management features

- Introduced Telegram.Bot package for bot functionality. - Updated user management to include Telegram linking and blocking features. - Enhanced activation request handling with notifications via Telegram. - Added new database entities for Telegram link tokens and login requests. - Implemented traffic synchronization for client stats in the XuiPanelGateway. - Updated application structure to support new test projects and improved dependency injection for Telegram services.
2026-07-02 01:01:03 +03:00
parent 1a8d33efa3
commit 7b6fe9ad78
142 changed files with 7570 additions and 22 deletions
@@ -0,0 +1,116 @@
+using NSubstitute;
+using PnvPanel.Application.Auth;
+using PnvPanel.Application.Common.Interfaces;
+using PnvPanel.Application.Tests.TestSupport;
+using PnvPanel.Domain.Telegram;
+using Xunit;
+using TelegramNs = PnvPanel.Application.Telegram;
+
+namespace PnvPanel.Application.Tests.Telegram;
+
+public class GetLoginRequestStatusQueryHandlerTests
+{
+    private readonly IIdentityService _identityService = Substitute.For<IIdentityService>();
+    private readonly IJwtTokenService _jwtTokenService = Substitute.For<IJwtTokenService>();
+    private readonly IRefreshTokenService _refreshTokenService = Substitute.For<IRefreshTokenService>();
+
+    private TelegramNs.GetLoginRequestStatusQueryHandler CreateHandler(PnvPanel.Infrastructure.Persistence.AppDbContext dbContext)
+        => new(dbContext, _identityService, _jwtTokenService, _refreshTokenService);
+
+    [Fact]
+    public async Task Handle_WhenRequestNotFound_ReturnsNotFound()
+    {
+        using var dbContext = InMemoryDbContextFactory.Create();
+
+        var handler = CreateHandler(dbContext);
+
+        var result = await handler.Handle(new TelegramNs.GetLoginRequestStatusQuery(Guid.NewGuid()), CancellationToken.None);
+
+        Assert.False(result.IsSuccess);
+        Assert.Equal(TelegramNs.TelegramErrors.LoginRequestNotFound, result.Error);
+    }
+
+    [Fact]
+    public async Task Handle_WhenPendingAndExpired_ReturnsExpiredStatusWithoutIssuingTokens()
+    {
+        using var dbContext = InMemoryDbContextFactory.Create();
+        var request = TelegramLoginRequest.Create(TimeSpan.FromMinutes(-5), null);
+        dbContext.TelegramLoginRequests.Add(request);
+        await dbContext.SaveChangesAsync(CancellationToken.None);
+
+        var handler = CreateHandler(dbContext);
+
+        var result = await handler.Handle(new TelegramNs.GetLoginRequestStatusQuery(request.Id), CancellationToken.None);
+
+        Assert.True(result.IsSuccess);
+        Assert.Equal(TelegramLoginStatus.Expired, result.Value.Status);
+        Assert.Null(result.Value.Auth);
+        await _refreshTokenService.DidNotReceive().IssueAsync(Arg.Any<Guid>(), Arg.Any<CancellationToken>());
+    }
+
+    [Fact]
+    public async Task Handle_WhenStillPending_ReturnsPendingStatusWithoutAuth()
+    {
+        using var dbContext = InMemoryDbContextFactory.Create();
+        var request = TelegramLoginRequest.Create(TimeSpan.FromMinutes(5), null);
+        dbContext.TelegramLoginRequests.Add(request);
+        await dbContext.SaveChangesAsync(CancellationToken.None);
+
+        var handler = CreateHandler(dbContext);
+
+        var result = await handler.Handle(new TelegramNs.GetLoginRequestStatusQuery(request.Id), CancellationToken.None);
+
+        Assert.True(result.IsSuccess);
+        Assert.Equal(TelegramLoginStatus.Pending, result.Value.Status);
+        Assert.Null(result.Value.Auth);
+    }
+
+    [Fact]
+    public async Task Handle_WhenApproved_ConsumesRequestAndReturnsAuthResult()
+    {
+        using var dbContext = InMemoryDbContextFactory.Create();
+        var userId = Guid.NewGuid();
+        var request = TelegramLoginRequest.Create(TimeSpan.FromMinutes(5), null);
+        request.Approve(userId);
+        dbContext.TelegramLoginRequests.Add(request);
+        await dbContext.SaveChangesAsync(CancellationToken.None);
+
+        var profile = new CurrentUserProfile(userId, "alice", Guid.NewGuid(), "user", true, false, 3);
+        _identityService.GetProfileAsync(userId, Arg.Any<CancellationToken>()).Returns(profile);
+        _jwtTokenService.GenerateAccessToken(Arg.Any<AuthenticatedUser>())
+            .Returns(("access-token", DateTimeOffset.UtcNow.AddMinutes(15)));
+        _refreshTokenService.IssueAsync(userId, Arg.Any<CancellationToken>())
+            .Returns(new IssuedRefreshToken("refresh-token", DateTimeOffset.UtcNow.AddDays(30)));
+
+        var handler = CreateHandler(dbContext);
+
+        var result = await handler.Handle(new TelegramNs.GetLoginRequestStatusQuery(request.Id), CancellationToken.None);
+
+        Assert.True(result.IsSuccess);
+        Assert.Equal(TelegramLoginStatus.Approved, result.Value.Status);
+        Assert.NotNull(result.Value.Auth);
+        Assert.Equal("access-token", result.Value.Auth!.AccessToken);
+        Assert.True(result.Value.Auth.User.TelegramLinked);
+        Assert.Equal(TelegramLoginStatus.Consumed, request.Status);
+    }
+
+    [Fact]
+    public async Task Handle_WhenApprovedButProfileMissing_ReturnsUnauthorized()
+    {
+        using var dbContext = InMemoryDbContextFactory.Create();
+        var userId = Guid.NewGuid();
+        var request = TelegramLoginRequest.Create(TimeSpan.FromMinutes(5), null);
+        request.Approve(userId);
+        dbContext.TelegramLoginRequests.Add(request);
+        await dbContext.SaveChangesAsync(CancellationToken.None);
+
+        _identityService.GetProfileAsync(userId, Arg.Any<CancellationToken>()).Returns((CurrentUserProfile?)null);
+
+        var handler = CreateHandler(dbContext);
+
+        var result = await handler.Handle(new TelegramNs.GetLoginRequestStatusQuery(request.Id), CancellationToken.None);
+
+        Assert.False(result.IsSuccess);
+        Assert.Equal(AuthErrors.Unauthorized, result.Error);
+    }
+}