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()))