Use GITHUB_REF when available

achamayou · Jun 28, 2024 · 7888b70 · 7888b70
1 parent cd70520
commit 7888b70
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/infra/github.py b/tests/infra/github.py
@@ -197,12 +197,12 @@ def local_branch():
         if tag_or_none:
             return tag_or_none
 
-        # ADO
+        # ADO or GHA
         short = None
         if "SYSTEM_PULLREQUEST_SOURCEBRANCH" in os.environ:
             short = os.environ["SYSTEM_PULLREQUEST_SOURCEBRANCH"]
         else:
-            ref = os.environ["BUILD_SOURCEBRANCH"]
+            ref = os.environ.get("BUILD_SOURCEBRANCH") or os.environ.get("GITHUB_REF")
             for prefix in ["refs/heads/", "refs/tags/", "refs/pull/"]:
                 if ref.startswith(prefix):
                     short = ref[len(prefix) :]