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")); }