Merge branch 'master' into 3634-tab-state

[arvados.git] / sdk / python / tests / arvados_testutil.py

diff --git a/sdk/python/tests/arvados_testutil.py b/sdk/python/tests/arvados_testutil.py
index 48fb013891564f04fdb69e19d71c6c648b01e97d..0dbf9bcee52b2b420f92f06611ec858911cad2ed 100644 (file)
--- a/sdk/python/tests/arvados_testutil.py
+++ b/sdk/python/tests/arvados_testutil.py
@@ -20,9 +20,9 @@ def fake_httplib2_response(code, **headers):
                    reason=httplib.responses.get(code, "Unknown Response"))
     return httplib2.Response(headers)
 
-def mock_responses(body, *codes):
+def mock_responses(body, *codes, **headers):
     return mock.patch('httplib2.Http.request', side_effect=(
-            (fake_httplib2_response(code), body) for code in codes))
+            (fake_httplib2_response(code, **headers), body) for code in codes))
 
 class ArvadosBaseTestCase(unittest.TestCase):
     # This class provides common utility functions for our tests.