Make shell tests run on latest release too.

Fixes #1448.

--
MOS_MIGRATED_REVID=125943283

1 files changed, 11 insertions, 0 deletions

diff --git a/src/test/shell/bazel/testenv.sh b/src/test/shell/bazel/testenv.sh
index 43c14fd4a9..f51954248b 100755
--- a/src/test/shell/bazel/testenv.sh
+++ b/src/test/shell/bazel/testenv.sh
@@ -20,6 +20,17 @@
 [ -z "$TEST_SRCDIR" ] && { echo "TEST_SRCDIR not set!" >&2; exit 1; }
 BAZEL_RUNFILES="$TEST_SRCDIR/io_bazel"
 
+if [ ! type rlocation &> /dev/null ]; then
+  function rlocation() {
+    if [[ "$1" = /* ]]; then
+      echo $1
+    else
+      echo "$TEST_SRCDIR/$1"
+    fi
+  }
+  export -f rlocation
+fi
+
 # Load the unit-testing framework
 source "$(rlocation io_bazel/src/test/shell/unittest.bash)" || \
   { echo "Failed to source unittest.bash" >&2; exit 1; }