using System; using System.Collections.Generic; using System.IO; using System.Net; using Consul; using Microsoft.AspNetCore.Builder; using Microsoft.AspNetCore.Hosting; using Microsoft.AspNetCore.Http; using Ocelot.Configuration.File; using Shouldly; using TestStack.BDDfy; using Xunit; namespace Ocelot.AcceptanceTests { public class TwoDownstreamServicesTests : IDisposable { private IWebHost _builderOne; private IWebHost _builderTwo; private IWebHost _fakeConsulBuilder; private readonly Steps _steps; private readonly List _serviceEntries; private string _downstreamPathOne; private string _downstreamPathTwo; public TwoDownstreamServicesTests() { _steps = new Steps(); _serviceEntries = new List(); } [Fact] public void should_fix_issue_194() { var consulPort = 8503; var downstreamServiceOneUrl = "http://localhost:8362"; var downstreamServiceTwoUrl = "http://localhost:8330"; var fakeConsulServiceDiscoveryUrl = $"http://localhost:{consulPort}"; // http://localhost:8362/api/user/info?id=1 is ok // http://localhost:3164/api/user/info?id=1 is ok // http://localhost:8330/api/product/info?id=1 is ok // http://localhost:3164/api/product/info?id=1 is 404 // is my configuration.json // { // "ReRoutes": [ // //{ // // "DownstreamPathTemplate": "/{product}", // // "DownstreamScheme": "http", // // "UpstreamPathTemplate": "/{product}", // // "UpstreamHttpMethod": [ "Get", "Post" ], // // "ServiceName": "api-product", // // "LoadBalancer": "LeastConnection", // // "UseServiceDiscovery": true // //}, // //{ // // "DownstreamPathTemplate": "/{user}", // // "DownstreamScheme": "http", // // "UpstreamPathTemplate": "/{user}", // // "UpstreamHttpMethod": [ "Get", "Post" ], // // "ServiceName": "api-user", // // "LoadBalancer": "LeastConnection", // // "UseServiceDiscovery": true // //}, // { // "DownstreamPathTemplate": "/api/user/{user}", // "DownstreamScheme": "http", // "DownstreamHost": "localhost", // "DownstreamPort": 8362, // "UpstreamPathTemplate": "/api/user/{user}", // "UpstreamHttpMethod": [ "Get" ] // }, // { // "DownstreamPathTemplate": "/api/product/{product}", // "DownstreamScheme": "http", // "DownstreamHost": "localhost", // "DownstreamPort": 8330, // "UpstreamPathTemplate": "//api/product/{product}", // "UpstreamHttpMethod": [ "Get" ] // } // ], // "GlobalConfiguration": { // "ServiceDiscoveryProvider": { // "Host": "localhost", // "Port": 8500 // } // } // } var configuration = new FileConfiguration { ReRoutes = new List { new FileReRoute { DownstreamPathTemplate = "/api/user/{user}", DownstreamScheme = "http", DownstreamHost = "localhost", DownstreamPort = 8362, UpstreamPathTemplate = "/api/user/{user}", UpstreamHttpMethod = new List { "Get" }, }, new FileReRoute { DownstreamPathTemplate = "/api/product/{product}", DownstreamScheme = "http", DownstreamHost = "localhost", DownstreamPort = 8330, UpstreamPathTemplate = "/api/product/{product}", UpstreamHttpMethod = new List { "Get" }, } }, GlobalConfiguration = new FileGlobalConfiguration() { ServiceDiscoveryProvider = new FileServiceDiscoveryProvider() { Host = "localhost", Port = consulPort } } }; this.Given(x => x.GivenProductServiceOneIsRunning(downstreamServiceOneUrl, "/api/user/info", 200, "user")) .And(x => x.GivenProductServiceTwoIsRunning(downstreamServiceTwoUrl, "/api/product/info", 200, "product")) .And(x => x.GivenThereIsAFakeConsulServiceDiscoveryProvider(fakeConsulServiceDiscoveryUrl)) .And(x => _steps.GivenThereIsAConfiguration(configuration)) .And(x => _steps.GivenOcelotIsRunning()) .When(x => _steps.WhenIGetUrlOnTheApiGateway("/api/user/info?id=1")) .Then(x => _steps.ThenTheStatusCodeShouldBe(HttpStatusCode.OK)) .And(x => _steps.ThenTheResponseBodyShouldBe("user")) .When(x => _steps.WhenIGetUrlOnTheApiGateway("/api/product/info?id=1")) .Then(x => _steps.ThenTheStatusCodeShouldBe(HttpStatusCode.OK)) .And(x => _steps.ThenTheResponseBodyShouldBe("product")) .BDDfy(); } private void GivenThereIsAFakeConsulServiceDiscoveryProvider(string url) { _fakeConsulBuilder = new WebHostBuilder() .UseUrls(url) .UseKestrel() .UseContentRoot(Directory.GetCurrentDirectory()) .UseIISIntegration() .UseUrls(url) .Configure(app => { app.Run(async context => { if(context.Request.Path.Value == "/v1/health/service/product") { await context.Response.WriteJsonAsync(_serviceEntries); } }); }) .Build(); _fakeConsulBuilder.Start(); } private void GivenProductServiceOneIsRunning(string baseUrl, string basePath, int statusCode, string responseBody) { _builderOne = new WebHostBuilder() .UseUrls(baseUrl) .UseKestrel() .UseContentRoot(Directory.GetCurrentDirectory()) .UseIISIntegration() .Configure(app => { app.UsePathBase(basePath); app.Run(async context => { _downstreamPathOne = !string.IsNullOrEmpty(context.Request.PathBase.Value) ? context.Request.PathBase.Value : context.Request.Path.Value; if(_downstreamPathOne != basePath) { context.Response.StatusCode = statusCode; await context.Response.WriteAsync("downstream path didnt match base path"); } else { context.Response.StatusCode = statusCode; await context.Response.WriteAsync(responseBody); } }); }) .Build(); _builderOne.Start(); } private void GivenProductServiceTwoIsRunning(string baseUrl, string basePath, int statusCode, string responseBody) { _builderTwo = new WebHostBuilder() .UseUrls(baseUrl) .UseKestrel() .UseContentRoot(Directory.GetCurrentDirectory()) .UseIISIntegration() .Configure(app => { app.UsePathBase(basePath); app.Run(async context => { _downstreamPathTwo = !string.IsNullOrEmpty(context.Request.PathBase.Value) ? context.Request.PathBase.Value : context.Request.Path.Value; if(_downstreamPathTwo != basePath) { context.Response.StatusCode = statusCode; await context.Response.WriteAsync("downstream path didnt match base path"); } else { context.Response.StatusCode = statusCode; await context.Response.WriteAsync(responseBody); } }); }) .Build(); _builderTwo.Start(); } public void Dispose() { _builderOne?.Dispose(); _builderTwo?.Dispose(); _steps.Dispose(); } } }