[experimental] tests

Author: injectives

driver/src/test/java/org/neo4j/driver/stress/CausalClusteringIT.java

@@ -334,7 +334,7 @@ void shouldHandleGracefulLeaderSwitch() throws Exception
 
         try ( Driver driver = createDriver( leader.getRoutingUri() ) )
         {
-            Session session1 = driver.session();
+            Session session1 = driver.session( builder().withDatabase( "neo4j" ).build() );
             Transaction tx1 = session1.beginTransaction();
 
             // gracefully stop current leader to force re-election
@@ -346,7 +346,7 @@ void shouldHandleGracefulLeaderSwitch() throws Exception
 
             session1.close();
 
-            Bookmark bookmark = inExpirableSession( driver, Driver::session, session ->
+            Bookmark bookmark = inExpirableSession( driver, d -> d.session( builder().withDatabase( "neo4j" ).build() ), session ->
             {
                 try ( Transaction tx = session.beginTransaction() )
                 {
@@ -357,7 +357,8 @@ void shouldHandleGracefulLeaderSwitch() throws Exception
                 return session.lastBookmark();
             } );
 
-            try ( Session session2 = driver.session( builder().withDefaultAccessMode( AccessMode.READ ).withBookmarks( bookmark ).build() );
+            try ( Session session2 = driver.session(
+                    builder().withDatabase( "neo4j" ).withDefaultAccessMode( AccessMode.READ ).withBookmarks( bookmark ).build() );
                   Transaction tx2 = session2.beginTransaction() )
             {
                 Record record = tx2.run( "MATCH (n:Person) RETURN COUNT(*) AS count" ).next();
@@ -538,20 +539,20 @@ void shouldRespectMaxConnectionPoolSizePerClusterMember()
 
         try ( Driver driver = createDriver( leader.getRoutingUri(), config ) )
         {
-            Session writeSession1 = driver.session( builder().withDefaultAccessMode( AccessMode.WRITE ).build() );
+            Session writeSession1 = driver.session( builder().withDatabase( "neo4j" ).withDefaultAccessMode( AccessMode.WRITE ).build() );
             writeSession1.beginTransaction();
 
-            Session writeSession2 = driver.session( builder().withDefaultAccessMode( AccessMode.WRITE ).build() );
+            Session writeSession2 = driver.session( builder().withDatabase( "neo4j" ).withDefaultAccessMode( AccessMode.WRITE ).build() );
             writeSession2.beginTransaction();
 
             // should not be possible to acquire more connections towards leader because limit is 2
-            Session writeSession3 = driver.session( builder().withDefaultAccessMode( AccessMode.WRITE ).build() );
+            Session writeSession3 = driver.session( builder().withDatabase( "neo4j" ).withDefaultAccessMode( AccessMode.WRITE ).build() );
             ClientException e = assertThrows( ClientException.class, writeSession3::beginTransaction );
             assertThat( e, is( connectionAcquisitionTimeoutError( 42 ) ) );
 
             // should be possible to acquire new connection towards read server
             // it's a different machine, not leader, so different max connection pool size limit applies
-            Session readSession = driver.session( builder().withDefaultAccessMode( AccessMode.READ ).build() );
+            Session readSession = driver.session( builder().withDatabase( "neo4j" ).withDefaultAccessMode( AccessMode.READ ).build() );
             Record record = readSession.readTransaction( tx -> tx.run( "RETURN 1" ).single() );
             assertEquals( 1, record.get( 0 ).asInt() );
         }
@@ -610,7 +611,7 @@ void shouldRediscoverWhenConnectionsToAllCoresBreak()
         try ( Driver driver = driverFactory.newInstance( leader.getRoutingUri(), clusterRule.getDefaultAuthToken(),
                                                          RoutingSettings.DEFAULT, RetrySettings.DEFAULT, configWithoutLogging(), SecurityPlanImpl.insecure() ) )
         {
-            try ( Session session = driver.session() )
+            try ( Session session = driver.session( builder().withDatabase( "neo4j" ).build() ) )
             {
                 createNode( session, "Person", "name", "Vision" );
 
@@ -626,18 +627,18 @@ RoutingSettings.DEFAULT, RetrySettings.DEFAULT, configWithoutLogging(), Security
             makeAllChannelsFailToRunQueries( driverFactory, ServerVersion.version( driver ) );
 
             // observe that connection towards writer is broken
-            try ( Session session = driver.session( builder().withDefaultAccessMode( AccessMode.WRITE ).build() ) )
+            try ( Session session = driver.session( builder().withDatabase( "neo4j" ).withDefaultAccessMode( AccessMode.WRITE ).build() ) )
             {
                 SessionExpiredException e = assertThrows( SessionExpiredException.class,
-                        () -> runCreateNode( session, "Person", "name", "Vision" ).consume() );
+                                                          () -> runCreateNode( session, "Person", "name", "Vision" ).consume() );
                 assertEquals( "Disconnected", e.getCause().getMessage() );
             }
 
             // probe connections to all readers
             int readersCount = cluster.followers().size() + cluster.readReplicas().size();
             for ( int i = 0; i < readersCount; i++ )
             {
-                try ( Session session = driver.session( builder().withDefaultAccessMode( AccessMode.READ ).build() ) )
+                try ( Session session = driver.session( builder().withDatabase( "neo4j" ).withDefaultAccessMode( AccessMode.READ ).build() ) )
                 {
                     runCountNodes( session, "Person", "name", "Vision" );
                 }
@@ -646,7 +647,7 @@ RoutingSettings.DEFAULT, RetrySettings.DEFAULT, configWithoutLogging(), Security
                 }
             }
 
-            try ( Session session = driver.session() )
+            try ( Session session = driver.session( builder().withDatabase( "neo4j" ).build() ) )
             {
                 updateNode( session, "Person", "name", "Vision", "Thanos" );
                 assertEquals( 0, countNodes( session, "Person", "name", "Vision" ) );