Improve TestKit glue

 * add option to choose interpreter version
 * make Python invocations more strict (warnings as errors)
 * refactor and simplify code

Author: robsdedude

testkit/Dockerfile

@@ -44,9 +44,10 @@ RUN pyenv install 3.6.13
 RUN pyenv rehash
 RUN pyenv global 3.6.13
 
-# Install Latest pip for each environment
+# Install Latest pip and setuptools for each environment
 # https://pip.pypa.io/en/stable/news/
 RUN python -m pip install --upgrade pip
+RUN python -m pip install --upgrade setuptools
 
 # Install Python Testing Tools
 RUN python -m pip install coverage tox

testkit/_common.py

@@ -0,0 +1,17 @@
+import os
+import subprocess
+import sys
+
+
+TEST_BACKEND_VERSION = os.getenv("TEST_BACKEND_VERSION", "python")
+
+
+def run(args, env=None):
+    return subprocess.run(
+        args, universal_newlines=True, stdout=sys.stdout, stderr=sys.stderr,
+        check=True, env=env
+    )
+
+
+def run_python(args, env=None):
+    run([TEST_BACKEND_VERSION, "-W", "error", *args], env=env)

testkit/backend.py

@@ -1,8 +1,25 @@
-import subprocess
-import sys
+#!/usr/bin/env python
+
+# Copyright (c) "Neo4j"
+# Neo4j Sweden AB [https://neo4j.com]
+#
+# This file is part of Neo4j.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from _common import run_python
+
 
 if __name__ == "__main__":
-    subprocess.check_call(
-        ["python", "-W", "error", "-m", "testkitbackend"],
-        stdout=sys.stdout, stderr=sys.stderr
-    )
+    run_python(["-m", "testkitbackend"])

testkit/build.py

@@ -4,17 +4,11 @@
 """
 
 
-import subprocess
-import sys
-
-
-def run(args, env=None):
-    subprocess.run(args, universal_newlines=True, stdout=sys.stdout,
-                   stderr=sys.stderr, check=True, env=env)
+from _common import run_python
 
 
 if __name__ == "__main__":
-    run(["python", "setup.py", "build"])
-    run(["python", "-m", "pip", "install", "-U", "pip"])
-    run(["python", "-m", "pip", "install", "-Ur",
-         "testkitbackend/requirements.txt"])
+    run_python(["setup.py", "build"])
+    run_python(["-m", "pip", "install", "-U", "pip"])
+    run_python(["-m", "pip", "install", "-Ur",
+                "testkitbackend/requirements.txt"])

testkit/integration.py

@@ -1,2 +1,22 @@
+#!/usr/bin/env python
+
+# Copyright (c) "Neo4j"
+# Neo4j Sweden AB [https://neo4j.com]
+#
+# This file is part of Neo4j.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
 if __name__ == "__main__":
     pass

testkit/unittests.py

@@ -1,14 +1,25 @@
-import subprocess
-import sys
+#!/usr/bin/env python
 
+# Copyright (c) "Neo4j"
+# Neo4j Sweden AB [https://neo4j.com]
+#
+# This file is part of Neo4j.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
-def run(args):
-    subprocess.run(
-        args, universal_newlines=True, stdout=sys.stdout, stderr=sys.stderr,
-        check=True
-    )
+
+from _common import run_python
 
 
 if __name__ == "__main__":
-    run([
-        "python", "-m", "tox", "-c", "tox-unit.ini"])
+    run_python(["-m", "tox", "-c", "tox-unit.ini"])