Skip to content

Commit f5d467d

Browse files
committed
minor fix
1 parent a4ad641 commit f5d467d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/resumable_media/unit/requests/test_download.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,7 @@ def test__write_to_stream_single_shot_download(self, checksum):
230230

231231
response.__enter__.assert_called_once_with()
232232
response.__exit__.assert_called_once_with(None, None, None)
233-
response.raw.read.assert_called_once_with()
233+
response.raw.read.assert_called_once_with(decode_content=True)
234234

235235
def _consume_helper(
236236
self,
@@ -1370,7 +1370,7 @@ def _mock_response(status_code=http.client.OK, chunks=None, headers=None):
13701370
response.__enter__.return_value = response
13711371
response.__exit__.return_value = None
13721372
response.iter_content.return_value = iter(chunks)
1373-
response.raw.read = mock.Mock(return_value=b"".join(chunks))
1373+
response.raw.read = mock.Mock(side_effect=lambda *args, **kwargs: b"".join(chunks))
13741374
return response
13751375
else:
13761376
return mock.Mock(

0 commit comments

Comments
 (0)