Allow tx timeout to be 0 or null.

driver/src/main/java/org/neo4j/driver/TransactionConfig.java

@@ -183,26 +183,33 @@ public static class Builder
         private Duration timeout;
         private Map<String,Object> metadata = emptyMap();
 
+        /**
+         * Value used to signal {@link #withTimeout(Duration)} to use the server-side configured default timeout.
+         */
+        public static final Duration SERVER_DEFAULT_TIMEOUT = null;
+
         private Builder()
         {
         }
 
         /**
          * Set the transaction timeout. Transactions that execute longer than the configured timeout will be terminated by the database.
+         * Use {@link #SERVER_DEFAULT_TIMEOUT SERVER_DEFAULT_TIMEOUT} (default) to rely on the server-side configured timeout.
          * <p>
          * This functionality allows to limit query/transaction execution time. Specified timeout overrides the default timeout configured in the database
          * using {@code dbms.transaction.timeout} setting.
          * <p>
-         * Provided value should not be {@code null} and should not represent a duration of zero or negative duration.
+         * Provided value should not represent a negative duration.
          *
          * @param timeout the timeout.
          * @return this builder.
          */
         public Builder withTimeout( Duration timeout )
         {
-            requireNonNull( timeout, "Transaction timeout should not be null" );
-            checkArgument( !timeout.isZero(), "Transaction timeout should not be zero" );
-            checkArgument( !timeout.isNegative(), "Transaction timeout should not be negative" );
+            if (timeout != null)
+            {
+                checkArgument( !timeout.isNegative(), "Transaction timeout should not be negative" );
+            }
 
             this.timeout = timeout;
             return this;

driver/src/test/java/org/neo4j/driver/TransactionConfigTest.java

@@ -51,18 +51,6 @@ void emptyConfigShouldHaveNoMetadata()
         assertEquals( emptyMap(), TransactionConfig.empty().metadata() );
     }
 
-    @Test
-    void shouldDisallowNullTimeout()
-    {
-        assertThrows( NullPointerException.class, () -> TransactionConfig.builder().withTimeout( null ) );
-    }
-
-    @Test
-    void shouldDisallowZeroTimeout()
-    {
-        assertThrows( IllegalArgumentException.class, () -> TransactionConfig.builder().withTimeout( Duration.ZERO ) );
-    }
-
     @Test
     void shouldDisallowNegativeTimeout()
     {
@@ -98,6 +86,26 @@ void shouldHaveTimeout()
         assertEquals( Duration.ofSeconds( 3 ), config.timeout() );
     }
 
