Ensure error is propagated from rx queries correctly.

Zhen Li · Zhen Li · commit e818e55a1568 · 2020-02-24T17:34:40.000+01:00
The rx queries should be semantically the same as the async queries.
Fixed a few wrong handling of finish future and error propergation used in rx queries.
diff --git a/driver/src/test/java/org/neo4j/driver/stress/RxFailingQuery.java b/driver/src/test/java/org/neo4j/driver/stress/RxFailingQuery.java
@@ -23,6 +23,7 @@
 
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.CompletionStage;
+import java.util.concurrent.atomic.AtomicInteger;
 
 import org.neo4j.driver.AccessMode;
 import org.neo4j.driver.Driver;
@@ -51,8 +52,7 @@ public CompletionStage<Void> execute( C context )
                 RxSession::close )
                 .subscribe( record -> {
                     assertThat( record.get( 0 ).asInt(), either( equalTo( 1 ) ).or( equalTo( 2 ) ) );
-                    queryFinished.complete( null );
-                }, error -> {
+                    }, error -> {
                     Throwable cause = Futures.completionExceptionCause( error );
                     assertThat( cause, is( arithmeticError() ) );
                     queryFinished.complete( null );
diff --git a/driver/src/test/java/org/neo4j/driver/stress/RxFailingQueryInTx.java b/driver/src/test/java/org/neo4j/driver/stress/RxFailingQueryInTx.java
@@ -52,7 +52,6 @@ public CompletionStage<Void> execute( C context )
                 RxTransaction::commit, ( tx, error ) -> tx.rollback(), null )
                 .subscribe( record -> {
                     assertThat( record.get( 0 ).asInt(), either( equalTo( 1 ) ).or( equalTo( 2 ) ) );
-                    queryFinished.complete( null );
                 }, error -> {
                     Throwable cause = Futures.completionExceptionCause( error );
                     assertThat( cause, is( arithmeticError() ) );
diff --git a/driver/src/test/java/org/neo4j/driver/stress/RxFailingQueryWithRetries.java b/driver/src/test/java/org/neo4j/driver/stress/RxFailingQueryWithRetries.java
@@ -51,7 +51,6 @@ public CompletionStage<Void> execute( C context )
                 RxSession::close )
                 .subscribe( record -> {
                     assertThat( record.get( 0 ).asInt(), either( equalTo( 1 ) ).or( equalTo( 2 ) ) );
-                    queryFinished.complete( null );
                 }, error -> {
                     Throwable cause = Futures.completionExceptionCause( error );
                     assertThat( cause, is( arithmeticError() ) );
diff --git a/driver/src/test/java/org/neo4j/driver/stress/RxWriteQuery.java b/driver/src/test/java/org/neo4j/driver/stress/RxWriteQuery.java
@@ -48,22 +48,23 @@ public CompletionStage<Void> execute( C context )
         Flux.usingWhen( Mono.fromSupplier( () -> newSession( AccessMode.WRITE, context ) ),
                 session -> session.run( "CREATE ()" ).consume(), RxSession::close )
                 .subscribe( summary -> {
-                    queryFinished.complete( null );
                     assertEquals( 1, summary.counters().nodesCreated() );
                     context.nodeCreated();
-                }, error -> {
                     queryFinished.complete( null );
-                    handleError( Futures.completionExceptionCause( error ), context );
-                } );
+                }, error -> handleError( Futures.completionExceptionCause( error ), context, queryFinished ) );
 
         return queryFinished;
     }
 
-    private void handleError( Throwable error, C context )
+    private void handleError( Throwable error, C context, CompletableFuture<Void> queryFinished )
     {
         if ( !stressTest.handleWriteFailure( error, context ) )
         {
-            throw new RuntimeException( error );
+            queryFinished.completeExceptionally( error );
+        }
+        else
+        {
+            queryFinished.complete( null );
         }
     }
 }
diff --git a/driver/src/test/java/org/neo4j/driver/stress/RxWriteQueryInTx.java b/driver/src/test/java/org/neo4j/driver/stress/RxWriteQueryInTx.java
@@ -53,18 +53,21 @@ public CompletionStage<Void> execute( C context )
                     context.nodeCreated();
                     queryFinished.complete( null );
                 }, error -> {
-                    handleError( Futures.completionExceptionCause( error ), context );
-                    queryFinished.complete( null );
+                    handleError( Futures.completionExceptionCause( error ), context, queryFinished );
                 } );
 
         return queryFinished;
     }
 
