def test_index_document(self): connection = manager.get_connection() connection.update(Index2, {"id": "1", "content": "Kaka"}) response = connection.search("Kaka", index=Index2) self.assertEqual(len(response), 1)
def test_create_destroy_index(self): connection = manager.get_connection() connection.create_index(Index1) with self.assertRaises(exceptions.IndexAlreadyExists): connection.create_index(Index1) connection.delete_index(Index1)
def test_update_document(self): connection = manager.get_connection() connection.update("index2", {"id": "1", "content": "Kaka1"}) connection.update("index2", {"id": "1", "content": "Kaka2"}) response = connection.search("*", index="index2") self.assertEqual(len(response), 1) self.assertEqual(response.items[0].data["content"], "Kaka2")
def test_index_document_with_custom_parser_01(self): parser = qparser.QueryParser("text", Index3.get_schema()) doc = self._create_document(id=1) connection = manager.get_connection() connection.update("index3", doc) result = connection.search("speci", index="index3", parser=parser) self.assertEqual(len(result), 1)
def test_index_document(self): connection = manager.get_connection() connection.update(Index2, {"id": "1", "content": "Kaka"}) connection.refresh(Index2) response = connection.search({"query": {"match_all":{}}}, index=Index2) self.assertIsInstance(response, result.SearchResult) self.assertEqual(len(response), 1)
def test_match_all_with_slicing(self): connection = manager.get_connection() for i in range(30): doc = {"id": "foo.{0}".format(i), "content": "kaka-{0}".format(i)} connection.update(Index2, doc) connection.refresh(Index2) response = connection.search({"query":{"match_all":{}}}, size=2, index=Index2) self.assertEqual(len(response), 2) self.assertEqual(response.total_results, 30)
def test_index_document(self): connection = manager.get_connection() connection.update(Index2, {"id": "1", "content": "Kaka"}) connection.refresh(Index2) response = connection.search({"query": { "match_all": {} }}, index=Index2) self.assertIsInstance(response, result.SearchResult) self.assertEqual(len(response), 1)
def test_index_document_01(self): doc = self._create_document(id=1) connection = manager.get_connection() connection.update("index3", doc) result = connection.search("Kaka", index="index3") self.assertEqual(len(result), 0) result = connection.search("this is", index="index3") self.assertEqual(len(result), 0) result = connection.search("speci", index="index3") self.assertEqual(len(result), 1)
def test_index_document_with_custom_parser_03(self): parser = qparser.QueryParser("tags", Index3.get_schema()) doc1 = self._create_document(id=1) doc2 = self._create_document(id=2) connection = manager.get_connection() connection.update("index3", doc1) connection.update("index3", doc2) result = connection.search("foo1", index="index3", parser=parser) self.assertEqual(len(result), 1) result = connection.search("foo1 OR bar2", index="index3", parser=parser) self.assertEqual(len(result), 2)
def test_index_document_with_custom_parser_02(self): parser = qparser.QueryParser("number", Index3.get_schema()) doc1 = self._create_document(id=1) doc2 = self._create_document(id=2) connection = manager.get_connection() connection.update("index3", doc1) connection.update("index3", doc2) result = connection.search("2", index="index3", parser=parser) self.assertEqual(len(result), 1) result = connection.search("2 OR 3", index="index3", parser=parser) self.assertEqual(len(result), 2)
def test_match_all_with_slicing(self): connection = manager.get_connection() for i in range(30): doc = {"id": "foo.{0}".format(i), "content": "kaka-{0}".format(i)} connection.update(Index2, doc) connection.refresh(Index2) response = connection.search({"query": { "match_all": {} }}, size=2, index=Index2) self.assertEqual(len(response), 2) self.assertEqual(response.total_results, 30)
def test_get_default_connection_twice(self): connection1 = manager.get_connection() connection2 = manager.get_connection() self.assertEqual(connection1, connection2)
def test_get_default_connection(self): from needlestack.whoosh.base import Whoosh connection = manager.get_connection("default") self.assertIsInstance(connection, Whoosh)
def setUp(self): connection = manager.get_connection() connection.delete_all_indexes() connection.create_index(Index3)
def test_index_document_using_string_resolve(self): connection = manager.get_connection() connection.update("index2", {"id": "1", "content": "Kaka"}) response = connection.search("Kaka", index="index2") self.assertEqual(len(response), 1)
def tearDown(self): connection = manager.get_connection() connection.delete_all_indexes()
def test_get_inexistent_connection(self): from django.core.exceptions import ImproperlyConfigured with self.assertRaises(ImproperlyConfigured): connection = manager.get_connection("inexistent")
def test_get_default_connection(self): from needlestack.elasticsearch.base import ElasticSearch connection = manager.get_connection("default") self.assertIsInstance(connection, ElasticSearch)
def setUp(self): connection = manager.get_connection() connection.delete_all_indexes() connection.create_index(Index2)
def setUp(self): connection = manager.get_connection() connection.create_index(Index2)
def tearDown(self): connection = manager.get_connection() connection.delete_index(Index2)