add test

Author: rickeylev

tests/py_zipapp/BUILD.bazel

@@ -98,6 +98,17 @@ sh_test(
     toolchains = ["//python:current_py_toolchain"],
 )
 
+sh_test(
+    name = "extract_root_test",
+    srcs = ["extract_root_test.sh"],
+    data = [
+        ":system_python_zipapp",
+    ],
+    env = {
+        "ZIPAPP": "$(rootpath :system_python_zipapp)",
+    },
+)
+
 py_library(
     name = "some_dep",
     srcs = ["some_dep.py"],

tests/py_zipapp/extract_root_test.sh

@@ -0,0 +1,21 @@
+#!/usr/bin/env bash
+# Verify that the RULES_PYTHON_EXTRACT_ROOT env variable is respected.
+
+set -euo pipefail
+
+export RULES_PYTHON_EXTRACT_ROOT="${TEST_TMPDIR:-/tmp}/extract_root_test"
+
+echo "Running zipapp the first time..."
+"$ZIPAPP"
+
+# Verify that the directory was created
+if [[ ! -d "$RULES_PYTHON_EXTRACT_ROOT" ]]; then
+  echo "Error: Extract root directory $RULES_PYTHON_EXTRACT_ROOT was not created!"
+  exit 1
+fi
+
+# Run a second time to ensure it can re-extract successfully.
+echo "Running zipapp the second time..."
+"$ZIPAPP"
+
+echo "Success!"