Stress startable from testkit

Needed check on encryption on/off in test code by probing the scheme.

Author: Peter Wilhelmsson

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

@@ -63,7 +63,7 @@ public AuthToken getAuthToken()
 
     public Config.ConfigBuilder config( Config.ConfigBuilder builder )
     {
-        if ( remoteClusterExists() )
+        if ( useEncryption() )
         {
             builder.withEncryption();
         }
@@ -153,6 +153,20 @@ private static boolean remoteClusterExists()
         return remoteClusterUriFromSystemProperty() != null;
     }
 
+    private static boolean useEncryption()
+    {
+        URI uri = remoteClusterUriFromSystemProperty();
+        if (uri == null) {
+            return false;
+        }
+        String scheme = uri.getScheme();
+        return
+            scheme == "neo4j+ssc" ||
+            scheme == "neo4j+s" ||
+            scheme == "bolt+ssc" ||
+            scheme == "bolt+s";
+    }
+
     private static URI remoteClusterUriFromSystemProperty()
     {
         String uri = System.getProperty( CLUSTER_URI_SYSTEM_PROPERTY_NAME );

testkit/build.py

@@ -2,12 +2,13 @@
 Executed in java driver container.
 Responsible for building driver and test backend.
 """
-import os, subprocess
+import subprocess
+
 
 def run(args):
     subprocess.run(
         args, universal_newlines=True, stderr=subprocess.STDOUT, check=True)
 
+
 if __name__ == "__main__":
     run(["mvn", "clean", "install", "-P", "!determine-revision", "-DskipTests"])
-

testkit/stress.py

@@ -1,4 +1,27 @@
+"""
+Executed in java driver container.
+Responsible for invoking Java stress test suite.
+The stress test might be invoked multiple times against different versions
+of Neo4j.
+Assumes driver has been built before.
+"""
+import subprocess
+import os
 
 if __name__ == "__main__":
-    print("Stress tests not ported to testkit")
-
+    uri = "%s://%s:%s" % (
+            os.environ["TEST_NEO4J_SCHEME"],
+            os.environ["TEST_NEO4J_HOST"],
+            os.environ["TEST_NEO4J_PORT"])
+    password = os.environ["TEST_NEO4J_PASS"]
+    cmd = [
+            "mvn", "surefire:test",
+            "-Dtest=CausalClusteringStressIT,AbstractStressTestBase",
+            "-DexternalClusterUri=%s" % uri,
+            "-Dneo4jUserPassword=%s" % password,
+            "-DthreadCount=10",
+            "-DexecutionTimeSeconds=10",
+            "-Dmaven.gitcommitid.skip=true",
+    ]
+    subprocess.run(cmd, universal_newlines=True,
+                   stderr=subprocess.STDOUT, check=True, cwd="driver")

testkit/unittests.py

@@ -3,12 +3,13 @@
 Responsible for running unit tests.
 Assumes driver has been setup by build script prior to this.
 """
+import subprocess
 
-import os, subprocess
 
 def run(args):
     subprocess.run(
         args, universal_newlines=True, stderr=subprocess.STDOUT, check=True)
 
+
 if __name__ == "__main__":
     run(["mvn", "test", "-Dmaven.gitcommitid.skip"])