def mock_http_with_file(request):
content = ''.join(
chr(randrange(0, 256))
for x in xrange(0, 1024 * randrange(3, 10) + randrange(0, 1024)))
def iter_content(chunk_size=1024):
for i in xrange(0, len(content), chunk_size):
yield content[i:i + chunk_size]
mock_response = NonCallableMagicMock()
mock_response.iter_content = MagicMock(side_effect=iter_content)
mock_requests = NonCallableMagicMock()
mock_requests.get = MagicMock(return_value=mock_response)
mock_requests._mock_response = mock_response
mock_requests._content = content
patcher = patch(
target='fabric_ovirt.lib.remote_files.requests',
new=mock_requests,
)
def finalizer():
patcher.stop()
return patcher.start()
def mock_http_with_digest_file(request, mock_digests_and_files):
digest_file_lines = (' '.join(pair) for pair in mock_digests_and_files)
mock_response = NonCallableMagicMock()
mock_response.iter_lines = MagicMock(return_value=digest_file_lines)
mock_requests = NonCallableMagicMock()
mock_requests.get = MagicMock(return_value=mock_response)
mock_requests._mock_response = mock_response
patcher = patch(
target='fabric_ovirt.lib.remote_files.requests',
new=mock_requests,
)
def finalizer():
patcher.stop()
return patcher.start()
