fix testkit build

Author: bigmontz

testkit/build.py

@@ -50,11 +50,10 @@ def build_testkit_backend(isLite):
 
 
 if __name__ == "__main__":
-    isLite = os.environ["TEST_DRIVER_LITE"]
+    isLite = os.environ.get("TEST_DRIVER_LITE", False)
     build_core()
     build_bolt_connection()
     if isLite:
         build_driver_lite()
-    else:
-        build_driver()
+    build_driver()
     build_testkit_backend(isLite)

testkit/integration.py

@@ -18,7 +18,7 @@ def test_driver_lite():
 
 if __name__ == "__main__":
     os.environ["TEST_NEO4J_IPV6_ENABLED"] = "False"
-    if os.environ["TEST_DRIVER_LITE"]:
+    if os.environ.get("TEST_DRIVER_LITE", False):
         test_driver_lite()
     else:
         test_driver()

testkit/stress.py

@@ -19,7 +19,7 @@ def test_driver_lite():
     os.environ['STRESS_TEST_MODE'] = 'fastest'
     os.environ['RUNNING_TIME_IN_SECONDS'] = \
         os.environ.get('TEST_NEO4J_STRESS_DURATION', 0)
-    if os.environ["TEST_DRIVER_LITE"]:
+    if os.environ.get("TEST_DRIVER_LITE", False):
         test_driver_lite()
     else:
         test_driver()

testkit/unittests.py

@@ -25,7 +25,7 @@ def test_driver_lite():
 if __name__ == "__main__":
     run(['npm', '--prefix', 'core', 'run', 'test'])
     run(['npm', '--prefix', 'bolt-connection', 'run', 'test'])
-    if os.environ["TEST_DRIVER_LITE"]:
+    if os.environ.get("TEST_DRIVER_LITE", False):
         test_driver_lite()
     else:
         test_driver()