yaml tests: more improvements to branching logic (#3197)

(cherry picked from commit 01d1e81)

Author: miguelgrinberg

.buildkite/run-tests

@@ -10,6 +10,9 @@ export TEST_SUITE="${TEST_SUITE:=platinum}"
 export PYTHON_VERSION="${PYTHON_VERSION:=3.14}"
 export PYTHON_CONNECTION_CLASS="${PYTHON_CONNECTION_CLASS:=urllib3}"
 export ES_YAML_TESTS_BRANCH="${BUILDKITE_PULL_REQUEST_BASE_BRANCH}"
+if [[ ! -n "$ES_YAML_TESTS_BRANCH" ]]; then
+    export ES_YAML_TESTS_BRANCH="${BUILDKITE_BRANCH}"
+fi
 
 script_path=$(dirname $(realpath $0))
 source $script_path/functions/imports.sh

test_elasticsearch/test_server/test_rest_api_spec.py

@@ -505,10 +505,10 @@ def remove_implicit_resolver(cls, tag_to_remove):
     if "ES_YAML_TESTS_BRANCH" in os.environ and os.environ["ES_YAML_TESTS_BRANCH"]:
         branch_candidates.append(os.environ["ES_YAML_TESTS_BRANCH"])
     git_branch = subprocess.getoutput("git branch --show-current")
-    if git_branch not in branch_candidates:
+    if git_branch and git_branch not in branch_candidates:
         branch_candidates.append(git_branch)
     package_version = __versionstr__.rsplit(".", 1)[0]
-    if package_version not in branch_candidates:
+    if package_version and package_version not in branch_candidates:
         branch_candidates.append(package_version)
     if "main" not in branch_candidates:
         branch_candidates.append("main")