Honza Kral
81 lines
1.8 KiB
Python
81 lines
1.8 KiB
Python
import time
|
|
import subprocess
|
|
import tempfile
|
|
import os
|
|
|
|
import requests
|
|
|
|
from unittest import SkipTest
|
|
|
|
data_dir = None
|
|
|
|
CMD = """
|
|
elasticsearch \
|
|
-f \
|
|
-D es.cluster.name=%(cluster_name)s \
|
|
-D es.node.name=test_name \
|
|
-D es.http.port=%(port)s \
|
|
-D es.gateway.type=none \
|
|
-D es.index.store.type=memory \
|
|
-D es.discovery.zen.ping.multicast.enabled=false \
|
|
-D es.path.data=%(data_dir)s \
|
|
-D es.pidfile=%(pidfile)s \
|
|
>/dev/null 2>&1
|
|
"""
|
|
|
|
server = None
|
|
pidfile = tempfile.mktemp()
|
|
|
|
def setup():
|
|
if 'YAML_TEST_DIR' not in os.environ:
|
|
raise SkipTest('')
|
|
global server
|
|
|
|
# check installed
|
|
if subprocess.call('which vim >/dev/null 2>&1', shell=True) != 0:
|
|
raise SkipTest("No Elasticsearch server, skipping integration tests.")
|
|
|
|
args = {
|
|
'cluster_name': 'es_client_test',
|
|
'port': 9900,
|
|
'data_dir': tempfile.tempdir,
|
|
'pidfile': pidfile
|
|
}
|
|
|
|
# check running
|
|
try:
|
|
requests.get('http://localhost:%(port)s' % args)
|
|
except requests.ConnectionError:
|
|
pass
|
|
else:
|
|
raise SkipTest('Elasticsearch already running!')
|
|
|
|
|
|
cmd = CMD % args
|
|
|
|
server = subprocess.Popen(cmd, shell=True)
|
|
|
|
# wait for green status
|
|
for _ in range(100):
|
|
response = None
|
|
time.sleep(.1)
|
|
try:
|
|
response = requests.get('http://localhost:%(port)s/_cluster/health?wait_for_status=green' % args)
|
|
except requests.ConnectionError:
|
|
continue
|
|
|
|
if response.status_code == 200:
|
|
break
|
|
|
|
else:
|
|
# timeout
|
|
raise SkipTest("Elasticsearch failed to start.")
|
|
|
|
|
|
def teardown():
|
|
if server is not None:
|
|
with open(pidfile) as pidf:
|
|
pid = pidf.read()
|
|
os.kill(int(pid), 15)
|
|
server.wait()
|