tk/Ocelot
1
0
Fork 0
mirror of https://github.com/nsnail/Ocelot.git synced 2025-06-19 12:38:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

added houses

Browse Source
This commit is contained in:
Tom Gardham-Pallister
2017-02-02 08:00:37 +00:00
parent 074ae4d609
commit 37aaeeed82
8 changed files with 229 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
test/Ocelot.UnitTests/LoadBalancer/LoadBalancerHouseTests.cs
View File

@ -38,7 +38,7 @@ namespace Ocelot.UnitTests.LoadBalancer
var key = "test";
this.Given(x => x.GivenThereIsALoadBalancer(key, new FakeLoadBalancer()))
.When(x => x.WhenWeGetThatLoadBalancer(key))
.When(x => x.WhenWeGetTheLoadBalancer(key))
.Then(x => x.ThenItIsReturned())
.BDDfy();
}
@ -51,13 +51,27 @@ namespace Ocelot.UnitTests.LoadBalancer
this.Given(x => x.GivenThereIsALoadBalancer(key, new FakeLoadBalancer()))
.And(x => x.GivenThereIsALoadBalancer(keyTwo, new FakeRoundRobinLoadBalancer()))
.When(x => x.WhenWeGetThatLoadBalancer(key))
.When(x => x.WhenWeGetTheLoadBalancer(key))
.Then(x => x.ThenTheLoadBalancerIs<FakeLoadBalancer>())
.When(x => x.WhenWeGetThatLoadBalancer(keyTwo))
.When(x => x.WhenWeGetTheLoadBalancer(keyTwo))
.Then(x => x.ThenTheLoadBalancerIs<FakeRoundRobinLoadBalancer>())
.BDDfy();
}
[Fact]
public void should_return_error_if_no_load_balancer_with_key()
{
this.When(x => x.WhenWeGetTheLoadBalancer("test"))
.Then(x => x.ThenAnErrorIsReturned())
.BDDfy();
}
private void ThenAnErrorIsReturned()
{
_getResult.IsError.ShouldBeTrue();
_getResult.Errors[0].ShouldBeOfType<UnableToFindLoadBalancerError>();
}
private void ThenTheLoadBalancerIs<T>()
{
_getResult.Data.ShouldBeOfType<T>();
@ -82,7 +96,7 @@ namespace Ocelot.UnitTests.LoadBalancer
WhenIAddTheLoadBalancer();
}
private void WhenWeGetThatLoadBalancer(string key)
private void WhenWeGetTheLoadBalancer(string key)
{
_getResult = _loadBalancerHouse.Get(key);
}

126
test/Ocelot.UnitTests/ServiceDiscovery/ServiceProviderHouseTests.cs Normal file
View File

@ -0,0 +1,126 @@
using System;
using System.Collections.Generic;
using Ocelot.LoadBalancer.LoadBalancers;
using Ocelot.Responses;
using Ocelot.ServiceDiscovery;
using Ocelot.Values;
using Shouldly;
using TestStack.BDDfy;
using Xunit;
namespace Ocelot.UnitTests.LoadBalancer
{
public class ServiceProviderHouseTests
{
private Ocelot.ServiceDiscovery.IServiceProvider _serviceProvider;
private readonly ServiceProviderHouse _serviceProviderHouse;
private Response _addResult;
private Response<Ocelot.ServiceDiscovery.IServiceProvider> _getResult;
private string _key;
public ServiceProviderHouseTests()
{
_serviceProviderHouse = new ServiceProviderHouse();
}
[Fact]
public void should_store_service_provider()
{
var key = "test";
this.Given(x => x.GivenThereIsAServiceProvider(key, new FakeServiceProvider()))
.When(x => x.WhenIAddTheServiceProvider())
.Then(x => x.ThenItIsAdded())
.BDDfy();
}
[Fact]
public void should_get_service_provider()
{
var key = "test";
this.Given(x => x.GivenThereIsAServiceProvider(key, new FakeServiceProvider()))
.When(x => x.WhenWeGetTheServiceProvider(key))
.Then(x => x.ThenItIsReturned())
.BDDfy();
}
[Fact]
public void should_store_service_providers_by_key()
{
var key = "test";
var keyTwo = "testTwo";
this.Given(x => x.GivenThereIsAServiceProvider(key, new FakeServiceProvider()))
.And(x => x.GivenThereIsAServiceProvider(keyTwo, new FakeConsulServiceProvider()))
.When(x => x.WhenWeGetTheServiceProvider(key))
.Then(x => x.ThenTheServiceProviderIs<FakeServiceProvider>())
.When(x => x.WhenWeGetTheServiceProvider(keyTwo))
.Then(x => x.ThenTheServiceProviderIs<FakeConsulServiceProvider>())
.BDDfy();
}
[Fact]
public void should_return_error_if_no_service_provider_house_with_key()
{
this.When(x => x.WhenWeGetTheServiceProvider("test"))
.Then(x => x.ThenAnErrorIsReturned())
.BDDfy();
}
private void ThenAnErrorIsReturned()
{
_getResult.IsError.ShouldBeTrue();
_getResult.Errors[0].ShouldBeOfType<UnableToFindServiceProviderError>();
}
private void ThenTheServiceProviderIs<T>()
{
_getResult.Data.ShouldBeOfType<T>();
}
private void ThenItIsAdded()
{
_addResult.IsError.ShouldBe(false);
_addResult.ShouldBeOfType<OkResponse>();
}
private void WhenIAddTheServiceProvider()
{
_addResult = _serviceProviderHouse.Add(_key, _serviceProvider);
}
private void GivenThereIsAServiceProvider(string key, Ocelot.ServiceDiscovery.IServiceProvider serviceProvider)
{
_key = key;
_serviceProvider = serviceProvider;
WhenIAddTheServiceProvider();
}
private void WhenWeGetTheServiceProvider(string key)
{
_getResult = _serviceProviderHouse.Get(key);
}
private void ThenItIsReturned()
{
_getResult.Data.ShouldBe(_serviceProvider);
}
class FakeServiceProvider : Ocelot.ServiceDiscovery.IServiceProvider
{
public List<Service> Get()
{
throw new NotImplementedException();
}
}
class FakeConsulServiceProvider : Ocelot.ServiceDiscovery.IServiceProvider
{
public List<Service> Get()
{
throw new NotImplementedException();
}
}
}
}