actions/opensearch-pyd
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Support for running different connection class in tests

Browse Source
This commit is contained in:
Honza Kral
2013-08-27 02:53:01 +02:00
parent 966128a5c8
commit 610226e91c
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_elasticsearch/test_server/__init__.py
+5 -1
View File
@@ -85,7 +85,11 @@ def teardown():
class ElasticTestCase(TestCase): class ElasticTestCase(TestCase):
def setUp(self): def setUp(self):
self.client = Elasticsearch([os.environ['TEST_ES_SERVER']]) kw = {}
if 'TEST_ES_CONNECTION' in os.environ:
from elasticsearch import connection
kw['connection_class'] = getattr(connection, os.environ['TEST_ES_CONNECTION'])
self.client = Elasticsearch([os.environ['TEST_ES_SERVER']], **kw)
def tearDown(self): def tearDown(self):
self.client.indices.delete() self.client.indices.delete()