From 610226e91cdf64c074b7da7ff8d4d49379324938 Mon Sep 17 00:00:00 2001 From: Honza Kral Date: Tue, 27 Aug 2013 02:53:01 +0200 Subject: [PATCH] Support for running different connection class in tests --- test_elasticsearch/test_server/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/test_elasticsearch/test_server/__init__.py b/test_elasticsearch/test_server/__init__.py index 80bcfd37..eea524bd 100644 --- a/test_elasticsearch/test_server/__init__.py +++ b/test_elasticsearch/test_server/__init__.py @@ -85,7 +85,11 @@ def teardown(): class ElasticTestCase(TestCase): 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): self.client.indices.delete()