QueryService: Rewrite tests

ydb-platform · Nov 14, 2023 · a9fe3f6 · a9fe3f6
1 parent 29123b1
commit a9fe3f6
Show file tree

Hide file tree

Showing 3 changed files with 199 additions and 346 deletions.
diff --git a/src/Ydb.Sdk/src/Services/Query/QueryClient.cs b/src/Ydb.Sdk/src/Services/Query/QueryClient.cs
@@ -685,12 +685,16 @@ public async Task<QueryResponseWithResult<T>> DoTx<T>(Func<Tx, Task<T>> func,
                 }
                 catch (StatusUnsuccessfulException e)
                 {
-                    return await Rollback<T>(session, tx, e.Status);
+                    var rollbackResponse = await Rollback<T>(session, tx, e.Status);
+                    return rollbackResponse;
                 }
                 catch (Exception e)
                 {
-                    return await Rollback<T>(session, tx,
-                        new Status(StatusCode.InternalError, $"Failed to execute lambda on tx {tx.TxId}: {e.Message}"));
+                    var status = new Status(
+                        StatusCode.InternalError,
+                        $"Failed to execute lambda on tx {tx.TxId}: {e.Message}");
+                    var rollbackResponse = await Rollback<T>(session, tx, status);
+                    return rollbackResponse;
                 }
 
                 // var commitResponse = await CommitTransaction(session.Id, tx);

diff --git a/src/Ydb.Sdk/tests/Query/TestExecuteQuery.cs b/src/Ydb.Sdk/tests/Query/TestExecuteQuery.cs