public void FindAffinitizedDestination_AffinityKeyIsSetOnRequest_Success()
{
var provider = new CookieSessionAffinityProvider(
Options.Create(_defaultProviderOptions),
AffinityTestHelper.GetDataProtector().Object,
AffinityTestHelper.GetLogger <CookieSessionAffinityProvider>().Object);
var context = new DefaultHttpContext();
var affinitizedDestination = _destinations[1];
context.Request.Headers["Cookie"] = GetCookieWithAffinity(affinitizedDestination);
var affinityResult = provider.FindAffinitizedDestinations(context, _destinations, "cluster-1", _defaultOptions);
Assert.Equal(AffinityStatus.OK, affinityResult.Status);
Assert.Equal(1, affinityResult.Destinations.Count);
Assert.Same(affinitizedDestination, affinityResult.Destinations[0]);
}
public void FindAffinitizedDestination_AffinityKeyIsNotSetOnRequest_ReturnKeyNotSet()
{
var provider = new CookieSessionAffinityProvider(
Options.Create(_defaultProviderOptions),
AffinityTestHelper.GetDataProtector().Object,
AffinityTestHelper.GetLogger <CookieSessionAffinityProvider>().Object);
Assert.Equal(SessionAffinityConstants.Modes.Cookie, provider.Mode);
var context = new DefaultHttpContext();
context.Request.Headers["Cookie"] = new[] { $"Some-Cookie=ZZZ" };
var affinityResult = provider.FindAffinitizedDestinations(context, _destinations, "cluster-1", _defaultOptions);
Assert.Equal(AffinityStatus.AffinityKeyNotSet, affinityResult.Status);
Assert.Null(affinityResult.Destinations);
}
public void AffinitizedRequest_AffinityKeyIsExtracted_DoNothing()
{
var provider = new CookieSessionAffinityProvider(
Options.Create(_defaultProviderOptions),
AffinityTestHelper.GetDataProtector().Object,
AffinityTestHelper.GetLogger <CookieSessionAffinityProvider>().Object);
var context = new DefaultHttpContext();
var affinitizedDestination = _destinations[0];
context.Request.Headers["Cookie"] = GetCookieWithAffinity(affinitizedDestination);
var affinityResult = provider.FindAffinitizedDestinations(context, _destinations, "cluster-1", _defaultOptions);
Assert.Equal(AffinityStatus.OK, affinityResult.Status);
provider.AffinitizeRequest(context, _defaultOptions, affinitizedDestination);
Assert.False(context.Response.Headers.ContainsKey("Cookie"));
}
