Longer timeout for the test client to fix travis CI

This commit is contained in:
Honza Král
2015-11-02 16:01:48 +01:00
parent 0e4b68c1aa
commit 328bdb73bc
+3 -2
View File
@@ -9,13 +9,14 @@ except ImportError:
from elasticsearch import Elasticsearch
from elasticsearch.exceptions import ConnectionError
def get_test_client(nowait=False):
def get_test_client(nowait=False, **kwargs):
# construct kwargs from the environment
kw = {}
kw = {'timeout': 30}
if 'TEST_ES_CONNECTION' in os.environ:
from elasticsearch import connection
kw['connection_class'] = getattr(connection, os.environ['TEST_ES_CONNECTION'])
kw.update(kwargs)
client = Elasticsearch([os.environ.get('TEST_ES_SERVER', {})], **kw)
# wait for yellow status