1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
import os
import unittest
from django.conf import settings
from haystack.utils import log as logging
def load_tests(loader, standard_tests, pattern):
log = logging.getLogger("haystack")
try:
import elasticsearch
if not ((1, 0, 0) <= elasticsearch.__version__ < (2, 0, 0)):
raise ImportError
from elasticsearch import Elasticsearch, ElasticsearchException
except ImportError:
log.error(
"Skipping ElasticSearch 1 tests: 'elasticsearch>=1.0.0,<2.0.0' not installed."
)
raise unittest.SkipTest("'elasticsearch>=1.0.0,<2.0.0' not installed.")
es = Elasticsearch(settings.HAYSTACK_CONNECTIONS["elasticsearch"]["URL"])
try:
es.info()
except ElasticsearchException as e:
log.error(
"elasticsearch not running on %r"
% settings.HAYSTACK_CONNECTIONS["elasticsearch"]["URL"],
exc_info=True,
)
raise unittest.SkipTest(
"elasticsearch not running on %r"
% settings.HAYSTACK_CONNECTIONS["elasticsearch"]["URL"],
e,
)
package_tests = loader.discover(
start_dir=os.path.dirname(__file__), pattern=pattern
)
standard_tests.addTests(package_tests)
return standard_tests
|
