Add bulk_write

arctic/store/bson_store.py

@@ -119,6 +119,17 @@ def find_one_and_replace(self, filter, replacement, **kwargs):
         self._arctic_lib.check_quota()
         return self._collection.find_one_and_replace(filter, replacement, **kwargs)
 
+    @mongo_retry
+    def bulk_write(self, requests, **kwargs):
+        """
+        See http://api.mongodb.com/python/current/api/pymongo/collection.html#pymongo.collection.Collection.bulk_write
+
+        Warning: this is wrapped in mongo_retry, and is therefore potentially unsafe if the write you want to execute
+        isn't idempotent.
+        """
+        self._arctic_lib.check_quota()
+        return self._collection.bulk_write(requests, **kwargs)
+
     @mongo_retry
     def count(self, filter, **kwargs):
         """

tests/unit/store/test_bson_store.py

@@ -109,6 +109,19 @@ def test_find_one_and_replace():
     assert collection.find_one_and_replace.call_args_list == [call(sentinel.filter, sentinel.replacement)]
 
 
+def test_bulk_write():
+    arctic_lib = create_autospec(ArcticLibraryBinding, instance=True)
+    collection = create_autospec(Collection, instance=True)
+    arctic_lib.get_top_level_collection.return_value = collection
+
+    bsons = BSONStore(arctic_lib)
+    bsons.bulk_write(sentinel.requests)
+
+    assert arctic_lib.check_quota.call_count == 1
+    assert collection.bulk_write.call_count == 1
+    assert collection.bulk_write.call_args_list == [call(sentinel.requests)]
+
+
 def test_delete_one():
     arctic_lib = create_autospec(ArcticLibraryBinding, instance=True)
     collection = create_autospec(Collection, instance=True)