diff --git a/src/Cassandra.IntegrationTests/Cassandra.IntegrationTests.csproj b/src/Cassandra.IntegrationTests/Cassandra.IntegrationTests.csproj
index da119bc7d..76d5dbb8f 100644
--- a/src/Cassandra.IntegrationTests/Cassandra.IntegrationTests.csproj
+++ b/src/Cassandra.IntegrationTests/Cassandra.IntegrationTests.csproj
@@ -61,7 +61,7 @@
-
+
diff --git a/src/Cassandra.Tests/Cassandra.Tests.csproj b/src/Cassandra.Tests/Cassandra.Tests.csproj
index 4cc3c088d..c0cc70ff9 100644
--- a/src/Cassandra.Tests/Cassandra.Tests.csproj
+++ b/src/Cassandra.Tests/Cassandra.Tests.csproj
@@ -30,7 +30,7 @@
-
+
diff --git a/src/Cassandra.Tests/Mapping/BatchTests.cs b/src/Cassandra.Tests/Mapping/BatchTests.cs
index fd697d0a1..1abf03d4a 100644
--- a/src/Cassandra.Tests/Mapping/BatchTests.cs
+++ b/src/Cassandra.Tests/Mapping/BatchTests.cs
@@ -48,12 +48,12 @@ private IMapper GetMapper(Func> getRowSetFunc, Action s.ExecuteAsync(It.IsAny()))
.Returns(getRowSetFunc)
- .Callback(statementCallback)
+ .Callback(stmt => statementCallback((BatchStatement)stmt))
.Verifiable();
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny(), It.IsAny()))
.Returns(getRowSetFunc)
- .Callback((bs, profile) => statementCallback(bs))
+ .Callback((stmt, profile) => statementCallback((BatchStatement)stmt))
.Verifiable();
sessionMock
.Setup(s => s.PrepareAsync(It.IsAny()))
@@ -77,7 +77,7 @@ private MapperAndSessionTuple GetMapperAndSession(Func> getRowSetFu
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny()))
.Returns(getRowSetFunc)
- .Callback(statementCallback)
+ .Callback(stmt => statementCallback((BatchStatement)stmt))
.Verifiable();
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny(), It.IsAny()))
diff --git a/src/Cassandra.Tests/Mapping/DeleteTests.cs b/src/Cassandra.Tests/Mapping/DeleteTests.cs
index 48785e45a..8861110a2 100644
--- a/src/Cassandra.Tests/Mapping/DeleteTests.cs
+++ b/src/Cassandra.Tests/Mapping/DeleteTests.cs
@@ -97,7 +97,7 @@ public void Insert_SetTimestamp_Test()
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny()))
.Returns(() => TestHelper.DelayedTask(RowSet.Empty()))
- .Callback(stmt => statement = stmt)
+ .Callback(stmt => statement = (BoundStatement)stmt)
.Verifiable();
sessionMock
.Setup(s => s.PrepareAsync(It.IsAny()))
@@ -106,7 +106,7 @@ public void Insert_SetTimestamp_Test()
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny(), It.IsAny()))
.Returns(() => TestHelper.DelayedTask(RowSet.Empty()))
- .Callback((stmt, profile) => statement = stmt)
+ .Callback((stmt, profile) => statement = (BoundStatement)stmt)
.Verifiable();
sessionMock
.Setup(s => s.PrepareAsync(It.IsAny()))
diff --git a/src/Cassandra.Tests/Mapping/InsertTests.cs b/src/Cassandra.Tests/Mapping/InsertTests.cs
index ce6656a79..53da130ec 100644
--- a/src/Cassandra.Tests/Mapping/InsertTests.cs
+++ b/src/Cassandra.Tests/Mapping/InsertTests.cs
@@ -199,10 +199,11 @@ public void Insert_Without_Nulls()
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny(), It.IsAny()))
.Returns(TaskHelper.ToTask(new RowSet()))
- .Callback((stmt, profile) =>
+ .Callback((stmt, profile) =>
{
- query = stmt.PreparedStatement.Cql;
- parameters = stmt.QueryValues;
+ var boundStmt = (BoundStatement)stmt;
+ query = boundStmt.PreparedStatement.Cql;
+ parameters = boundStmt.QueryValues;
})
.Verifiable();
sessionMock
@@ -268,7 +269,7 @@ public void InsertIfNotExists_Poco_AppliedInfo_True_Test()
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny(), It.IsAny()))
.Returns(TestHelper.DelayedTask(TestDataHelper.CreateMultipleValuesRowSet(new[] { "[applied]" }, new[] { true })))
- .Callback((b, profile) => query = b.PreparedStatement.Cql)
+ .Callback((stmt, profile) => query = ((BoundStatement)stmt).PreparedStatement.Cql)
.Verifiable();
sessionMock
.Setup(s => s.PrepareAsync(It.IsAny()))
@@ -299,7 +300,7 @@ public void InsertIfNotExists_Poco_AppliedInfo_False_Test()
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny(), It.IsAny()))
.Returns(TestHelper.DelayedTask(TestDataHelper.CreateMultipleValuesRowSet(new[] { "[applied]", "userid", "name" }, new object[] { false, newUser.Id, "existing-name" })))
- .Callback((b, profile) => query = b.PreparedStatement.Cql)
+ .Callback((stmt, profile) => query = ((BoundStatement)stmt).PreparedStatement.Cql)
.Verifiable();
sessionMock
.Setup(s => s.PrepareAsync(It.IsAny()))
@@ -364,7 +365,7 @@ public void Insert_SetTimestamp_Test()
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny(), It.IsAny()))
.Returns(TaskHelper.ToTask(new RowSet()))
- .Callback((stmt, profile) => statement = stmt)
+ .Callback((stmt, profile) => statement = (BoundStatement)stmt)
.Verifiable();
sessionMock
.Setup(s => s.PrepareAsync(It.IsAny()))
diff --git a/src/Cassandra.Tests/Mapping/MappingTestBase.cs b/src/Cassandra.Tests/Mapping/MappingTestBase.cs
index 1f5b62464..be77d1fb3 100644
--- a/src/Cassandra.Tests/Mapping/MappingTestBase.cs
+++ b/src/Cassandra.Tests/Mapping/MappingTestBase.cs
@@ -67,12 +67,12 @@ protected MapperAndSessionTuple GetMappingClientAndSession(Func> ge
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny()))
.Returns(getRowSetFunc)
- .Callback(s => queryCallback(s.PreparedStatement.Cql, s.QueryValues))
+ .Callback(stmt => { var s = (BoundStatement)stmt; queryCallback(s.PreparedStatement.Cql, s.QueryValues); })
.Verifiable();
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny(), It.IsAny()))
.Returns(getRowSetFunc)
- .Callback((s, profile) => queryCallback(s.PreparedStatement.Cql, s.QueryValues))
+ .Callback((stmt, profile) => { var s = (BoundStatement)stmt; queryCallback(s.PreparedStatement.Cql, s.QueryValues); })
.Verifiable();
sessionMock
.Setup(s => s.PrepareAsync(It.IsAny()))
@@ -117,7 +117,7 @@ protected ISession GetSession(RowSet rs, Action callback
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny()))
.Returns(() => TaskHelper.ToTask(rs))
- .Callback(callback)
+ .Callback(stmt => callback((TStatement)stmt))
.Verifiable();
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny(), It.IsAny()))
diff --git a/src/Cassandra.Tests/Mapping/UpdateTests.cs b/src/Cassandra.Tests/Mapping/UpdateTests.cs
index dd65adb69..782f3eab7 100644
--- a/src/Cassandra.Tests/Mapping/UpdateTests.cs
+++ b/src/Cassandra.Tests/Mapping/UpdateTests.cs
@@ -211,8 +211,9 @@ public void UpdateIf_AppliedInfo_True_Test()
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny()))
.Returns(TestHelper.DelayedTask(TestDataHelper.CreateMultipleValuesRowSet(new[] { "[applied]" }, new[] { true })))
- .Callback(b =>
+ .Callback(stmt =>
{
+ var b = (BoundStatement)stmt;
parameters = b.QueryValues;
query = b.PreparedStatement.Cql;
})
@@ -246,8 +247,9 @@ public void UpdateIf_AppliedInfo_False_Test()
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny()))
.Returns(TestHelper.DelayedTask(TestDataHelper.CreateMultipleValuesRowSet(new[] { "[applied]", "id", "artist" }, new object[] { false, id, "Jimmy Page" })))
- .Callback(b =>
+ .Callback(stmt =>
{
+ var b = (BoundStatement)stmt;
parameters = b.QueryValues;
query = b.PreparedStatement.Cql;
})
@@ -278,12 +280,12 @@ public void Update_SetTimestamp_Test()
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny(), It.IsAny()))
.Returns(() => TestHelper.DelayedTask(RowSet.Empty()))
- .Callback((stmt, profile) => statement = stmt)
+ .Callback((stmt, profile) => statement = (BoundStatement)stmt)
.Verifiable();
sessionMock
.Setup(s => s.ExecuteAsync(It.IsAny()))
.Returns(() => TestHelper.DelayedTask(RowSet.Empty()))
- .Callback(stmt => statement = stmt)
+ .Callback(stmt => statement = (BoundStatement)stmt)
.Verifiable();
sessionMock
.Setup(s => s.PrepareAsync(It.IsAny()))