Leonid Pershin
b2ae358250
- Introduced billing capabilities, allowing users to request payments for subscription periods (3/6/12 months) with admin approval via Telegram. - Updated role management to include a `BillingEnabled` property, preventing billing for admin roles. - Enhanced the `CreateRoleCommand` and `UpdateRoleCommand` to accept billing parameters, ensuring proper handling during role creation and updates. - Added new endpoints for billing management and integrated billing checks into VPN config creation to enforce payment requirements. - Updated related services, models, and tests to support the new billing features, ensuring comprehensive coverage and functionality. - Enhanced documentation to reflect the new billing processes and role management changes.
83 lines
2.6 KiB
C#
83 lines
2.6 KiB
C#
using NSubstitute;
|
|
using PnvPanel.Application.Auth;
|
|
using PnvPanel.Application.Auth.Me;
|
|
using PnvPanel.Application.Common.Interfaces;
|
|
using PnvPanel.Application.Common.Models;
|
|
using PnvPanel.Application.Tests.TestSupport;
|
|
using Xunit;
|
|
|
|
namespace PnvPanel.Application.Tests.Auth;
|
|
|
|
public class GetCurrentUserQueryHandlerTests
|
|
{
|
|
private readonly IIdentityService _identityService = Substitute.For<IIdentityService>();
|
|
|
|
[Fact]
|
|
public async Task Handle_Unauthenticated_ReturnsUnauthorized()
|
|
{
|
|
var handler = new GetCurrentUserQueryHandler(_identityService, FakeCurrentUser.Anonymous());
|
|
|
|
var result = await handler.Handle(new GetCurrentUserQuery(), CancellationToken.None);
|
|
|
|
Assert.False(result.IsSuccess);
|
|
Assert.Equal(AuthErrors.Unauthorized, result.Error);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Handle_ProfileMissing_ReturnsUnauthorized()
|
|
{
|
|
var userId = Guid.NewGuid();
|
|
_identityService
|
|
.GetProfileAsync(userId, Arg.Any<CancellationToken>())
|
|
.Returns((CurrentUserProfile?)null);
|
|
|
|
var handler = new GetCurrentUserQueryHandler(
|
|
_identityService,
|
|
FakeCurrentUser.Authenticated(userId)
|
|
);
|
|
|
|
var result = await handler.Handle(new GetCurrentUserQuery(), CancellationToken.None);
|
|
|
|
Assert.False(result.IsSuccess);
|
|
Assert.Equal(AuthErrors.Unauthorized, result.Error);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Handle_AuthenticatedWithProfile_ReturnsCurrentUserDto()
|
|
{
|
|
var userId = Guid.NewGuid();
|
|
var profile = new CurrentUserProfile(
|
|
userId,
|
|
"alice",
|
|
Guid.NewGuid(),
|
|
"user",
|
|
true,
|
|
false,
|
|
3,
|
|
RoleQuota.Unlimited,
|
|
"sub-token",
|
|
false,
|
|
null,
|
|
false
|
|
);
|
|
_identityService.GetProfileAsync(userId, Arg.Any<CancellationToken>()).Returns(profile);
|
|
_identityService
|
|
.GetTelegramLinkInfoAsync(userId, Arg.Any<CancellationToken>())
|
|
.Returns(new TelegramLinkInfo(true, 42, "alice_tg"));
|
|
|
|
var handler = new GetCurrentUserQueryHandler(
|
|
_identityService,
|
|
FakeCurrentUser.Authenticated(userId, "alice")
|
|
);
|
|
|
|
var result = await handler.Handle(new GetCurrentUserQuery(), CancellationToken.None);
|
|
|
|
Assert.True(result.IsSuccess);
|
|
Assert.Equal(userId, result.Value.Id);
|
|
Assert.Equal("alice", result.Value.UserName);
|
|
Assert.Equal("user", result.Value.Role);
|
|
Assert.True(result.Value.IsActivated);
|
|
Assert.True(result.Value.TelegramLinked);
|
|
}
|
|
}
|