public void Cluster_Uses_Provided_Timestamp_Generator()
{
var generator = new TestTimestampGenerator();
using (var cluster = Cluster.Builder()
.WithTimestampGenerator(generator)
.AddContactPoint(TestCluster.InitialContactPoint)
.Build())
{
var session = cluster.Connect();
TestHelper.ParallelInvoke(() => session.Execute("SELECT * FROM system.local"), 10);
// The driver should use the generator against C* 2.1+
Assert.AreEqual(GetProtocolVersion() < ProtocolVersion.V3 ? 0 : 10, generator.GetCounter());
}
}
public void Should_Use_Statement_Timestamp_Precedence_Over_Cluster_Timestamp_Generator()
{
var generator = new TestTimestampGenerator();
using (var cluster = Cluster.Builder()
.WithTimestampGenerator(generator)
.AddContactPoint(TestCluster.InitialContactPoint)
.Build())
{
var session = cluster.Connect();
var stmt = new SimpleStatement("SELECT * FROM system.local");
stmt.SetTimestamp(DateTimeOffset.Now);
if (GetProtocolVersion() < ProtocolVersion.V3)
{
Assert.Throws <NotSupportedException>(() => session.Execute(stmt));
}
else
{
session.Execute(stmt);
}
// The driver should use the generator against C* 2.1+
Assert.AreEqual(0, generator.GetCounter());
}
}
