bitwarden-server/test/Core.Test/Utilities/SelfHostedAttributeTests.cs

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;
    }
}