using Microsoft.Extensions.Logging; using NSubstitute; using PnvPanel.Application.Admin.Billing; using PnvPanel.Application.Billing; using PnvPanel.Application.Common.Interfaces; using PnvPanel.Application.Common.Models; using PnvPanel.Application.Tests.TestSupport; using PnvPanel.Domain.Billing; using PnvPanel.Domain.Configs; using PnvPanel.Domain.Inbounds; using PnvPanel.Domain.Nodes; using Xunit; namespace PnvPanel.Application.Tests.Admin.Billing; public class ConfirmPaymentRequestCommandHandlerTests { private readonly IIdentityService _identityService = Substitute.For(); private readonly IXuiPanelGateway _gateway = Substitute.For(); private readonly IRealtimeNotifier _notifier = Substitute.For(); private readonly ITelegramNotifier _telegramNotifier = Substitute.For(); private readonly ILogger _logger = Substitute.For< ILogger >(); private static CurrentUserProfile Profile(Guid userId, DateTimeOffset? paidUntil) => new( userId, "alice", Guid.NewGuid(), "premium", IsActivated: true, IsBlocked: false, MaxConfigs: 5, MaxIpLimit: 3, SubscriptionToken: "sub-token", BillingEnabled: true, BillingPaidUntil: paidUntil, BillingSuspended: paidUntil is null ); [Fact] public async Task Handle_WhenNoPriorPayment_ExtendsFromNow() { using var dbContext = InMemoryDbContextFactory.Create(); var adminId = Guid.NewGuid(); var userId = Guid.NewGuid(); var request = PaymentRequest.Create(userId, PaymentPeriod.Quarter, 1500); request.MarkPaymentSent(); dbContext.PaymentRequests.Add(request); await dbContext.SaveChangesAsync(CancellationToken.None); _identityService .GetProfileAsync(userId, Arg.Any()) .Returns(Profile(userId, paidUntil: null)); _identityService .ExtendBillingPaidUntilAsync(userId, Arg.Any(), Arg.Any()) .Returns(Result.Success()); var handler = new ConfirmPaymentRequestCommandHandler( dbContext, _identityService, _gateway, _notifier, _telegramNotifier, FakeCurrentUser.Authenticated(adminId, "admin"), _logger ); var before = DateTimeOffset.UtcNow; var result = await handler.Handle( new ConfirmPaymentRequestCommand(request.Id), CancellationToken.None ); Assert.True(result.IsSuccess); Assert.Equal(PaymentRequestStatus.Confirmed, request.Status); await _identityService .Received(1) .ExtendBillingPaidUntilAsync( userId, Arg.Is(d => d >= before.AddMonths(3).AddMinutes(-1)), Arg.Any() ); } [Fact] public async Task Handle_WhenPaidUntilInFuture_ExtendsFromExistingPaidUntil() { using var dbContext = InMemoryDbContextFactory.Create(); var adminId = Guid.NewGuid(); var userId = Guid.NewGuid(); var existingPaidUntil = DateTimeOffset.UtcNow.AddDays(20); var request = PaymentRequest.Create(userId, PaymentPeriod.Quarter, 1500); request.MarkPaymentSent(); dbContext.PaymentRequests.Add(request); await dbContext.SaveChangesAsync(CancellationToken.None); _identityService .GetProfileAsync(userId, Arg.Any()) .Returns(Profile(userId, paidUntil: existingPaidUntil)); _identityService .ExtendBillingPaidUntilAsync(userId, Arg.Any(), Arg.Any()) .Returns(Result.Success()); var handler = new ConfirmPaymentRequestCommandHandler( dbContext, _identityService, _gateway, _notifier, _telegramNotifier, FakeCurrentUser.Authenticated(adminId, "admin"), _logger ); await handler.Handle(new ConfirmPaymentRequestCommand(request.Id), CancellationToken.None); var expected = existingPaidUntil.AddMonths(3); await _identityService .Received(1) .ExtendBillingPaidUntilAsync( userId, Arg.Is(d => Math.Abs((d - expected).TotalSeconds) < 5), Arg.Any() ); } [Fact] public async Task Handle_ResumesExpiredConfigsAndSyncsExpiry() { using var dbContext = InMemoryDbContextFactory.Create(); var adminId = Guid.NewGuid(); var userId = Guid.NewGuid(); var node = Node.Register( "node-1", new Uri("https://node1.example.com"), new NodeCredentials("admin", "protected"), null ); var inbound = Inbound.FromRemote(node.Id, "1", VpnProtocol.Vless, "remark", 443); var expiredConfig = VpnConfig.Create(userId, inbound.Id, VpnProtocol.Vless, null); expiredConfig.AssignRemoteClient("ext-1"); expiredConfig.Suspend(); var activeConfig = VpnConfig.Create(userId, inbound.Id, VpnProtocol.Vless, null); activeConfig.AssignRemoteClient("ext-2"); dbContext.Nodes.Add(node); dbContext.Inbounds.Add(inbound); dbContext.VpnConfigs.AddRange(expiredConfig, activeConfig); var request = PaymentRequest.Create(userId, PaymentPeriod.Quarter, 1500); request.MarkPaymentSent(); dbContext.PaymentRequests.Add(request); await dbContext.SaveChangesAsync(CancellationToken.None); _identityService .GetProfileAsync(userId, Arg.Any()) .Returns(Profile(userId, paidUntil: null)); _identityService .ExtendBillingPaidUntilAsync(userId, Arg.Any(), Arg.Any()) .Returns(Result.Success()); _gateway .UpdateClientAsync( Arg.Any(), Arg.Any(), Arg.Any(), Arg.Any(), Arg.Any(), Arg.Any(), Arg.Any(), Arg.Any() ) .Returns(Result.Success()); var handler = new ConfirmPaymentRequestCommandHandler( dbContext, _identityService, _gateway, _notifier, _telegramNotifier, FakeCurrentUser.Authenticated(adminId, "admin"), _logger ); var result = await handler.Handle( new ConfirmPaymentRequestCommand(request.Id), CancellationToken.None ); Assert.True(result.IsSuccess); Assert.Equal(ConfigStatus.Active, expiredConfig.Status); Assert.NotNull(expiredConfig.ExpiresAt); Assert.NotNull(activeConfig.ExpiresAt); Assert.Equal(expiredConfig.ExpiresAt, activeConfig.ExpiresAt); // enable:true И expiresAt=новый newPaidUntil вместе (см. IXuiPanelGateway.UpdateClientAsync) — // и для ранее Expired конфига (ext-1), и для уже Active (ext-2): пользователь мог доплатить // заранее, панель должна узнать новый срок сразу, а не только когда конфиг реально просрочится. await _gateway .Received(1) .UpdateClientAsync( Arg.Any(), inbound.RemoteInboundId, "ext-1", VpnProtocol.Vless, Arg.Any(), true, Arg.Is(d => d == expiredConfig.ExpiresAt), Arg.Any() ); await _gateway .Received(1) .UpdateClientAsync( Arg.Any(), inbound.RemoteInboundId, "ext-2", VpnProtocol.Vless, Arg.Any(), true, Arg.Is(d => d == activeConfig.ExpiresAt), Arg.Any() ); await _notifier.Received(1).NotifyBillingStatusChangedAsync(userId, Arg.Any()); } [Fact] public async Task Handle_WhenRoleChangeTopUp_ConfirmsWithoutExtendingPaidUntil() { using var dbContext = InMemoryDbContextFactory.Create(); var adminId = Guid.NewGuid(); var userId = Guid.NewGuid(); var existingPaidUntil = DateTimeOffset.UtcNow.AddDays(20); var request = PaymentRequest.CreateRoleChangeTopUp(userId, 1200); request.MarkPaymentSent(); dbContext.PaymentRequests.Add(request); await dbContext.SaveChangesAsync(CancellationToken.None); _identityService .GetProfileAsync(userId, Arg.Any()) .Returns(Profile(userId, paidUntil: existingPaidUntil)); var handler = new ConfirmPaymentRequestCommandHandler( dbContext, _identityService, _gateway, _notifier, _telegramNotifier, FakeCurrentUser.Authenticated(adminId, "admin"), _logger ); var result = await handler.Handle( new ConfirmPaymentRequestCommand(request.Id), CancellationToken.None ); Assert.True(result.IsSuccess); Assert.Equal(PaymentRequestStatus.Confirmed, request.Status); await _identityService .DidNotReceive() .ExtendBillingPaidUntilAsync( Arg.Any(), Arg.Any(), Arg.Any() ); } [Fact] public async Task Handle_WhenRequestAlreadyDecided_ReturnsNotDecidable() { using var dbContext = InMemoryDbContextFactory.Create(); var adminId = Guid.NewGuid(); var userId = Guid.NewGuid(); var request = PaymentRequest.Create(userId, PaymentPeriod.Quarter, 1500); request.Cancel(); dbContext.PaymentRequests.Add(request); await dbContext.SaveChangesAsync(CancellationToken.None); var handler = new ConfirmPaymentRequestCommandHandler( dbContext, _identityService, _gateway, _notifier, _telegramNotifier, FakeCurrentUser.Authenticated(adminId, "admin"), _logger ); var result = await handler.Handle( new ConfirmPaymentRequestCommand(request.Id), CancellationToken.None ); Assert.False(result.IsSuccess); Assert.Equal(BillingErrors.RequestNotDecidable, result.Error); } }