Added return to bulk method and unit test for bulk method.

elasticmock/fake_elasticsearch.py

@@ -74,6 +74,7 @@ def index(self, index, body, doc_type='_doc', id=None, params=None):
     def bulk(self, body, params=None):
         id = None
         version = 1
+        items = []
 
         for line in body.splitlines():
             if len(line.strip()) == 0:
@@ -99,6 +100,20 @@ def bulk(self, body, params=None):
                     '_version': version
                 })
 
+                items.append({'index': {
+                    '_type': doc_type,
+                    '_id': id,
+                    '_index': index,
+                    '_version': version,
+                    'result': 'created',
+                    'status': 201
+                }})
+
+        return {
+            'errors': False,
+            'items': items
+        }
+
     @query_params('parent', 'preference', 'realtime', 'refresh', 'routing')
     def exists(self, index, doc_type, id, params=None):
         result = False

tests/test_elasticmock.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 
+import json
 import unittest
 import elasticsearch
 from elasticsearch.exceptions import NotFoundError
@@ -27,6 +28,32 @@ def test_should_index_document(self):
         self.assertEqual(1, data.get('_version'))
         self.assertEqual(self.index_name, data.get('_index'))
 
+    def test_should_bulk_index_documents(self):
+        action = {'index': {'_index': self.index_name, '_type': self.doc_type}}
+        action_json = json.dumps(action)
+        body_json = json.dumps(self.body)
+        num_of_documents = 10
+
+        lines = []
+        for count in range(0, num_of_documents):
+            lines.append(action_json)
+            lines.append(body_json)
+        body = '\n'.join(lines)
+
+        data = self.es.bulk(body=body)
+        items = data.get('items')
+
+        self.assertFalse(data.get('errors'))
+        self.assertEqual(num_of_documents, len(items))
+
+        for item in items:
+            index = item.get('index')
+
+            self.assertEqual(self.doc_type, index.get('_type'))
+            self.assertEqual(self.index_name, index.get('_index'))
+            self.assertEqual('created', index.get('result'))
+            self.assertEqual(201, index.get('status'))
+
     def test_should_index_document_without_doc_type(self):
         data = self.es.index(index=self.index_name, body=self.body)