Fix patch methods in batch requests (fixes #171)

Author: leplatrem

CHANGELOG.rst

@@ -7,8 +7,9 @@ This document describes changes between each past release.
 9.2.0 (unreleased)
 ==================
 
-- Nothing changed yet.
+**Bug fixes**
 
+- Fix patch methods in batch requests (fixes #171)
 
 9.1.0 (2018-02-05)
 ==================

kinto_http/batch.py

@@ -19,10 +19,17 @@ def __init__(self, client, batch_max_requests=0):
         self._results = []
 
     def request(self, method, endpoint, data=None, permissions=None,
-                headers=None):
+                payload=None, headers=None):
         # Store all the requests in a dict, to be read later when .send()
         # is called.
-        self.requests.append((method, endpoint, data, permissions, headers))
+        payload = payload or {}
+        if data is not None:
+            payload['data'] = data
+        if permissions is not None:
+            payload['permissions'] = permissions
+
+        self.requests.append((method, endpoint, payload.get("data"),
+                              payload.get("permissions"), headers))
         # This is the signature of the session request.
         return defaultdict(dict), defaultdict(dict)
 

kinto_http/tests/test_client.py

@@ -25,6 +25,7 @@ def test_context_manager_works_as_expected(self):
         with self.client.batch(bucket='mozilla', collection='test') as batch:
             batch.create_record(id=1234, data={'foo': 'bar'})
             batch.create_record(id=5678, data={'bar': 'baz'})
+            batch.patch_record(id=5678, data={'bar': 'biz'})
 
         self.session.request.assert_called_with(
             method='POST',
@@ -37,7 +38,11 @@ def test_context_manager_works_as_expected(self):
                 {'body': {'data': {'bar': 'baz'}},
                  'path': '/buckets/mozilla/collections/test/records/5678',
                  'method': 'PUT',
-                 'headers': {'If-None-Match': '*'}}]})
+                 'headers': {'If-None-Match': '*'}},
+                {'body': {'data': {'bar': 'biz'}},
+                 'path': '/buckets/mozilla/collections/test/records/5678',
+                 'method': 'PATCH',
+                 'headers': {'Content-Type': 'application/json'}}]})
 
     def test_batch_raises_exception(self):
         # Make the next call to sess.request raise a 403.