Enable encryption for cluster stress test if test with a remote cluster.

Author: Zhen Li

driver/src/test/java/org/neo4j/driver/v1/stress/AbstractStressTestBase.java

@@ -67,7 +67,6 @@
 import static java.util.Collections.nCopies;
 import static java.util.Collections.singletonMap;
 import static java.util.concurrent.CompletableFuture.completedFuture;
-import static java.util.concurrent.TimeUnit.MINUTES;
 import static java.util.concurrent.TimeUnit.NANOSECONDS;
 import static java.util.concurrent.TimeUnit.SECONDS;
 import static org.hamcrest.Matchers.equalTo;
@@ -99,14 +98,7 @@ void setUp()
         System.setProperty( DRIVER_METRICS_ENABLED_KEY, "true" );
         logging = new LoggerNameTrackingLogging();
 
-        Config config = Config.builder()
-                .withoutEncryption()
-                .withLogging( logging )
-                .withMaxConnectionPoolSize( 100 )
-                .withConnectionAcquisitionTimeout( 1, MINUTES )
-                .build();
-
-        driver = (InternalDriver) GraphDatabase.driver( databaseUri(), authToken(), config );
+        driver = (InternalDriver) GraphDatabase.driver( databaseUri(), authToken(), config( logging ) );
         System.setProperty( DRIVER_METRICS_ENABLED_KEY, "false" );
 
         ThreadFactory threadFactory = new DaemonThreadFactory( getClass().getSimpleName() + "-worker-" );
@@ -186,6 +178,8 @@ private void runStressTest( Function<C,List<Future<?>>> threadLauncher ) throws
 
     abstract AuthToken authToken();
 
+    abstract Config config( Logging logging );
+
     abstract C createContext();
 
     abstract List<BlockingCommand<C>> createTestSpecificBlockingCommands();

driver/src/test/java/org/neo4j/driver/v1/stress/CausalClusteringStressIT.java

@@ -35,7 +35,9 @@
 import org.neo4j.driver.internal.BoltServerAddress;
 import org.neo4j.driver.internal.util.ServerVersion;
 import org.neo4j.driver.v1.AuthToken;
+import org.neo4j.driver.v1.Config;
 import org.neo4j.driver.v1.Driver;
+import org.neo4j.driver.v1.Logging;
 import org.neo4j.driver.v1.exceptions.SessionExpiredException;
 import org.neo4j.driver.v1.summary.ResultSummary;
 import org.neo4j.driver.v1.util.cc.ClusterMemberRole;
@@ -66,6 +68,12 @@ AuthToken authToken()
         return clusterRule.getAuthToken();
     }
 
+    @Override
+    Config config( Logging logging )
+    {
+        return clusterRule.getDriverConfig( logging );
+    }
+
     @Override
     Context createContext()
     {

driver/src/test/java/org/neo4j/driver/v1/util/cc/LocalOrRemoteClusterExtension.java

@@ -28,9 +28,12 @@
 import org.neo4j.driver.internal.util.DriverFactoryWithOneEventLoopThread;
 import org.neo4j.driver.v1.AuthToken;
 import org.neo4j.driver.v1.AuthTokens;
+import org.neo4j.driver.v1.Config;
 import org.neo4j.driver.v1.Driver;
+import org.neo4j.driver.v1.Logging;
 import org.neo4j.driver.v1.util.TestUtil;
 
+import static java.util.concurrent.TimeUnit.MINUTES;
 import static org.neo4j.driver.internal.DriverFactory.BOLT_ROUTING_URI_SCHEME;
 import static org.neo4j.driver.internal.DriverFactory.NEO4J_URI_SCHEME;
 import static org.neo4j.driver.v1.Config.defaultConfig;
@@ -62,6 +65,26 @@ public AuthToken getAuthToken()
         return localClusterExtension.getDefaultAuthToken();
     }
 
+
+    public Config getDriverConfig( Logging logging )
+    {
+        Config.ConfigBuilder builder = Config.builder()
+                .withLogging( logging )
+                .withMaxConnectionPoolSize( 100 )
+                .withConnectionAcquisitionTimeout( 1, MINUTES );
+
+        if ( remoteClusterExists() )
+        {
+            builder.withEncryption();
+        }
+        else
+        {
+            builder.withoutEncryption();
+        }
+
+        return builder.build();
+    }
+
     @Override
     public void beforeAll( ExtensionContext context ) throws Exception
     {