def test_urls(self): # This test backwards compatibility with old URLS setting backend = ElasticSearch(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 = ElasticSearch({}) query = mock.MagicMock() query.queryset = models.SearchTest.objects.all() query.get_query.return_value = 'QUERY' query.get_sort.return_value = None return backend.results_class(backend, query)
def test_default_settings(self): backend = ElasticSearch(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_old_name_deprecation(self): from wagtail.wagtailsearch.backends.elasticsearch import ElasticSearch with warnings.catch_warnings(record=True) as w: warnings.simplefilter('always') ElasticSearch({}) self.assertEqual(len(w), 1) self.assertIs(w[0].category, RemovedInWagtail18Warning)
def test_hosts(self): # This tests that HOSTS goes to es_hosts backend = ElasticSearch(params={ 'HOSTS': [ { 'host': '127.0.0.1', 'port': 9300, 'use_ssl': True, } ] }) self.assertEqual(len(backend.es_hosts), 1) self.assertEqual(backend.es_hosts[0]['host'], '127.0.0.1') self.assertEqual(backend.es_hosts[0]['port'], 9300) self.assertEqual(backend.es_hosts[0]['use_ssl'], True)