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

another quick hack to work this out

Browse Source
This commit is contained in:
Tom Gardham-Pallister 2018-05-03 22:32:07 +01:00
parent e57b72173e
commit 334d4bf368
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -17,8 +17,9 @@ namespace Ocelot.UnitTests.LoadBalancer
public class CookieStickySessionsTests public class CookieStickySessionsTests
{ {
private readonly CookieStickySessions _stickySessions; private CookieStickySessions _stickySessions;
private readonly Mock<ILoadBalancer> _loadBalancer; private readonly Mock<ILoadBalancer> _loadBalancer;
private readonly int _defaultExpiryInMs;
private DownstreamContext _downstreamContext; private DownstreamContext _downstreamContext;
private Response<ServiceHostAndPort> _result; private Response<ServiceHostAndPort> _result;
private Response<ServiceHostAndPort> _firstHostAndPort; private Response<ServiceHostAndPort> _firstHostAndPort;
@ -27,8 +28,8 @@ namespace Ocelot.UnitTests.LoadBalancer
public CookieStickySessionsTests() public CookieStickySessionsTests()
{ {
_loadBalancer = new Mock<ILoadBalancer>(); _loadBalancer = new Mock<ILoadBalancer>();
const int defaultExpiryInMs = 100; _defaultExpiryInMs = 100;
_stickySessions = new CookieStickySessions(_loadBalancer.Object, "sessionid", defaultExpiryInMs, 1); _stickySessions = new CookieStickySessions(_loadBalancer.Object, "sessionid", _defaultExpiryInMs, 1);
_downstreamContext = new DownstreamContext(new DefaultHttpContext()); _downstreamContext = new DownstreamContext(new DefaultHttpContext());
} }
@ -72,6 +73,8 @@ namespace Ocelot.UnitTests.LoadBalancer
[Fact] [Fact]
public void should_expire_sticky_session() public void should_expire_sticky_session()
{ {
_stickySessions = new CookieStickySessions(_loadBalancer.Object, "sessionid", _defaultExpiryInMs, 1);
this.Given(_ => GivenTheLoadBalancerReturnsSequence()) this.Given(_ => GivenTheLoadBalancerReturnsSequence())
.When(_ => WhenTheStickySessionExpires()) .When(_ => WhenTheStickySessionExpires())
.Then(_ => ThenANewHostAndPortIsReturned()) .Then(_ => ThenANewHostAndPortIsReturned())
@ -81,6 +84,8 @@ namespace Ocelot.UnitTests.LoadBalancer
[Fact] [Fact]
public void should_refresh_sticky_session() public void should_refresh_sticky_session()
{ {
_stickySessions = new CookieStickySessions(_loadBalancer.Object, "sessionid", _defaultExpiryInMs, 50);
this.Given(_ => GivenTheLoadBalancerReturnsSequence()) this.Given(_ => GivenTheLoadBalancerReturnsSequence())
.When(_ => WhenIMakeRequestsToKeepRefreshingTheSession()) .When(_ => WhenIMakeRequestsToKeepRefreshingTheSession())
.Then(_ => ThenTheSessionIsRefreshed()) .Then(_ => ThenTheSessionIsRefreshed())