mirror of
https://github.com/bitwarden/server.git
synced 2024-12-05 14:23:30 +01:00
92 lines
2.5 KiB
C#
92 lines
2.5 KiB
C#
using Bit.Core.Exceptions;
|
|
using Bit.Core.Settings;
|
|
using Bit.Core.Utilities;
|
|
using Microsoft.AspNetCore.Http;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.AspNetCore.Mvc.Abstractions;
|
|
using Microsoft.AspNetCore.Mvc.Filters;
|
|
using Microsoft.AspNetCore.Routing;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using NSubstitute;
|
|
using Xunit;
|
|
|
|
namespace Bit.Core.Test.Utilities;
|
|
|
|
public class SelfHostedAttributeTests
|
|
{
|
|
[Fact]
|
|
public void NotSelfHosted_Throws_When_SelfHosted()
|
|
{
|
|
// Arrange
|
|
var sha = new SelfHostedAttribute { NotSelfHostedOnly = true };
|
|
|
|
// Act & Assert
|
|
Assert.Throws<BadRequestException>(() => sha.OnActionExecuting(GetContext(selfHosted: true)));
|
|
}
|
|
|
|
[Fact]
|
|
public void NotSelfHosted_Success_When_NotSelfHosted()
|
|
{
|
|
// Arrange
|
|
var sha = new SelfHostedAttribute { NotSelfHostedOnly = true };
|
|
|
|
// Act
|
|
sha.OnActionExecuting(GetContext(selfHosted: false));
|
|
|
|
// Assert
|
|
// The Assert here is just NOT throwing an exception
|
|
}
|
|
|
|
|
|
[Fact]
|
|
public void SelfHosted_Success_When_SelfHosted()
|
|
{
|
|
// Arrange
|
|
var sha = new SelfHostedAttribute { SelfHostedOnly = true };
|
|
|
|
// Act
|
|
sha.OnActionExecuting(GetContext(selfHosted: true));
|
|
|
|
// Assert
|
|
// The Assert here is just NOT throwing an exception
|
|
}
|
|
|
|
[Fact]
|
|
public void SelfHosted_Throws_When_NotSelfHosted()
|
|
{
|
|
// Arrange
|
|
var sha = new SelfHostedAttribute { SelfHostedOnly = true };
|
|
|
|
// Act & Assert
|
|
Assert.Throws<BadRequestException>(() => sha.OnActionExecuting(GetContext(selfHosted: false)));
|
|
}
|
|
|
|
|
|
// This generates a ActionExecutingContext with the needed injected
|
|
// service with the given value.
|
|
private ActionExecutingContext GetContext(bool selfHosted)
|
|
{
|
|
IServiceCollection services = new ServiceCollection();
|
|
|
|
var globalSettings = new GlobalSettings
|
|
{
|
|
SelfHosted = selfHosted
|
|
};
|
|
|
|
services.AddSingleton(globalSettings);
|
|
|
|
var httpContext = new DefaultHttpContext();
|
|
httpContext.RequestServices = services.BuildServiceProvider();
|
|
|
|
var context = Substitute.For<ActionExecutingContext>(
|
|
Substitute.For<ActionContext>(httpContext,
|
|
new RouteData(),
|
|
Substitute.For<ActionDescriptor>()),
|
|
new List<IFilterMetadata>(),
|
|
new Dictionary<string, object>(),
|
|
Substitute.For<Controller>());
|
|
|
|
return context;
|
|
}
|
|
}
|