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