+    @Test
+    void shouldAllowDefaultTimeout()
+    {
+        TransactionConfig config = TransactionConfig.builder()
+                .withTimeout( TransactionConfig.Builder.SERVER_DEFAULT_TIMEOUT )
+                .build();
+
+        assertNull( config.timeout() );
+    }
+
+    @Test
+    void shouldAllowZeroTimeout()
+    {
+        TransactionConfig config = TransactionConfig.builder()
+                .withTimeout( Duration.ZERO )
+                .build();
+
+        assertEquals( Duration.ZERO, config.timeout() );
+    }
+
     @Test
     void shouldHaveMetadata()
     {

testkit-backend/src/main/java/neo4j/org/testkit/backend/CustomDriverError.java

@@ -0,0 +1,23 @@
+/*
+ * Copyright (c) "Neo4j"
+ * Neo4j Sweden AB [http://neo4j.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package neo4j.org.testkit.backend;
+
+public class CustomDriverError extends java.lang.RuntimeException
+{
+}

testkit-backend/src/main/java/neo4j/org/testkit/backend/channel/handler/TestkitRequestProcessorHandler.java

@@ -21,6 +21,7 @@
 import io.netty.channel.Channel;
 import io.netty.channel.ChannelHandlerContext;
 import io.netty.channel.ChannelInboundHandlerAdapter;
+import neo4j.org.testkit.backend.CustomDriverError;
 import neo4j.org.testkit.backend.TestkitState;
 import neo4j.org.testkit.backend.messages.requests.TestkitRequest;
 import neo4j.org.testkit.backend.messages.responses.BackendError;
@@ -145,6 +146,20 @@ else if ( isConnectionPoolClosedException( throwable ) || throwable instanceof U
                               )
                               .build();
         }
+        else if ( throwable instanceof CustomDriverError )
+        {
+            throwable = throwable.getCause();
+            String id = testkitState.newId();
+            return DriverError.builder()
+                              .data(
+                                      DriverError.DriverErrorBody.builder()
+                                                                 .id( id )
+                                                                 .errorType( throwable.getClass().getName() )
+                                                                 .msg( throwable.getMessage() )
+                                                                 .build()
+                              )
+                              .build();
+        }
         else
         {
             return BackendError.builder().data( BackendError.BackendErrorBody.builder().msg( throwable.toString() ).build() ).build();

testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionBeginTransaction.java

@@ -20,6 +20,7 @@
 
 import lombok.Getter;
 import lombok.Setter;
+import neo4j.org.testkit.backend.CustomDriverError;
 import neo4j.org.testkit.backend.TestkitState;
 import neo4j.org.testkit.backend.holder.AsyncTransactionHolder;
 import neo4j.org.testkit.backend.holder.RxTransactionHolder;
@@ -45,6 +46,30 @@ public class SessionBeginTransaction implements TestkitRequest
 {
     private SessionBeginTransactionBody data;
 
+    private void configureTimeout( TransactionConfig.Builder builder )
+    {
+        if ( data.getTimeoutPresent() )
+        {
+            try
+            {
+                if ( data.getTimeout() != null )
+                {
+                    builder.withTimeout( Duration.ofMillis( data.getTimeout() ) );
+                }
+                else
+                {
+                    builder.withTimeout( TransactionConfig.Builder.SERVER_DEFAULT_TIMEOUT );
+                }
+            }
+            catch ( IllegalArgumentException e )
+            {
+                CustomDriverError wrapped = new CustomDriverError();
+                wrapped.initCause( e );
+                throw wrapped;
+            }
+        }
+    }
+
     @Override
     public TestkitResponse process( TestkitState testkitState )
     {
@@ -53,10 +78,7 @@ public TestkitResponse process( TestkitState testkitState )
         TransactionConfig.Builder builder = TransactionConfig.builder();
         Optional.ofNullable( data.txMeta ).ifPresent( builder::withMetadata );
 
-        if ( data.getTimeout() != null )
-        {
-            builder.withTimeout( Duration.ofMillis( data.getTimeout() ) );
-        }
+        configureTimeout( builder );
 
         org.neo4j.driver.Transaction transaction = session.beginTransaction( builder.build() );
         return transaction( testkitState.addTransactionHolder( new TransactionHolder( sessionHolder, transaction ) ) );
@@ -72,10 +94,7 @@ public CompletionStage<TestkitResponse> processAsync( TestkitState testkitState
                                              TransactionConfig.Builder builder = TransactionConfig.builder();
                                              Optional.ofNullable( data.txMeta ).ifPresent( builder::withMetadata );
 
-                                             if ( data.getTimeout() != null )
-                                             {
-                                                 builder.withTimeout( Duration.ofMillis( data.getTimeout() ) );
-                                             }
+                                             configureTimeout( builder );
 
                                              return session.beginTransactionAsync( builder.build() ).thenApply( tx -> transaction(
                                                      testkitState.addAsyncTransactionHolder( new AsyncTransactionHolder( sessionHolder, tx ) ) ) );
@@ -92,10 +111,7 @@ public Mono<TestkitResponse> processRx( TestkitState testkitState )
                                          TransactionConfig.Builder builder = TransactionConfig.builder();
                                          Optional.ofNullable( data.txMeta ).ifPresent( builder::withMetadata );
 
-                                         if ( data.getTimeout() != null )
-                                         {
-                                             builder.withTimeout( Duration.ofMillis( data.getTimeout() ) );
-                                         }
+                                         configureTimeout( builder );
 
                                          return Mono.fromDirect( session.beginTransaction( builder.build() ) )
                                                     .map( tx -> transaction(
@@ -108,12 +124,23 @@ private Transaction transaction( String txId )
         return Transaction.builder().data( Transaction.TransactionBody.builder().id( txId ).build() ).build();
     }
 
-    @Getter
-    @Setter
     public static class SessionBeginTransactionBody
     {
+        @Getter
+        @Setter
         private String sessionId;
+        @Getter
+        @Setter
         private Map<String,Object> txMeta;
+        @Getter
         private Integer timeout;
+        @Getter
+        private Boolean timeoutPresent = false;
+
+        public void setTimeout( Integer timeout )
+        {
+            this.timeout = timeout;
+            timeoutPresent = true;
+        }
     }
 }

testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionRun.java

@@ -21,6 +21,7 @@
 import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
 import lombok.Getter;
 import lombok.Setter;
+import neo4j.org.testkit.backend.CustomDriverError;
 import neo4j.org.testkit.backend.TestkitState;
 import neo4j.org.testkit.backend.holder.ResultCursorHolder;
 import neo4j.org.testkit.backend.holder.ResultHolder;
@@ -49,6 +50,30 @@ public class SessionRun implements TestkitRequest
 {
     private SessionRunBody data;
 
+    private void configureTimeout( TransactionConfig.Builder builder )
+    {
+        if ( data.getTimeoutPresent() )
+        {
+            try
+            {
+                if ( data.getTimeout() != null )
+                {
+                    builder.withTimeout( Duration.ofMillis( data.getTimeout() ) );
+                }
+                else
+                {
+                    builder.withTimeout( TransactionConfig.Builder.SERVER_DEFAULT_TIMEOUT );
+                }
+            }
+            catch ( IllegalArgumentException e )
+            {
+                CustomDriverError wrapped = new CustomDriverError();
+                wrapped.initCause( e );
+                throw wrapped;
+            }
+        }
+    }
+
     @Override
     public TestkitResponse process( TestkitState testkitState )
     {
@@ -59,7 +84,7 @@ public TestkitResponse process( TestkitState testkitState )
                               .orElseGet( () -> new Query( data.cypher ) );
         TransactionConfig.Builder transactionConfig = TransactionConfig.builder();
         Optional.ofNullable( data.getTxMeta() ).ifPresent( transactionConfig::withMetadata );
-        Optional.ofNullable( data.getTimeout() ).ifPresent( to -> transactionConfig.withTimeout( Duration.ofMillis( to ) ) );
+        configureTimeout( transactionConfig );
         org.neo4j.driver.Result result = session.run( query, transactionConfig.build() );
         String id = testkitState.addResultHolder( new ResultHolder( sessionHolder, result ) );
 
@@ -78,8 +103,7 @@ public CompletionStage<TestkitResponse> processAsync( TestkitState testkitState
                                                                    .orElseGet( () -> new Query( data.cypher ) );
                                              TransactionConfig.Builder transactionConfig = TransactionConfig.builder();
                                              Optional.ofNullable( data.getTxMeta() ).ifPresent( transactionConfig::withMetadata );
-                                             Optional.ofNullable( data.getTimeout() )
-                                                     .ifPresent( to -> transactionConfig.withTimeout( Duration.ofMillis( to ) ) );
+                                             configureTimeout( transactionConfig );
 
                                              return session.runAsync( query, transactionConfig.build() )
                                                            .thenApply( resultCursor ->
@@ -103,7 +127,7 @@ public Mono<TestkitResponse> processRx( TestkitState testkitState )
                                                                .orElseGet( () -> new Query( data.cypher ) );
                                          TransactionConfig.Builder transactionConfig = TransactionConfig.builder();
                                          Optional.ofNullable( data.getTxMeta() ).ifPresent( transactionConfig::withMetadata );
-                                         Optional.ofNullable( data.getTimeout() ).ifPresent( to -> transactionConfig.withTimeout( Duration.ofMillis( to ) ) );
+                                         configureTimeout( transactionConfig );
 
                                          RxResult result = session.run( query, transactionConfig.build() );
                                          String id = testkitState.addRxResultHolder( new RxResultHolder( sessionHolder, result ) );
@@ -120,17 +144,32 @@ private Result createResponse( String resultId )
         return Result.builder().data( Result.ResultBody.builder().id( resultId ).build() ).build();
     }
 
-    @Setter
-    @Getter
     public static class SessionRunBody
     {
         @JsonDeserialize( using = TestkitCypherParamDeserializer.class )
+        @Setter
+        @Getter
         private Map<String,Object> params;
 
+        @Setter
+        @Getter
         private String sessionId;
+        @Setter
+        @Getter
         private String cypher;
+        @Setter
+        @Getter
         private Map<String,Object> txMeta;
+        @Getter
         private Integer timeout;
+        @Getter
+        private Boolean timeoutPresent = false;
+
+        public void setTimeout( Integer timeout )
+        {
+            this.timeout = timeout;
+            timeoutPresent = true;
+        }
 
     }
 }