-    private void handleError( Throwable error, C context )
+    private void handleError( Throwable error, C context, CompletableFuture<Void> queryFinished )
     {
         if ( !stressTest.handleWriteFailure( error, context ) )
         {
-            throw new RuntimeException( error );
+            queryFinished.completeExceptionally( error );
+        }
+        else
+        {
+            queryFinished.complete( null );
         }
     }
 }
diff --git a/driver/src/test/java/org/neo4j/driver/stress/RxWriteQueryWithRetries.java b/driver/src/test/java/org/neo4j/driver/stress/RxWriteQueryWithRetries.java
@@ -48,22 +48,23 @@ public CompletionStage<Void> execute( C context )
         Flux.usingWhen( Mono.fromSupplier( () -> newSession( AccessMode.READ, context ) ),
                 session -> session.writeTransaction( tx -> tx.run( "CREATE ()" ).consume() ), RxSession::close )
                 .subscribe( summary -> {
-                    queryFinished.complete( null );
                     assertEquals( 1, summary.counters().nodesCreated() );
                     context.nodeCreated();
-                }, error -> {
                     queryFinished.complete( null );
-                    handleError( Futures.completionExceptionCause( error ), context );
-                } );
+                }, error -> handleError( Futures.completionExceptionCause( error ), context, queryFinished ) );
 
         return queryFinished;
     }
 
-    private void handleError( Throwable error, C context )
+    private void handleError( Throwable error, C context, CompletableFuture<Void> queryFinished )
     {
         if ( !stressTest.handleWriteFailure( error, context ) )
         {
-            throw new RuntimeException( error );
+            queryFinished.completeExceptionally( error );
+        }
+        else
+        {
+            queryFinished.complete( null );
         }
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -48,22 +48,23 @@ public CompletionStage<Void> execute( C context )`
`48`	`48`	`Flux.usingWhen( Mono.fromSupplier( () -> newSession( AccessMode.WRITE, context ) ),`
`49`	`49`	`session -> session.run( "CREATE ()" ).consume(), RxSession::close )`
`50`	`50`	`.subscribe( summary -> {`
`51`		`- queryFinished.complete( null );`
`52`	`51`	`assertEquals( 1, summary.counters().nodesCreated() );`
`53`	`52`	`context.nodeCreated();`
`54`		`- }, error -> {`
`55`	`53`	`queryFinished.complete( null );`
`56`		`- handleError( Futures.completionExceptionCause( error ), context );`
`57`		`- } );`
	`54`	`+ }, error -> handleError( Futures.completionExceptionCause( error ), context, queryFinished ) );`
`58`	`55`
`59`	`56`	`return queryFinished;`
`60`	`57`	`}`
`61`	`58`
`62`		`- private void handleError( Throwable error, C context )`
	`59`	`+ private void handleError( Throwable error, C context, CompletableFuture<Void> queryFinished )`
`63`	`60`	`{`
`64`	`61`	`if ( !stressTest.handleWriteFailure( error, context ) )`
`65`	`62`	`{`
`66`		`- throw new RuntimeException( error );`
	`63`	`+ queryFinished.completeExceptionally( error );`
	`64`	`+ }`
	`65`	`+ else`
	`66`	`+ {`
	`67`	`+ queryFinished.complete( null );`
`67`	`68`	`}`
`68`	`69`	`}`
`69`	`70`	`}`
Original file line number	Diff line number	Diff line change
`@@ -53,18 +53,21 @@ public CompletionStage<Void> execute( C context )`
`53`	`53`	`context.nodeCreated();`
`54`	`54`	`queryFinished.complete( null );`
`55`	`55`	`}, error -> {`
`56`		`- handleError( Futures.completionExceptionCause( error ), context );`
`57`		`- queryFinished.complete( null );`
	`56`	`+ handleError( Futures.completionExceptionCause( error ), context, queryFinished );`
`58`	`57`	`} );`
`59`	`58`
`60`	`59`	`return queryFinished;`
`61`	`60`	`}`
`62`	`61`
`63`		`- private void handleError( Throwable error, C context )`
	`62`	`+ private void handleError( Throwable error, C context, CompletableFuture<Void> queryFinished )`
`64`	`63`	`{`
`65`	`64`	`if ( !stressTest.handleWriteFailure( error, context ) )`
`66`	`65`	`{`
`67`		`- throw new RuntimeException( error );`
	`66`	`+ queryFinished.completeExceptionally( error );`
	`67`	`+ }`
	`68`	`+ else`
	`69`	`+ {`
	`70`	`+ queryFinished.complete( null );`
`68`	`71`	`}`
`69`	`72`	`}`
`70`	`73`	`}`