Use one global client instance for all test runs

This commit is contained in:
Honza Kral
2013-08-28 19:24:11 +02:00
parent fe7cc106f6
commit f603a834ca
+8 -5
View File
@@ -84,12 +84,15 @@ def teardown():
server.wait() server.wait()
class ElasticTestCase(TestCase): class ElasticTestCase(TestCase):
client = None
def setUp(self): def setUp(self):
kw = {} if ElasticTestCase.client is None:
if 'TEST_ES_CONNECTION' in os.environ: kw = {}
from elasticsearch import connection if 'TEST_ES_CONNECTION' in os.environ:
kw['connection_class'] = getattr(connection, os.environ['TEST_ES_CONNECTION']) from elasticsearch import connection
self.client = Elasticsearch([os.environ['TEST_ES_SERVER']], **kw) kw['connection_class'] = getattr(connection, os.environ['TEST_ES_CONNECTION'])
ElasticTestCase.client = Elasticsearch([os.environ['TEST_ES_SERVER']], **kw)
self.client = ElasticTestCase.client
def tearDown(self): def tearDown(self):
self.client.indices.delete() self.client.indices.delete()