Adjust unit tests

neo4j · Jan 12, 2022 · 49b203f · 49b203f
1 parent 5e57c9e
commit 49b203f
Showing 1 changed file with 20 additions and 12 deletions.
diff --git a/driver/src/test/java/org/neo4j/driver/TransactionConfigTest.java b/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()
     {