public void RetryPolicy_Extended(string resultError, Type exceptionType)
{
_scassandraManager = SCassandraManager.Instance;
var extendedRetryPolicy = new TestExtendedRetryPolicy();
var builder = Cluster.Builder()
.AddContactPoint("127.0.0.1")
.WithPort(_scassandraManager.BinaryPort)
.WithRetryPolicy(extendedRetryPolicy)
.WithReconnectionPolicy(new ConstantReconnectionPolicy(long.MaxValue));
using (var cluster = builder.Build())
{
var session = (Session)cluster.Connect();
const string cql = "select * from table1";
_scassandraManager.PrimeQuery(cql, "{\"result\" : \"" + resultError + "\"}").Wait();
Exception throwedException = null;
try
{
session.Execute(cql);
}
catch (Exception ex)
{
throwedException = ex;
}
finally
{
Assert.NotNull(throwedException);
Assert.AreEqual(throwedException.GetType(), exceptionType);
Assert.AreEqual(1, Interlocked.Read(ref extendedRetryPolicy.RequestErrorConter));
Assert.AreEqual(0, Interlocked.Read(ref extendedRetryPolicy.ReadTimeoutCounter));
Assert.AreEqual(0, Interlocked.Read(ref extendedRetryPolicy.WriteTimeoutCounter));
Assert.AreEqual(0, Interlocked.Read(ref extendedRetryPolicy.UnavailableCounter));
}
}
}
public void MarkHostDown_PartialPoolConnection()
{
//start scassandra
_scassandraManager = new SCassandraManager();
_scassandraManager.Start();
_scassandraManager.SetupInitialConf().Wait();
const int connectionLength = 4;
var builder = Cluster.Builder()
.AddContactPoint("127.0.0.1")
.WithPort(_scassandraManager.BinaryPort)
.WithPoolingOptions(new PoolingOptions()
.SetCoreConnectionsPerHost(HostDistance.Local, connectionLength)
.SetMaxConnectionsPerHost(HostDistance.Local, connectionLength)
.SetHeartBeatInterval(2000))
.WithReconnectionPolicy(new ConstantReconnectionPolicy(long.MaxValue));
using (var cluster = builder.Build())
{
var session = (Session)cluster.Connect();
var allHosts = cluster.AllHosts();
TestHelper.WaitUntil(() =>
allHosts.Sum(h => session
.GetOrCreateConnectionPool(h, HostDistance.Local)
.OpenConnections
) == allHosts.Count * connectionLength);
var h1 = allHosts.FirstOrDefault();
var pool = session.GetOrCreateConnectionPool(h1, HostDistance.Local);
var ports = _scassandraManager.GetListOfConnectedPorts().Result;
// 4 pool connections + the control connection
Assert.AreEqual(5, ports.Length);
_scassandraManager.DisableConnectionListener().Wait();
// Remove the first connections
for (var i = 0; i < 3; i++)
{
// Closure
var index = i;
var expectedOpenConnections = 5 - index;
WaitSimulatorConnections(_scassandraManager, expectedOpenConnections);
ports = _scassandraManager.GetListOfConnectedPorts().Result;
Assert.AreEqual(expectedOpenConnections, ports.Length, "Cassandra simulator contains unexpected number of connected clients");
_scassandraManager.DropConnection(ports.Last()).Wait();
// Host pool could have between pool.OpenConnections - i and pool.OpenConnections - i - 1
TestHelper.WaitUntil(() => pool.OpenConnections >= 4 - index - 1 && pool.OpenConnections <= 4 - index);
Assert.LessOrEqual(pool.OpenConnections, 4 - index);
Assert.GreaterOrEqual(pool.OpenConnections, 4 - index - 1);
Assert.IsTrue(h1.IsUp);
}
WaitSimulatorConnections(_scassandraManager, 2);
ports = _scassandraManager.GetListOfConnectedPorts().Result;
Assert.AreEqual(2, ports.Length);
_scassandraManager.DropConnection(ports[1]).Wait();
_scassandraManager.DropConnection(ports[0]).Wait();
TestHelper.WaitUntil(() => pool.OpenConnections == 0 && !h1.IsUp);
Assert.IsFalse(h1.IsUp);
}
}
/// <summary>
/// Waits for the simulator to have the expected number of connections
/// </summary>
private static void WaitSimulatorConnections(SCassandraManager scassandra, int expected)
{
TestHelper.WaitUntil(() => scassandra.GetListOfConnectedPorts().Result.Length == expected);
}
