public void EnableSession_InvalidMode_Fails()
{
var affinityProvider = new Mock <ISessionAffinityProvider>(MockBehavior.Strict);
affinityProvider.SetupGet(p => p.Mode).Returns("Mode");
var transformProvider = new AffinitizeTransformProvider(new[] { affinityProvider.Object });
var cluster = new Cluster
{
Id = "cluster1",
SessionAffinity = new SessionAffinityOptions()
{
Enabled = true,
Mode = "Invalid",
}
};
var validationContext = new TransformClusterValidationContext()
{
Cluster = cluster,
};
transformProvider.ValidateCluster(validationContext);
var ex = Assert.Single(validationContext.Errors);
Assert.Equal("No matching ISessionAffinityProvider found for the session affinity mode 'Invalid' set on the cluster 'cluster1'.", ex.Message);
var builderContext = new TransformBuilderContext()
{
Cluster = cluster,
};
ex = Assert.Throws <ArgumentException>(() => transformProvider.Apply(builderContext));
Assert.Equal("No Yarp.ReverseProxy.Service.SessionAffinity.ISessionAffinityProvider was found for the id 'Invalid'. (Parameter 'id')", ex.Message);
}
public void EnableSessionAffinity_AddsTransform()
{
var affinityProvider = new Mock <ISessionAffinityProvider>(MockBehavior.Strict);
affinityProvider.SetupGet(p => p.Mode).Returns("Mode");
var transformProvider = new AffinitizeTransformProvider(new[] { affinityProvider.Object });
var cluster = new Cluster
{
Id = "cluster1",
SessionAffinity = new SessionAffinityOptions()
{
Enabled = true,
Mode = "Mode",
}
};
var validationContext = new TransformClusterValidationContext()
{
Cluster = cluster,
};
transformProvider.ValidateCluster(validationContext);
Assert.Empty(validationContext.Errors);
var builderContext = new TransformBuilderContext()
{
Cluster = cluster,
};
transformProvider.Apply(builderContext);
Assert.IsType <AffinitizeTransform>(builderContext.ResponseTransforms.Single());
}