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)