def test_urls(self): # This test backwards compatibility with old URLS setting backend = Elasticsearch2SearchBackend( 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 get_results(self): backend = Elasticsearch2SearchBackend({}) 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_default_settings(self): backend = Elasticsearch2SearchBackend(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): Elasticsearch2SearchBackend(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_as_string_works(self, Elasticsearch): Elasticsearch2SearchBackend(params={'URLS': 'http://localhost:9200'}) 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 Elasticsearch2SearchBackend(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 = Elasticsearch2SearchBackend( 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): Elasticsearch2SearchBackend( 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 = Elasticsearch2SearchBackend.mapping_class( models.Book) self.obj = models.Book.objects.get(id=4)
def setUp(self): # Create ES mapping self.es_mapping = Elasticsearch2SearchBackend.mapping_class(models.Novel) self.obj = models.Novel.objects.get(id=4)