98 lines
2.8 KiB
Python
Executable File
98 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# Licensed to Elasticsearch B.V under one or more agreements.
|
|
# Elasticsearch B.V licenses this file to you under the Apache 2.0 License.
|
|
# See the LICENSE file in the project root for more information
|
|
|
|
from __future__ import print_function
|
|
|
|
import sys
|
|
from os import environ
|
|
from os.path import dirname, join, pardir, abspath, exists
|
|
import subprocess
|
|
|
|
|
|
def fetch_es_repo():
|
|
# user is manually setting YAML dir, don't tamper with it
|
|
if "TEST_ES_YAML_DIR" in environ:
|
|
return
|
|
|
|
repo_path = environ.get(
|
|
"TEST_ES_REPO",
|
|
abspath(join(dirname(__file__), pardir, pardir, "elasticsearch")),
|
|
)
|
|
|
|
# no repo
|
|
if not exists(repo_path) or not exists(join(repo_path, ".git")):
|
|
subprocess.check_call(
|
|
"git clone https://github.com/elastic/elasticsearch %s" % repo_path,
|
|
shell=True,
|
|
)
|
|
|
|
# set YAML test dir
|
|
environ["TEST_ES_YAML_DIR"] = join(
|
|
repo_path, "rest-api-spec", "src", "main", "resources", "rest-api-spec", "test"
|
|
)
|
|
|
|
# fetching of yaml tests disabled, we'll run with what's there
|
|
if environ.get("TEST_ES_NOFETCH", False):
|
|
return
|
|
|
|
from test_elasticsearch.test_server import get_client
|
|
from test_elasticsearch.test_cases import SkipTest
|
|
|
|
# find out the sha of the running es
|
|
try:
|
|
es = get_client()
|
|
sha = es.info()["version"]["build_hash"]
|
|
except (SkipTest, KeyError):
|
|
print("No running elasticsearch >1.X server...")
|
|
return
|
|
|
|
# fetch new commits to be sure...
|
|
print("Fetching elasticsearch repo...")
|
|
subprocess.check_call(
|
|
"cd %s && git fetch https://github.com/elasticsearch/elasticsearch.git"
|
|
% repo_path,
|
|
shell=True,
|
|
)
|
|
# reset to the version from info()
|
|
subprocess.check_call("cd %s && git fetch" % repo_path, shell=True)
|
|
subprocess.check_call("cd %s && git reset --hard %s" % (repo_path, sha), shell=True)
|
|
|
|
|
|
def run_all(argv=None):
|
|
sys.exitfunc = lambda: sys.stderr.write("Shutting down....\n")
|
|
|
|
# fetch yaml tests
|
|
fetch_es_repo()
|
|
|
|
# always insert coverage when running tests
|
|
if argv is None:
|
|
junit_xml = join(
|
|
abspath(dirname(dirname(__file__))), "junit", "elasticsearch-py-junit.xml"
|
|
)
|
|
argv = [
|
|
"pytest",
|
|
"--cov=elasticsearch",
|
|
"--junitxml=%s" % junit_xml,
|
|
"--log-level=DEBUG",
|
|
"--cache-clear",
|
|
"-vv",
|
|
]
|
|
|
|
if sys.version_info < (3, 6):
|
|
argv.append("--ignore=test_elasticsearch/test_async/")
|
|
|
|
argv.append(abspath(dirname(__file__)),)
|
|
|
|
exit_code = 0
|
|
try:
|
|
subprocess.check_call(argv, stdout=sys.stdout, stderr=sys.stderr)
|
|
except subprocess.CalledProcessError as e:
|
|
exit_code = e.returncode
|
|
sys.exit(exit_code)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_all(sys.argv)
|