tk/Ocelot
1
0
Fork 0
mirror of https://github.com/nsnail/Ocelot.git synced 2025-04-22 06:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

sorted sticky session cookie tests

Browse Source
This commit is contained in:
Tom Pallister 2018-05-05 15:42:26 +01:00
parent fb3af754ab
commit b536a88dca
1 changed files with 43 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
test/Ocelot.UnitTests/LoadBalancer/CookieStickySessionsTests.cs
View File

@ -11,7 +11,6 @@ namespace Ocelot.UnitTests.LoadBalancer
using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Http;
using System.Collections.Generic; using System.Collections.Generic;
using System.Collections; using System.Collections;
using System.Threading;
using Ocelot.Middleware; using Ocelot.Middleware;
using Ocelot.UnitTests.Responder; using Ocelot.UnitTests.Responder;
using TestStack.BDDfy; using TestStack.BDDfy;
@ -19,7 +18,7 @@ namespace Ocelot.UnitTests.LoadBalancer
public class CookieStickySessionsTests public class CookieStickySessionsTests
{ {
private CookieStickySessions _stickySessions; private readonly CookieStickySessions _stickySessions;
private readonly Mock<ILoadBalancer> _loadBalancer; private readonly Mock<ILoadBalancer> _loadBalancer;
private readonly int _defaultExpiryInMs; private readonly int _defaultExpiryInMs;
private DownstreamContext _downstreamContext; private DownstreamContext _downstreamContext;
@ -37,6 +36,18 @@ namespace Ocelot.UnitTests.LoadBalancer
_downstreamContext = new DownstreamContext(new DefaultHttpContext()); _downstreamContext = new DownstreamContext(new DefaultHttpContext());
} }
[Fact]
public void should_expire_sticky_session()
{
this.Given(_ => GivenTheLoadBalancerReturns())
.And(_ => GivenTheDownstreamRequestHasSessionId("321"))
.And(_ => GivenIHackAMessageInWithAPastExpiry())
.And(_ => WhenILease())
.When(_ => WhenTheMessagesAreProcessed())
.Then(_ => ThenTheLoadBalancerIsCalled())
.BDDfy();
}
[Fact] [Fact]
public void should_return_host_and_port() public void should_return_host_and_port()
{ {
@ -81,6 +92,22 @@ namespace Ocelot.UnitTests.LoadBalancer
_stickySessions.Release(new ServiceHostAndPort("", 0)); _stickySessions.Release(new ServiceHostAndPort("", 0));
} }
private void ThenTheLoadBalancerIsCalled()
{
_loadBalancer.Verify(x => x.Release(It.IsAny<ServiceHostAndPort>()), Times.Once);
}
private void WhenTheMessagesAreProcessed()
{
_bus.Process();
}
private void GivenIHackAMessageInWithAPastExpiry()
{
var hostAndPort = new ServiceHostAndPort("999", 999);
_bus.Publish(new StickySession(hostAndPort, DateTime.UtcNow.AddDays(-1), "321"), 0);
}
private void ThenAnErrorIsReturned() private void ThenAnErrorIsReturned()
{ {
_result.IsError.ShouldBeTrue(); _result.IsError.ShouldBeTrue();
@ -208,17 +235,31 @@ namespace Ocelot.UnitTests.LoadBalancer
public FakeBus() public FakeBus()
{ {
Messages = new List<T>(); Messages = new List<T>();
Subscriptions = new List<Action<T>>();
} }
public List<T> Messages { get; } public List<T> Messages { get; }
public List<Action<T>> Subscriptions { get; }
public void Subscribe(Action<T> action) public void Subscribe(Action<T> action)
{ {
Subscriptions.Add(action);
} }
public void Publish(T message, int delay) public void Publish(T message, int delay)
{ {
Messages.Add(message); Messages.Add(message);
} }
public void Process()
{
foreach (var message in Messages)
{
foreach (var subscription in Subscriptions)
{
subscription(message);
}
}
}
} }
} }