Updating some storage mocks after google-resumable-media==0.3.1 release. (googleapis#4244)

dhermes · web-flow · commit 99db799c5ad4 · 2017-10-25T08:49:40.000-07:00
Fixes googleapis#4243. Also increasing the lower-bound on `google-resumable-media` to `0.3.1`.
diff --git a/storage/setup.py b/storage/setup.py
@@ -53,7 +53,7 @@
 REQUIREMENTS = [
     'google-cloud-core >= 0.27.0, < 0.28dev',
     'google-auth >= 1.0.0',
-    'google-resumable-media >= 0.3.0',
+    'google-resumable-media >= 0.3.1',
     'requests >= 2.18.0',
 ]
 
diff --git a/storage/tests/unit/test_blob.py b/storage/tests/unit/test_blob.py
@@ -441,7 +441,9 @@ def _mock_requests_response(
         response.status_code = status_code
         response.headers.update(headers)
         if stream:
-            response.raw = io.BytesIO(content)
+            raw = io.BytesIO(content)
+            raw.headers = headers
+            response.raw = raw
             response._content = False
         else:
             response.raw = None
@@ -677,15 +679,18 @@ def test_download_to_filename_corrupted(self):
         empty_hash = base64.b64encode(
             hashlib.md5(b'').digest()).decode(u'utf-8')
         headers = {'x-goog-hash': 'md5=' + empty_hash}
+        mock_raw = mock.Mock(headers=headers, spec=['headers'])
         response = mock.MagicMock(
             headers=headers,
             status_code=http_client.OK,
+            raw=mock_raw,
             spec=[
                 '__enter__',
                 '__exit__',
                 'headers',
                 'iter_content',
                 'status_code',
+                'raw',
             ],
         )
         # i.e. context manager returns ``self``.
