tk/Ocelot
1
0
Fork 0
mirror of https://github.com/nsnail/Ocelot.git synced 2025-11-04 17:30:48 +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
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 System.Collections.Generic;
using System.Collections;
using System.Threading;
using Ocelot.Middleware;
using Ocelot.UnitTests.Responder;
using TestStack.BDDfy;
@@ -19,7 +18,7 @@ namespace Ocelot.UnitTests.LoadBalancer
public class CookieStickySessionsTests
{
private CookieStickySessions _stickySessions;
private readonly CookieStickySessions _stickySessions;
private readonly Mock<ILoadBalancer> _loadBalancer;
private readonly int _defaultExpiryInMs;
private DownstreamContext _downstreamContext;
@@ -37,6 +36,18 @@ namespace Ocelot.UnitTests.LoadBalancer
_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]
public void should_return_host_and_port()
{
@@ -81,6 +92,22 @@ namespace Ocelot.UnitTests.LoadBalancer
_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()
{
_result.IsError.ShouldBeTrue();
@@ -208,17 +235,31 @@ namespace Ocelot.UnitTests.LoadBalancer
public FakeBus()
{
Messages = new List<T>();
Subscriptions = new List<Action<T>>();
}
public List<T> Messages { get; }
public List<Action<T>> Subscriptions { get; }
public void Subscribe(Action<T> action)
{
Subscriptions.Add(action);
}
public void Publish(T message, int delay)
{
Messages.Add(message);
}
public void Process()
{
foreach (var message in Messages)
{
foreach (var subscription in Subscriptions)
{
subscription(message);
}
}
}
}
}