actions/opensearch-pyd
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Better tests for the bulk_index helper

Browse Source
This commit is contained in:
Honza Kral
2013-10-06 17:54:07 +02:00
parent cc524e9c9d
commit 44ed34771e
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_elasticsearch/test_server/test_helpers.py
+3 -1
View File
@@ -4,12 +4,13 @@ from . import ElasticTestCase
class TestBulkIndex(ElasticTestCase):
def test_all_documents_get_inserted(self):
docs = [{"answer": x} for x in range(100)]
docs = [{"answer": x, '_id': x} for x in range(100)]
success, failed = helpers.bulk_index(self.client, docs, index='test-index', doc_type='answers', refresh=True)
self.assertEquals(100, len(success))
self.assertFalse(failed)
self.assertEquals(100, self.client.count(index='test-index', doc_type='answers')['count'])
self.assertEquals({"answer": 42}, self.client.get(index='test-index', doc_type='answers', id=42)['_source'])
def test_stats_only_reports_numbers(self):
docs = [{"answer": x} for x in range(100)]
@@ -49,3 +50,4 @@ class TestReindex(ElasticTestCase):
self.assertTrue(self.client.indices.exists_type("prod_index", "questions"))
self.assertEquals(100, self.client.count(index='prod_index')['count'])
self.assertEquals({"answer": 42, "correct": True}, self.client.get(index="prod_index", doc_type="answers", id=42)['_source'])