mirror of
https://github.com/nsnail/Ocelot.git
synced 2025-05-03 05:12:51 +08:00
108 lines
3.4 KiB
C#
108 lines
3.4 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Net.Http;
|
|
using Microsoft.AspNetCore.Hosting;
|
|
using Microsoft.AspNetCore.TestHost;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Moq;
|
|
using Ocelot.Authentication.Handler.Factory;
|
|
using Ocelot.Authentication.Middleware;
|
|
using Ocelot.Cache.Middleware;
|
|
using Ocelot.Configuration.Builder;
|
|
using Ocelot.DownstreamRouteFinder;
|
|
using Ocelot.DownstreamRouteFinder.UrlMatcher;
|
|
using Ocelot.Infrastructure.RequestData;
|
|
using Ocelot.Logging;
|
|
using Ocelot.Responses;
|
|
using TestStack.BDDfy;
|
|
using Xunit;
|
|
|
|
namespace Ocelot.UnitTests.Authentication
|
|
{
|
|
public class AuthenticationMiddlewareTests : IDisposable
|
|
{
|
|
private readonly Mock<IRequestScopedDataRepository> _scopedRepository;
|
|
private readonly Mock<IAuthenticationHandlerFactory> _authFactory;
|
|
private readonly string _url;
|
|
private readonly TestServer _server;
|
|
private readonly HttpClient _client;
|
|
private HttpResponseMessage _result;
|
|
private OkResponse<DownstreamRoute> _downstreamRoute;
|
|
private Mock<IOcelotLoggerFactory> _mockLoggerFactory;
|
|
|
|
public AuthenticationMiddlewareTests()
|
|
{
|
|
_url = "http://localhost:51879";
|
|
_scopedRepository = new Mock<IRequestScopedDataRepository>();
|
|
_authFactory = new Mock<IAuthenticationHandlerFactory>();
|
|
SetUpLogger();
|
|
|
|
var builder = new WebHostBuilder()
|
|
.ConfigureServices(x =>
|
|
{
|
|
x.AddSingleton(_mockLoggerFactory.Object);
|
|
x.AddSingleton(_authFactory.Object);
|
|
x.AddSingleton(_scopedRepository.Object);
|
|
})
|
|
.UseUrls(_url)
|
|
.UseKestrel()
|
|
.UseContentRoot(Directory.GetCurrentDirectory())
|
|
.UseIISIntegration()
|
|
.UseUrls(_url)
|
|
.Configure(app =>
|
|
{
|
|
app.UseAuthenticationMiddleware();
|
|
});
|
|
|
|
_server = new TestServer(builder);
|
|
_client = _server.CreateClient();
|
|
}
|
|
|
|
[Fact]
|
|
public void happy_path()
|
|
{
|
|
this.Given(x => x.GivenTheDownStreamRouteIs(new DownstreamRoute(new List<UrlPathPlaceholderNameAndValue>(), new ReRouteBuilder().Build())))
|
|
.When(x => x.WhenICallTheMiddleware())
|
|
.Then(x => x.ThenNoExceptionsAreThrown())
|
|
.BDDfy();
|
|
}
|
|
|
|
private void SetUpLogger()
|
|
{
|
|
_mockLoggerFactory = new Mock<IOcelotLoggerFactory>();
|
|
|
|
var logger = new Mock<IOcelotLogger>();
|
|
|
|
_mockLoggerFactory
|
|
.Setup(x => x.CreateLogger<AuthenticationMiddleware>())
|
|
.Returns(logger.Object);
|
|
}
|
|
|
|
private void ThenNoExceptionsAreThrown()
|
|
{
|
|
//todo not suck
|
|
}
|
|
|
|
private void GivenTheDownStreamRouteIs(DownstreamRoute downstreamRoute)
|
|
{
|
|
_downstreamRoute = new OkResponse<DownstreamRoute>(downstreamRoute);
|
|
_scopedRepository
|
|
.Setup(x => x.Get<DownstreamRoute>(It.IsAny<string>()))
|
|
.Returns(_downstreamRoute);
|
|
}
|
|
|
|
private void WhenICallTheMiddleware()
|
|
{
|
|
_result = _client.GetAsync(_url).Result;
|
|
}
|
|
|
|
|
|
public void Dispose()
|
|
{
|
|
_client.Dispose();
|
|
_server.Dispose();
|
|
}
|
|
}
|
|
}
|