def test_archive(self):
headers = {'content-disposition': 'filename=foo'}
self.response('archive', 200, **headers)
self.get(self.api + 'tarball/master')
self.conf.update({'stream': True})
assert self.repo.archive(None) is False
assert os.path.isfile('foo') is False
assert self.repo.archive('tarball')
assert os.path.isfile('foo')
os.unlink('foo')
self.mock_assertions()
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
assert os.path.isfile('path_to_file') is False
assert self.repo.archive('tarball', 'path_to_file')
assert os.path.isfile('path_to_file')
os.unlink('path_to_file')
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
self.get(self.api + 'zipball/randomref')
assert self.repo.archive('zipball', ref='randomref')
os.unlink('foo')
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
o = mock.mock_open()
with mock.patch('{0}.open'.format(__name__), o, create=True):
with open('archive', 'wb+') as fd:
self.repo.archive('tarball', fd)
o.assert_called_once_with('archive', 'wb+')
fd = o()
fd.write.assert_called_once_with(b'archive_data')
def test_archive(self):
headers = {'content-disposition': 'filename=foo'}
self.response('archive', 200, **headers)
self.get(self.api + 'tarball/master')
self.conf.update({'stream': True})
assert self.repo.archive(None) is False
assert os.path.isfile('foo') is False
assert self.repo.archive('tarball')
assert os.path.isfile('foo')
os.unlink('foo')
self.mock_assertions()
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
assert os.path.isfile('path_to_file') is False
assert self.repo.archive('tarball', 'path_to_file')
assert os.path.isfile('path_to_file')
os.unlink('path_to_file')
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
self.get(self.api + 'zipball/randomref')
assert self.repo.archive('zipball', ref='randomref')
os.unlink('foo')
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
o = mock.mock_open()
with mock.patch('{0}.open'.format(__name__), o, create=True):
with open('archive', 'wb+') as fd:
self.repo.archive('tarball', fd)
o.assert_called_once_with('archive', 'wb+')
fd = o()
fd.write.assert_called_once_with(b'archive_data')
def test_download(self):
headers = {'content-disposition': 'filename=foo'}
self.response('archive', 200, **headers)
self.get(self.api)
self.conf.update({
'stream': True,
'allow_redirects': False,
'headers': {
'Accept': 'application/octet-stream'
}
})
# 200, to default location
assert os.path.isfile('foo') is False
assert self.asset.download()
assert os.path.isfile('foo')
os.unlink('foo')
self.mock_assertions()
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
# 200, to path
assert os.path.isfile('path_to_file') is False
assert self.asset.download('path_to_file')
assert os.path.isfile('path_to_file')
os.unlink('path_to_file')
self.mock_assertions()
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
# 200, to file-like object
o = mock.mock_open()
with mock.patch('{0}.open'.format(__name__), o, create=True):
with open('download', 'wb+') as fd:
self.asset.download(fd)
o.assert_called_once_with('download', 'wb+')
fd = o()
fd.write.assert_called_once_with(b'archive_data')
self.mock_assertions()
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
# 302, to file-like object
r = self.request.return_value
target = 'http://github.s3.example.com/foo'
self.response('', 302, location=target)
self.get(target)
self.request.side_effect = [self.request.return_value, r]
self.conf['headers'].update({
'Authorization': None,
'Content-Type': None,
})
del self.conf['allow_redirects']
o = mock.mock_open()
with mock.patch('{0}.open'.format(__name__), o, create=True):
with open('download', 'wb+') as fd:
self.asset.download(fd)
o.assert_called_once_with('download', 'wb+')
fd = o()
fd.write.assert_called_once_with(b'archive_data')
self.mock_assertions()
# 404
self.response('', 404)
self.request.side_effect = None
assert self.asset.download() is False
def test_download(self):
headers = {'content-disposition': 'filename=foo'}
self.response('archive', 200, **headers)
self.get(self.api)
self.conf.update({
'stream': True,
'allow_redirects': False,
'headers': {'Accept': 'application/octet-stream'}
})
# 200, to default location
assert os.path.isfile('foo') is False
assert self.asset.download()
assert os.path.isfile('foo')
os.unlink('foo')
self.mock_assertions()
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
# 200, to path
assert os.path.isfile('path_to_file') is False
assert self.asset.download('path_to_file')
assert os.path.isfile('path_to_file')
os.unlink('path_to_file')
self.mock_assertions()
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
# 200, to file-like object
o = mock.mock_open()
with mock.patch('{0}.open'.format(__name__), o, create=True):
with open('download', 'wb+') as fd:
self.asset.download(fd)
o.assert_called_once_with('download', 'wb+')
fd = o()
fd.write.assert_called_once_with(b'archive_data')
self.mock_assertions()
self.request.return_value.raw.seek(0)
self.request.return_value._content_consumed = False
# 302, to file-like object
r = self.request.return_value
target = 'http://github.s3.example.com/foo'
self.response('', 302, location=target)
self.get(target)
self.request.side_effect = [self.request.return_value, r]
self.conf['headers'].update({
'Authorization': None,
'Content-Type': None,
})
del self.conf['allow_redirects']
o = mock.mock_open()
with mock.patch('{0}.open'.format(__name__), o, create=True):
with open('download', 'wb+') as fd:
self.asset.download(fd)
o.assert_called_once_with('download', 'wb+')
fd = o()
fd.write.assert_called_once_with(b'archive_data')
self.mock_assertions()
# 404
self.response('', 404)
self.request.side_effect = None
assert self.asset.download() is False