fix: updated .kokoro/build.sh for nox usage

mf2199 · mf2199 · commit 43b186ee4d8f · 2020-08-25T19:26:19.000-04:00
diff --git a/.kokoro/build.sh b/.kokoro/build.sh
@@ -29,13 +29,20 @@ export GOOGLE_APPLICATION_CREDENTIALS=${KOKORO_GFILE_DIR}/service-account.json
 # Setup project id.
 export PROJECT_ID=$(cat "${KOKORO_GFILE_DIR}/project-id.json")
 
-# Install tox
-python3.6 -m pip install --upgrade --quiet tox flake8 isort
-python3.6 -m tox --version
+# Remove old nox
+python3.6 -m pip uninstall --yes --quiet nox-automation
 
-python3.6 -m tox
-python3.6 -m isort --recursive --check-only --diff
-python3.6 -m flake8
+# Install nox
+python3.6 -m pip install --upgrade --quiet nox
+python3.6 -m nox --version
+
+# If NOX_SESSION is set, it only runs the specified session,
+# otherwise run all the sessions.
+if [[ -n "${NOX_SESSION:-}" ]]; then
+    python3.6 -m nox -s "${NOX_SESSION:-}"
+else
+    python3.6 -m nox
+fi
 
 # Export essential environment variables for Django tests.
 export RUNNING_SPANNER_BACKEND_TESTS=1
