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
|
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 ((7, 0, 0) <= elasticsearch.__version__ < (8, 0, 0)):
raise ImportError
from elasticsearch import Elasticsearch, exceptions
except ImportError:
log.error(
"Skipping ElasticSearch 7 tests: 'elasticsearch>=7.0.0,<8.0.0' not installed."
)
raise unittest.SkipTest("'elasticsearch>=7.0.0,<8.0.0' not installed.")
url = settings.HAYSTACK_CONNECTIONS["elasticsearch"]["URL"]
es = Elasticsearch(url)
try:
es.info()
except exceptions.ConnectionError as e:
log.error("elasticsearch not running on %r" % url, exc_info=True)
raise unittest.SkipTest("elasticsearch not running on %r" % url, e)
package_tests = loader.discover(
start_dir=os.path.dirname(__file__), pattern=pattern
)
standard_tests.addTests(package_tests)
return standard_tests
|
