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_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_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_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_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(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_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_get_default_connection(self):
from needlestack.whoosh.base import Whoosh
connection = manager.get_connection("default")
self.assertIsInstance(connection, Whoosh)
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_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 test_get_default_connection(self):
from needlestack.elasticsearch.base import ElasticSearch
connection = manager.get_connection("default")
self.assertIsInstance(connection, ElasticSearch)
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 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)
def tearDown(self):
connection = manager.get_connection()
connection.delete_index(Index2)
def test_get_default_connection_twice(self):
connection1 = manager.get_connection()
connection2 = manager.get_connection()
self.assertEqual(connection1, connection2)
def tearDown(self):
connection = manager.get_connection()
connection.delete_all_indexes()
