From c4098fa1a9fda63dd6165d8c28cf73b4c21b0f5e Mon Sep 17 00:00:00 2001
From: Alexey Volkov <avolkov@google.com>
Date: Mon, 26 Aug 2019 17:03:58 -0700
Subject: [PATCH] Test infra - Fixed the batch Prow tests Fixes
 https://github.com/kubeflow/pipelines/issues/1957

---
 test/presubmit-tests-with-pipeline-deployment.sh | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/test/presubmit-tests-with-pipeline-deployment.sh b/test/presubmit-tests-with-pipeline-deployment.sh
index 3282fd99f6a..a10f5cc309e 100755
--- a/test/presubmit-tests-with-pipeline-deployment.sh
+++ b/test/presubmit-tests-with-pipeline-deployment.sh
@@ -63,6 +63,12 @@ while [ "$1" != "" ]; do
     shift
 done
 
+# PULL_PULL_SHA is empty whne Pros/Tide tests the batches.
+# PULL_BASE_SHA cannot be used here as it still points to master tip in that case.
+if [ -z "${PULL_PULL_SHA:-''}" ]; then
+    PULL_PULL_SHA=$(git rev-parse HEAD)
+fi
+
 # Variables
 GCR_IMAGE_BASE_DIR=gcr.io/${PROJECT}/${PULL_PULL_SHA}
 TEST_RESULTS_GCS_DIR=gs://${TEST_RESULT_BUCKET}/${PULL_PULL_SHA}/${TEST_RESULT_FOLDER}