mirror of
https://github.com/bitwarden/server.git
synced 2025-04-05 05:00:19 -05:00
Add tests
This commit is contained in:
parent
aa31e985c4
commit
883a66ef5e
@ -11,8 +11,6 @@ namespace Bit.Api.AdminConsole.Authorization;
|
|||||||
/// Retrieves the Organization ID from the route and then passes it to the requirement's AuthorizeAsync callback to
|
/// Retrieves the Organization ID from the route and then passes it to the requirement's AuthorizeAsync callback to
|
||||||
/// determine whether the action is authorized.
|
/// determine whether the action is authorized.
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="currentContext"></param>
|
|
||||||
/// <param name="httpContextAccessor"></param>
|
|
||||||
public class OrganizationRequirementHandler(
|
public class OrganizationRequirementHandler(
|
||||||
ICurrentContext currentContext,
|
ICurrentContext currentContext,
|
||||||
IProviderOrganizationContext providerOrganizationContext,
|
IProviderOrganizationContext providerOrganizationContext,
|
||||||
|
@ -12,7 +12,7 @@ public static class OrganizationRequirementHelpers
|
|||||||
}
|
}
|
||||||
|
|
||||||
httpContextAccessor.HttpContext.GetRouteData().Values.TryGetValue("orgId", out var orgIdParam);
|
httpContextAccessor.HttpContext.GetRouteData().Values.TryGetValue("orgId", out var orgIdParam);
|
||||||
if (!Guid.TryParse(orgIdParam?.ToString(), out var orgId))
|
if (orgIdParam == null || !Guid.TryParse(orgIdParam.ToString(), out var orgId))
|
||||||
{
|
{
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,92 @@
|
|||||||
|
using System.Security.Claims;
|
||||||
|
using Bit.Api.AdminConsole.Authorization;
|
||||||
|
using Bit.Api.AdminConsole.Context;
|
||||||
|
using Bit.Test.Common.AutoFixture;
|
||||||
|
using Bit.Test.Common.AutoFixture.Attributes;
|
||||||
|
using Microsoft.AspNetCore.Authorization;
|
||||||
|
using Microsoft.AspNetCore.Http;
|
||||||
|
using NSubstitute;
|
||||||
|
using Xunit;
|
||||||
|
|
||||||
|
namespace Bit.Api.Test.AdminConsole.Authorization;
|
||||||
|
|
||||||
|
[SutProviderCustomize]
|
||||||
|
public class OrganizationRequirementHandlerTests
|
||||||
|
{
|
||||||
|
[Theory, BitAutoData]
|
||||||
|
public async Task IfNoOrganizationId_Throws(SutProvider<OrganizationRequirementHandler> sutProvider)
|
||||||
|
{
|
||||||
|
// Arrange
|
||||||
|
ArrangeRouteValues(sutProvider, null); // no orgId in route
|
||||||
|
var testRequirement = Substitute.For<IOrganizationRequirement>();
|
||||||
|
var authContext = new AuthorizationHandlerContext([testRequirement], new ClaimsPrincipal(), null);
|
||||||
|
|
||||||
|
// Act
|
||||||
|
var exception = await Assert.ThrowsAsync<Exception>(() => sutProvider.Sut.HandleAsync(authContext));
|
||||||
|
Assert.Contains("No organizationId found", exception.Message);
|
||||||
|
Assert.False(authContext.HasSucceeded);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Theory, BitAutoData]
|
||||||
|
public async Task IfInvalidOrganizationId_Throws(SutProvider<OrganizationRequirementHandler> sutProvider)
|
||||||
|
{
|
||||||
|
// Arrange
|
||||||
|
ArrangeRouteValues(sutProvider, "malformed guid");
|
||||||
|
var testRequirement = Substitute.For<IOrganizationRequirement>();
|
||||||
|
var authContext = new AuthorizationHandlerContext([testRequirement], new ClaimsPrincipal(), null);
|
||||||
|
|
||||||
|
// Act
|
||||||
|
var exception = await Assert.ThrowsAsync<Exception>(() => sutProvider.Sut.HandleAsync(authContext));
|
||||||
|
Assert.Contains("No organizationId found", exception.Message);
|
||||||
|
Assert.False(authContext.HasSucceeded);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Theory, BitAutoData]
|
||||||
|
public async Task DoesNotAuthorize_IfAuthorizeAsync_ReturnsFalse(SutProvider<OrganizationRequirementHandler> sutProvider, Guid organizationId)
|
||||||
|
{
|
||||||
|
// Arrange route values
|
||||||
|
ArrangeRouteValues(sutProvider, organizationId.ToString());
|
||||||
|
|
||||||
|
// Arrange requirement
|
||||||
|
var testRequirement = Substitute.For<IOrganizationRequirement>();
|
||||||
|
testRequirement
|
||||||
|
.AuthorizeAsync(organizationId, null, Arg.Any<IProviderOrganizationContext>())
|
||||||
|
.ReturnsForAnyArgs(false);
|
||||||
|
var authContext = new AuthorizationHandlerContext([testRequirement], new ClaimsPrincipal(), null);
|
||||||
|
|
||||||
|
// Act
|
||||||
|
await sutProvider.Sut.HandleAsync(authContext);
|
||||||
|
|
||||||
|
// Assert
|
||||||
|
await testRequirement.Received(1).AuthorizeAsync(organizationId, null, Arg.Any<IProviderOrganizationContext>());
|
||||||
|
Assert.False(authContext.HasSucceeded);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Theory, BitAutoData]
|
||||||
|
public async Task Authorizes_IfAuthorizeAsync_ReturnsTrue(SutProvider<OrganizationRequirementHandler> sutProvider, Guid organizationId)
|
||||||
|
{
|
||||||
|
// Arrange route values
|
||||||
|
ArrangeRouteValues(sutProvider, organizationId.ToString());
|
||||||
|
|
||||||
|
// Arrange requirement
|
||||||
|
var testRequirement = Substitute.For<IOrganizationRequirement>();
|
||||||
|
testRequirement
|
||||||
|
.AuthorizeAsync(organizationId, null, Arg.Any<IProviderOrganizationContext>())
|
||||||
|
.ReturnsForAnyArgs(true);
|
||||||
|
var authContext = new AuthorizationHandlerContext([testRequirement], new ClaimsPrincipal(), null);
|
||||||
|
|
||||||
|
// Act
|
||||||
|
await sutProvider.Sut.HandleAsync(authContext);
|
||||||
|
|
||||||
|
// Assert
|
||||||
|
await testRequirement.Received(1).AuthorizeAsync(organizationId, null, Arg.Any<IProviderOrganizationContext>());
|
||||||
|
Assert.True(authContext.HasSucceeded);
|
||||||
|
}
|
||||||
|
|
||||||
|
private static void ArrangeRouteValues(SutProvider<OrganizationRequirementHandler> sutProvider, string orgIdRouteValue)
|
||||||
|
{
|
||||||
|
var httpContext = new DefaultHttpContext();
|
||||||
|
httpContext.Request.RouteValues["orgId"] = orgIdRouteValue;
|
||||||
|
sutProvider.GetDependency<IHttpContextAccessor>().HttpContext = httpContext;
|
||||||
|
}
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user