using System.Collections.Generic; 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(() => 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(() => 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( Substitute.For(httpContext, new RouteData(), Substitute.For()), new List(), new Dictionary(), Substitute.For()); return context; } } }