public virtual void TestLoadBalancingWithFailure()
{
Configuration conf = new Configuration();
KMSClientProvider p1 = Org.Mockito.Mockito.Mock <KMSClientProvider>();
Org.Mockito.Mockito.When(p1.CreateKey(Org.Mockito.Mockito.AnyString(), Org.Mockito.Mockito
.Any <KeyProvider.Options>())).ThenReturn(new KMSClientProvider.KMSKeyVersion("p1"
, "v1", new byte[0]));
Org.Mockito.Mockito.When(p1.GetKMSUrl()).ThenReturn("p1");
// This should not be retried
KMSClientProvider p2 = Org.Mockito.Mockito.Mock <KMSClientProvider>();
Org.Mockito.Mockito.When(p2.CreateKey(Org.Mockito.Mockito.AnyString(), Org.Mockito.Mockito
.Any <KeyProvider.Options>())).ThenThrow(new NoSuchAlgorithmException("p2"));
Org.Mockito.Mockito.When(p2.GetKMSUrl()).ThenReturn("p2");
KMSClientProvider p3 = Org.Mockito.Mockito.Mock <KMSClientProvider>();
Org.Mockito.Mockito.When(p3.CreateKey(Org.Mockito.Mockito.AnyString(), Org.Mockito.Mockito
.Any <KeyProvider.Options>())).ThenReturn(new KMSClientProvider.KMSKeyVersion("p3"
, "v3", new byte[0]));
Org.Mockito.Mockito.When(p3.GetKMSUrl()).ThenReturn("p3");
// This should be retried
KMSClientProvider p4 = Org.Mockito.Mockito.Mock <KMSClientProvider>();
Org.Mockito.Mockito.When(p4.CreateKey(Org.Mockito.Mockito.AnyString(), Org.Mockito.Mockito
.Any <KeyProvider.Options>())).ThenThrow(new IOException("p4"));
Org.Mockito.Mockito.When(p4.GetKMSUrl()).ThenReturn("p4");
KeyProvider kp = new LoadBalancingKMSClientProvider(new KMSClientProvider[] { p1,
p2, p3, p4 }, 0, conf);
Assert.Equal("p1", kp.CreateKey("test4", new KeyProvider.Options
(conf)).GetName());
// Exceptions other than IOExceptions will not be retried
try
{
kp.CreateKey("test1", new KeyProvider.Options(conf)).GetName();
NUnit.Framework.Assert.Fail("Should fail since its not an IOException");
}
catch (Exception e)
{
Assert.True(e is NoSuchAlgorithmException);
}
Assert.Equal("p3", kp.CreateKey("test2", new KeyProvider.Options
(conf)).GetName());
// IOException will trigger retry in next provider
Assert.Equal("p1", kp.CreateKey("test3", new KeyProvider.Options
(conf)).GetName());
}
/// <exception cref="System.IO.IOException"/>
private T DoOp <T>(LoadBalancingKMSClientProvider.ProviderCallable <T> op, int currPos
)
{
IOException ex = null;
for (int i = 0; i < providers.Length; i++)
{
KMSClientProvider provider = providers[(currPos + i) % providers.Length];
try
{
return(op.Call(provider));
}
catch (IOException ioe)
{
Log.Warn("KMS provider at [{}] threw an IOException [{}]!!", provider.GetKMSUrl()
, ioe.Message);
ex = ioe;
}
catch (Exception e)
{
if (e is RuntimeException)
{
throw (RuntimeException)e;
}
else
{
throw new LoadBalancingKMSClientProvider.WrapperException(e);
}
}
}
if (ex != null)
{
Log.Warn("Aborting since the Request has failed with all KMS" + " providers in the group. !!"
);
throw ex;
}
throw new IOException("No providers configured !!");
}
public virtual void TestLoadBalancingWithAllBadNodes()
{
Configuration conf = new Configuration();
KMSClientProvider p1 = Org.Mockito.Mockito.Mock <KMSClientProvider>();
Org.Mockito.Mockito.When(p1.CreateKey(Org.Mockito.Mockito.AnyString(), Org.Mockito.Mockito
.Any <KeyProvider.Options>())).ThenThrow(new IOException("p1"));
KMSClientProvider p2 = Org.Mockito.Mockito.Mock <KMSClientProvider>();
Org.Mockito.Mockito.When(p2.CreateKey(Org.Mockito.Mockito.AnyString(), Org.Mockito.Mockito
.Any <KeyProvider.Options>())).ThenThrow(new IOException("p2"));
KMSClientProvider p3 = Org.Mockito.Mockito.Mock <KMSClientProvider>();
Org.Mockito.Mockito.When(p3.CreateKey(Org.Mockito.Mockito.AnyString(), Org.Mockito.Mockito
.Any <KeyProvider.Options>())).ThenThrow(new IOException("p3"));
KMSClientProvider p4 = Org.Mockito.Mockito.Mock <KMSClientProvider>();
Org.Mockito.Mockito.When(p4.CreateKey(Org.Mockito.Mockito.AnyString(), Org.Mockito.Mockito
.Any <KeyProvider.Options>())).ThenThrow(new IOException("p4"));
Org.Mockito.Mockito.When(p1.GetKMSUrl()).ThenReturn("p1");
Org.Mockito.Mockito.When(p2.GetKMSUrl()).ThenReturn("p2");
Org.Mockito.Mockito.When(p3.GetKMSUrl()).ThenReturn("p3");
Org.Mockito.Mockito.When(p4.GetKMSUrl()).ThenReturn("p4");
KeyProvider kp = new LoadBalancingKMSClientProvider(new KMSClientProvider[] { p1,
p2, p3, p4 }, 0, conf);
try
{
kp.CreateKey("test3", new KeyProvider.Options(conf)).GetName();
NUnit.Framework.Assert.Fail("Should fail since all providers threw an IOException"
);
}
catch (Exception e)
{
Assert.True(e is IOException);
}
}