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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
import warnings
from django.core.exceptions import ImproperlyConfigured
from django.test import TestCase
from haystack.utils import loading
class LoadBackendTestCase(TestCase):
def test_load_solr(self):
try:
import pysolr
except ImportError:
warnings.warn(
"Pysolr doesn't appear to be installed. Unable to test loading the Solr backend."
)
return
backend = loading.load_backend("haystack.backends.solr_backend.SolrEngine")
self.assertEqual(backend.__name__, "SolrEngine")
def test_load_whoosh(self):
try:
import whoosh
except ImportError:
warnings.warn(
"Whoosh doesn't appear to be installed. Unable to test loading the Whoosh backend."
)
return
backend = loading.load_backend("haystack.backends.whoosh_backend.WhooshEngine")
self.assertEqual(backend.__name__, "WhooshEngine")
def test_load_elasticsearch(self):
try:
import elasticsearch
except ImportError:
warnings.warn(
"elasticsearch-py doesn't appear to be installed. Unable to test loading the ElasticSearch backend."
)
return
backend = loading.load_backend(
"haystack.backends.elasticsearch_backend.ElasticsearchSearchEngine"
)
self.assertEqual(backend.__name__, "ElasticsearchSearchEngine")
def test_load_simple(self):
backend = loading.load_backend("haystack.backends.simple_backend.SimpleEngine")
self.assertEqual(backend.__name__, "SimpleEngine")
def test_load_nonexistent(self):
try:
backend = loading.load_backend("foobar")
self.fail()
except ImproperlyConfigured as e:
self.assertEqual(
str(e),
"The provided backend 'foobar' is not a complete Python path to a BaseEngine subclass.",
)
try:
backend = loading.load_backend("foobar.FooEngine")
self.fail()
except ImportError as e:
pass
try:
backend = loading.load_backend("haystack.backends.simple_backend.FooEngine")
self.fail()
except ImportError as e:
self.assertEqual(
str(e),
"The Python module 'haystack.backends.simple_backend' has no 'FooEngine' class.",
)
|
