mirror of
				https://github.com/nsnail/Ocelot.git
				synced 2025-11-04 10:35:28 +08:00 
			
		
		
		
	passing test for issue (#197)
* passing test for issue * fix test when run together
This commit is contained in:
		
							
								
								
									
										230
									
								
								test/Ocelot.AcceptanceTests/TwoDownstreamServicesTests.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										230
									
								
								test/Ocelot.AcceptanceTests/TwoDownstreamServicesTests.cs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,230 @@
 | 
			
		||||
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<ServiceEntry> _serviceEntries;
 | 
			
		||||
        private string _downstreamPathOne;
 | 
			
		||||
        private string _downstreamPathTwo;
 | 
			
		||||
 | 
			
		||||
        public TwoDownstreamServicesTests()
 | 
			
		||||
        {
 | 
			
		||||
            _steps = new Steps();
 | 
			
		||||
            _serviceEntries = new List<ServiceEntry>();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        [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<FileReRoute>
 | 
			
		||||
                    {
 | 
			
		||||
                        new FileReRoute
 | 
			
		||||
                        {
 | 
			
		||||
                            DownstreamPathTemplate = "/api/user/{user}",
 | 
			
		||||
                            DownstreamScheme = "http",
 | 
			
		||||
                            DownstreamHost = "localhost",
 | 
			
		||||
                            DownstreamPort = 8362,
 | 
			
		||||
                            UpstreamPathTemplate = "/api/user/{user}",
 | 
			
		||||
                            UpstreamHttpMethod = new List<string> { "Get" },
 | 
			
		||||
                        },
 | 
			
		||||
                        new FileReRoute
 | 
			
		||||
                        {
 | 
			
		||||
                            DownstreamPathTemplate = "/api/product/{product}",
 | 
			
		||||
                            DownstreamScheme = "http",
 | 
			
		||||
                            DownstreamHost = "localhost",
 | 
			
		||||
                            DownstreamPort = 8330,
 | 
			
		||||
                            UpstreamPathTemplate = "/api/product/{product}",
 | 
			
		||||
                            UpstreamHttpMethod = new List<string> { "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();
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
		Reference in New Issue
	
	Block a user