def get_results(self):
backend = Elasticsearch5SearchBackend({})
query_compiler = mock.MagicMock()
query_compiler.queryset = models.Book.objects.all()
query_compiler.get_query.return_value = "QUERY"
query_compiler.get_sort.return_value = None
return backend.results_class(backend, query_compiler)
def test_urls(self):
# This test backwards compatibility with old URLS setting
backend = Elasticsearch5SearchBackend(
params={
'URLS': [
'http://*****:*****@elasticsearch.mysite.com',
'https://elasticsearch.mysite.com/hello',
],
})
self.assertEqual(len(backend.hosts), 4)
self.assertEqual(backend.hosts[0]['host'], 'localhost')
self.assertEqual(backend.hosts[0]['port'], 12345)
self.assertEqual(backend.hosts[0]['use_ssl'], False)
self.assertEqual(backend.hosts[1]['host'], '127.0.0.1')
self.assertEqual(backend.hosts[1]['port'], 54321)
self.assertEqual(backend.hosts[1]['use_ssl'], True)
self.assertEqual(backend.hosts[2]['host'], 'elasticsearch.mysite.com')
self.assertEqual(backend.hosts[2]['port'], 80)
self.assertEqual(backend.hosts[2]['use_ssl'], False)
self.assertEqual(backend.hosts[2]['http_auth'],
('username', 'password'))
self.assertEqual(backend.hosts[3]['host'], 'elasticsearch.mysite.com')
self.assertEqual(backend.hosts[3]['port'], 443)
self.assertEqual(backend.hosts[3]['use_ssl'], True)
self.assertEqual(backend.hosts[3]['url_prefix'], '/hello')
def setUp(self):
# Create ES mapping
self.es_mapping = Elasticsearch5SearchBackend.mapping_class(
models.Book)
# Create ES document
self.obj = models.Book.objects.get(id=4)
def test_default_settings(self):
backend = Elasticsearch5SearchBackend(params={})
self.assertEqual(len(backend.hosts), 1)
self.assertEqual(backend.hosts[0]['host'], 'localhost')
self.assertEqual(backend.hosts[0]['port'], 9200)
self.assertEqual(backend.hosts[0]['use_ssl'], False)
def test_default_settings(self, Elasticsearch):
Elasticsearch5SearchBackend(params={})
Elasticsearch.assert_called_with(hosts=[{
'host': 'localhost',
'port': 9200,
'url_prefix': '',
'use_ssl': False,
'verify_certs': False,
'http_auth': None
}],
timeout=10)
def test_default_settings(self, Elasticsearch):
Elasticsearch5SearchBackend(params={})
Elasticsearch.assert_called_with(
hosts=[{
"host": "localhost",
"port": 9200,
"url_prefix": "",
"use_ssl": False,
"verify_certs": False,
"http_auth": None,
}],
timeout=10,
)
def test_urls(self, Elasticsearch):
# This test backwards compatibility with old URLS setting
Elasticsearch5SearchBackend(
params={
"URLS": [
"http://*****:*****@elasticsearch.mysite.com",
"https://elasticsearch.mysite.com/hello",
],
})
Elasticsearch.assert_called_with(
hosts=[
{
"host": "localhost",
"port": 12345,
"url_prefix": "",
"use_ssl": False,
"verify_certs": False,
"http_auth": None,
},
{
"host": "127.0.0.1",
"port": 54321,
"url_prefix": "",
"use_ssl": True,
"verify_certs": True,
"http_auth": None,
},
{
"host": "elasticsearch.mysite.com",
"port": 80,
"url_prefix": "",
"use_ssl": False,
"verify_certs": False,
"http_auth": ("username", "password"),
},
{
"host": "elasticsearch.mysite.com",
"port": 443,
"url_prefix": "/hello",
"use_ssl": True,
"verify_certs": True,
"http_auth": None,
},
],
timeout=10,
)
def test_urls(self, Elasticsearch):
# This test backwards compatibility with old URLS setting
Elasticsearch5SearchBackend(params={
'URLS': [
'http://*****:*****@elasticsearch.mysite.com',
'https://elasticsearch.mysite.com/hello',
],
})
Elasticsearch.assert_called_with(
hosts=[
{
'host': 'localhost',
'port': 12345,
'url_prefix': '',
'use_ssl': False,
'verify_certs': False,
'http_auth': None,
},
{
'host': '127.0.0.1',
'port': 54321,
'url_prefix': '',
'use_ssl': True,
'verify_certs': True,
'http_auth': None,
},
{
'host': 'elasticsearch.mysite.com',
'port': 80,
'url_prefix': '',
'use_ssl': False,
'verify_certs': False,
'http_auth': ('username', 'password')
},
{
'host': 'elasticsearch.mysite.com',
'port': 443,
'url_prefix': '/hello',
'use_ssl': True,
'verify_certs': True,
'http_auth': None,
},
],
timeout=10
)
def test_hosts(self):
# This tests that HOSTS goes to es_hosts
backend = Elasticsearch5SearchBackend(
params={
'HOSTS': [{
'host': '127.0.0.1',
'port': 9300,
'use_ssl': True,
'verify_certs': True,
}]
})
self.assertEqual(len(backend.hosts), 1)
self.assertEqual(backend.hosts[0]['host'], '127.0.0.1')
self.assertEqual(backend.hosts[0]['port'], 9300)
self.assertEqual(backend.hosts[0]['use_ssl'], True)
def test_hosts(self, Elasticsearch):
Elasticsearch5SearchBackend(
params={
'HOSTS': [{
'host': '127.0.0.1',
'port': 9300,
'use_ssl': True,
'verify_certs': True,
}]
})
Elasticsearch.assert_called_with(hosts=[{
'host': '127.0.0.1',
'port': 9300,
'use_ssl': True,
'verify_certs': True,
}],
timeout=10)
def test_hosts(self, Elasticsearch):
Elasticsearch5SearchBackend(
params={
"HOSTS": [{
"host": "127.0.0.1",
"port": 9300,
"use_ssl": True,
"verify_certs": True,
}]
})
Elasticsearch.assert_called_with(
hosts=[{
"host": "127.0.0.1",
"port": 9300,
"use_ssl": True,
"verify_certs": True,
}],
timeout=10,
)
def setUp(self):
# Create ES mapping
self.es_mapping = Elasticsearch5SearchBackend.mapping_class(
models.Novel)
self.obj = models.Novel.objects.get(id=4)
def setUp(self):
# Create ES mapping
self.es_mapping = Elasticsearch5SearchBackend.mapping_class(models.Novel)
self.obj = models.Novel.objects.get(id=4)
def setUp(self):
# Create ES mapping
self.es_mapping = Elasticsearch5SearchBackend.mapping_class(models.Book)
# Create ES document
self.obj = models.Book.objects.get(id=4)
