Copy driver to insider testkit container

Author: bigmontz

testkit/backend.py

@@ -3,12 +3,12 @@
 Assumes driver and backend has been built.
 Responsible for starting the test backend.
 """
-import os
+from common import DRIVER_REPO
 import subprocess
 
 
 if __name__ == "__main__":
-    npm = ["npm", "--prefix", "packages/testkit-backend"]
     err = open("/artifacts/backenderr.log", "w")
     out = open("/artifacts/backendout.log", "w")
-    subprocess.check_call([*npm, "start"], stdout=out, stderr=err)
+    subprocess.check_call(["npm", "start"], stdout=out, stderr=err,
+                          cwd=DRIVER_REPO + "packages/testkit-backend")

testkit/build.py

@@ -3,45 +3,37 @@
 Responsible for building driver and test backend.
 """
 import os
-import subprocess
-import shutil
+from common import run, run_in, run_in_driver_repo, is_lite, DRIVER_REPO
 
 
-def run(args, env=None, cwd=None):
-    subprocess.run(
-        args, universal_newlines=True, stderr=subprocess.STDOUT,
-        check=True, env=env, cwd=cwd)
+def copy_files_to_workdir():
+    run(["mkdir", DRIVER_REPO])
+    run(["cp", "-fr", ".", DRIVER_REPO])
+    run(["chown", "-Rh", "driver:driver", DRIVER_REPO])
 
 
-def run_in(cwd):
-    def _runIn(args, env=None):
-        return run(args, env, cwd)
-    return _runIn
+def init_monorepo():
+    run_in_driver_repo(["rm", "-fr", "node_modules"])
+    run_in_driver_repo(["npm", "ci"])
+
 
+def clean_and_build():
+    run_in_driver_repo(["npm", "run", "clean"])
+    run_in_driver_repo(["npm", "run", "build"])
 
-def build_testkit_backend(isLite):
-    run_in_testkit_backend = run_in(cwd="./packages/testkit-backend/")
+
+def build_testkit_backend():
+    run_in_testkit_backend = run_in(
+        cwd=DRIVER_REPO + "packages/testkit-backend/")
     run_in_testkit_backend(["rm", "-fr", "node_modules"])
-    neo4jdriverPath = "neo4j@../neo4j-driver"
-    if isLite:
-        neo4jdriverPath = "neo4j@../neo4j-driver-lite"
+    neo4jdriverPath = "neo4j@../neo4j-driver" if not is_lite()\
+        else "neo4j@../neo4j-driver-lite"
     run_in_testkit_backend(["npm", "install", neo4jdriverPath])
     run_in_testkit_backend(["npm", "install"])
 
 
-def init_monorepo():
-    run(["rm", "-fr", "node_modules"])
-    run(["npm", "ci"])
-
-
-def clean_and_build():
-    run(["npm", "run", "clean"])
-    run(["npm", "run", "build"])
-
-
 if __name__ == "__main__":
-    isLite = os.environ.get("TEST_DRIVER_LITE", False)
-    run(["ls", "-la"], cwd="/driver")
+    copy_files_to_workdir()
     init_monorepo()
     clean_and_build()
-    build_testkit_backend(isLite)
+    build_testkit_backend()

testkit/common.py

@@ -0,0 +1,28 @@
+"""
+Define common commands available for all the scripts
+"""
+import subprocess
+import os
+
+
+DRIVER_REPO = "/home/driver/repo/"
+
+
+def run(args, env=None, cwd=None):
+    subprocess.run(
+        args, universal_newlines=True, stderr=subprocess.STDOUT,
+        check=True, env=env, cwd=cwd)
+
+
+def run_in(cwd):
+    def _runIn(args, env=None):
+        return run(args, env, cwd)
+    return _runIn
+
+
+def run_in_driver_repo(args, env=None):
+    return run(args, env, DRIVER_REPO)
+
+
+def is_lite():
+    return os.environ.get("TEST_DRIVER_LITE", "False").upper() in ["TRUE", "1"]

testkit/integration.py

@@ -1,19 +1,14 @@
-import subprocess
-import os
-
 
-def run(args):
-    subprocess.run(
-        args, universal_newlines=True, 
-        stderr=subprocess.STDOUT, check=True)
+import os
+from common import run_in_driver_repo, is_lite
 
 
 if __name__ == "__main__":
     os.environ["TEST_NEO4J_IPV6_ENABLED"] = "False"
 
-    if os.environ.get("TEST_DRIVER_LITE", False):
+    if is_lite():
         ignore = "--ignore=neo4j-driver"
     else:
         ignore = "--ignore=neo4j-driver-lite"
 
-    run(["npm", "run", "test::integration", "--", ignore])
+    run_in_driver_repo(["npm", "run", "test::integration", "--", ignore])

testkit/stress.py

@@ -1,20 +1,15 @@
-import subprocess
 import os
-
-
-def run(args, cwd=None):
-    subprocess.run(
-        args, universal_newlines=True, stderr=subprocess.STDOUT, check=True, cwd=cwd)
+from common import run_in_driver_repo, is_lite
 
 
 if __name__ == "__main__":
     os.environ["STRESS_TEST_MODE"] = "fastest"
     os.environ["RUNNING_TIME_IN_SECONDS"] = \
         os.environ.get("TEST_NEO4J_STRESS_DURATION", 0)
 
-    if os.environ.get("TEST_DRIVER_LITE", False):
+    if is_lite():
         ignore = "--ignore=neo4j-driver"
     else:
         ignore = "--ignore=neo4j-driver-lite"
 
-    run(["npm", "run", "test::stress", "--", ignore])
+    run_in_driver_repo(["npm", "run", "test::stress", "--", ignore])

testkit/unittests.py

@@ -3,19 +3,14 @@
 Responsible for running unit tests.
 Assumes driver has been setup by build script prior to this.
 """
-import subprocess
 import os
-
-
-def run(args, cwd=None):
-    subprocess.run(
-        args, universal_newlines=True, stderr=subprocess.STDOUT, check=True, cwd=cwd)
+from common import run_in_driver_repo, is_lite
 
 
 if __name__ == "__main__":
-    if os.environ.get("TEST_DRIVER_LITE", False):
+    if is_lite():
         ignore = "--ignore=neo4j-driver"
     else:
         ignore = "--ignore=neo4j-driver-lite"
 
-    run(["npm", "run", "test::unit", "--", ignore])
+    run_in_driver_repo(["npm", "run", "test::unit", "--", ignore])