diff --git a/testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionReadTransaction.java b/testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionReadTransaction.java index b032547164..350ce7d7f0 100644 --- a/testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionReadTransaction.java +++ b/testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionReadTransaction.java @@ -20,6 +20,7 @@ import lombok.Getter; import lombok.Setter; +import neo4j.org.testkit.backend.FrontendError; import neo4j.org.testkit.backend.TestkitState; import neo4j.org.testkit.backend.holder.AsyncTransactionHolder; import neo4j.org.testkit.backend.holder.RxTransactionHolder; @@ -123,10 +124,11 @@ private TransactionWork handle( TestkitState testkitState, SessionHolder s { throw (Neo4jException) workThrowable; } - else + if ( workThrowable instanceof FrontendError ) { - throw new RuntimeException( "Unexpected exception occurred in transaction work function", workThrowable ); + throw (FrontendError) workThrowable; } + throw new RuntimeException( "Unexpected exception occurred in transaction work function", workThrowable ); } }; } diff --git a/testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionWriteTransaction.java b/testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionWriteTransaction.java index 65caff854e..1d0a734ee1 100644 --- a/testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionWriteTransaction.java +++ b/testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionWriteTransaction.java @@ -20,6 +20,7 @@ import lombok.Getter; import lombok.Setter; +import neo4j.org.testkit.backend.FrontendError; import neo4j.org.testkit.backend.TestkitState; import neo4j.org.testkit.backend.holder.AsyncTransactionHolder; import neo4j.org.testkit.backend.holder.RxTransactionHolder; @@ -126,10 +127,11 @@ private TransactionWork handle( TestkitState testkitState, SessionHolder s { throw (Neo4jException) workThrowable; } - else + if ( workThrowable instanceof FrontendError ) { - throw new RuntimeException( "Unexpected exception occurred in transaction work function", workThrowable ); + throw (FrontendError) workThrowable; } + throw new RuntimeException( "Unexpected exception occurred in transaction work function", workThrowable ); } }; }