From 8fdf11cc5c541d6bea6066324e8686eecc31484b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Honza=20Kr=C3=A1l?= Date: Sun, 19 Jan 2020 00:02:02 +0000 Subject: [PATCH] make local.py work in tests --- test_elasticsearch/test_server/__init__.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/test_elasticsearch/test_server/__init__.py b/test_elasticsearch/test_server/__init__.py index 53adbe0f..9c116675 100644 --- a/test_elasticsearch/test_server/__init__.py +++ b/test_elasticsearch/test_server/__init__.py @@ -1,7 +1,5 @@ -from elasticsearch.helpers.test import ( - get_test_client, - ElasticsearchTestCase as BaseTestCase, -) +from elasticsearch.helpers import test +from elasticsearch.helpers.test import ElasticsearchTestCase as BaseTestCase client = None @@ -18,7 +16,7 @@ def get_client(**kwargs): new_client = local_get_client(**kwargs) except ImportError: # fallback to using vanilla client - new_client = get_test_client(**kwargs) + new_client = test.get_test_client(**kwargs) if not kwargs: client = new_client