Merge pull request #691 from zhenlineo/1.7-test-with-remote-server

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

Author: zhenlineo

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

@@ -99,14 +99,12 @@ void setUp()
         System.setProperty( DRIVER_METRICS_ENABLED_KEY, "true" );
         logging = new LoggerNameTrackingLogging();
 
-        Config config = Config.builder()
-                .withoutEncryption()
+        Config.ConfigBuilder builder = Config.builder()
                 .withLogging( logging )
                 .withMaxConnectionPoolSize( 100 )
-                .withConnectionAcquisitionTimeout( 1, MINUTES )
-                .build();
+                .withConnectionAcquisitionTimeout( 1, MINUTES );
 
-        driver = (InternalDriver) GraphDatabase.driver( databaseUri(), authToken(), config );
+        driver = (InternalDriver) GraphDatabase.driver( databaseUri(), authToken(), config( builder ) );
         System.setProperty( DRIVER_METRICS_ENABLED_KEY, "false" );
 
         ThreadFactory threadFactory = new DaemonThreadFactory( getClass().getSimpleName() + "-worker-" );
@@ -186,6 +184,8 @@ private void runStressTest( Function<C,List<Future<?>>> threadLauncher ) throws
 
     abstract AuthToken authToken();
 
+    abstract Config config( Config.ConfigBuilder builder );
+
     abstract C createContext();
 
     abstract List<BlockingCommand<C>> createTestSpecificBlockingCommands();

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

@@ -35,6 +35,7 @@
 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.exceptions.SessionExpiredException;
 import org.neo4j.driver.v1.summary.ResultSummary;
@@ -66,6 +67,12 @@ AuthToken authToken()
         return clusterRule.getAuthToken();
     }
 
+    @Override
+    Config config( Config.ConfigBuilder builder )
+    {
+        return clusterRule.getDriverConfig( builder );
+    }
+
     @Override
     Context createContext()
     {

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

@@ -26,6 +26,7 @@
 import java.util.concurrent.atomic.AtomicLong;
 
 import org.neo4j.driver.v1.AuthToken;
+import org.neo4j.driver.v1.Config;
 import org.neo4j.driver.v1.summary.ResultSummary;
 import org.neo4j.driver.v1.util.DatabaseExtension;
 import org.neo4j.driver.v1.util.ParallelizableIT;
@@ -52,6 +53,12 @@ AuthToken authToken()
         return neo4j.authToken();
     }
 
+    @Override
+    Config config( Config.ConfigBuilder builder )
+    {
+        return builder.build();
+    }
+
     @Override
     Context createContext()
     {

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

@@ -28,6 +28,7 @@
 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.util.TestUtil;
 
@@ -62,6 +63,20 @@ public AuthToken getAuthToken()
         return localClusterExtension.getDefaultAuthToken();
     }
 
+    public Config getDriverConfig( Config.ConfigBuilder builder )
+    {
+        if ( remoteClusterExists() )
+        {
+            builder.withEncryption();
+        }
+        else
+        {
+            builder.withoutEncryption();
+        }
+
+        return builder.build();
+    }
+
     @Override
     public void beforeAll( ExtensionContext context ) throws Exception
     {