From be766b4a3080feda1218bd26eeab3216ebcde831 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Honza=20Kr=C3=A1l?= Date: Wed, 4 Jun 2014 15:11:39 +0400 Subject: [PATCH] Adding a test for a single document in bulk helper --- test_elasticsearch/test_server/test_helpers.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/test_elasticsearch/test_server/test_helpers.py b/test_elasticsearch/test_server/test_helpers.py index e686de16..67bafb3b 100644 --- a/test_elasticsearch/test_server/test_helpers.py +++ b/test_elasticsearch/test_server/test_helpers.py @@ -54,6 +54,15 @@ class TestStreamingBulk(ElasticsearchTestCase): class TestBulk(ElasticsearchTestCase): + def test_bulk_works_with_single_item(self): + docs = [{"answer": 42, '_id': 1}] + success, failed = helpers.bulk(self.client, docs, index='test-index', doc_type='answers', refresh=True) + + self.assertEquals(1, success) + self.assertFalse(failed) + self.assertEquals(1, self.client.count(index='test-index', doc_type='answers')['count']) + self.assertEquals({"answer": 42}, self.client.get(index='test-index', doc_type='answers', id=1)['_source']) + def test_all_documents_get_inserted(self): docs = [{"answer": x, '_id': x} for x in range(100)] success, failed = helpers.bulk(self.client, docs, index='test-index', doc_type='answers', refresh=True)