def test_download(self, tmpdir, filename, httpauth):
url = 'http://example.org/test/file.dat'
filesize = 2 * 1024 * 1024 + 1
self.mox.StubOutWithMock(fileutils, 'open_url')
if httpauth:
fileutils.open_url(url, httpuser='******', httppassword='******'
).AndReturn(self.MockResponse(filesize))
else:
fileutils.open_url(url).AndReturn(self.MockResponse(filesize))
self.mox.ReplayAll()
with tmpdir.as_cwd():
if filename:
output = 'test.name'
if httpauth:
f = f = fileutils.download(url, output, httpuser='******',
httppassword='******')
else:
f = fileutils.download(url, output)
else:
output = 'file.dat'
if httpauth:
f = f = fileutils.download(url, output, httpuser='******',
httppassword='******')
else:
f = fileutils.download(url)
assert f == output
assert os.path.exists(output)
assert os.path.getsize(output) == filesize
self.mox.VerifyAll()
def partial_mock_artifacts(self, matrix):
# JenkinsArtifacts.__init__ does a lot of work, so we can't just
# stubout methods after constructing it
self.mox.StubOutWithMock(fileutils, 'open_url')
if matrix:
fileutils.open_url(MockUrl.unlabelledurl + 'api/xml').AndReturn(
MockUrl(True))
fileutils.open_url(MockUrl.labelledurl + 'api/xml').AndReturn(
MockUrl(False))
self.mox.ReplayAll()
return JenkinsArtifacts(Args(matrix))
def partial_mock_artifacts(self, matrix):
# Artifacts.__init__ does a lot of work, so we can't just
# stubout methods after constructing it
self.mox.StubOutWithMock(fileutils, 'open_url')
if matrix:
fileutils.open_url(
self.MockUrl.unlabelledurl + 'api/xml').AndReturn(
self.MockUrl(True))
fileutils.open_url(
self.MockUrl.labelledurl + 'api/xml').AndReturn(
self.MockUrl(False))
self.mox.ReplayAll()
return Artifacts(self.Args(matrix))
def test_read_downloads(self):
self.mox.StubOutWithMock(fileutils, 'open_url')
fileutils.open_url(MockDownloadUrl.pageurl).AndReturn(
MockDownloadUrl(True))
self.mox.ReplayAll()
fullpath = '%s%s' % (MockDownloadUrl.pageurl,
MockDownloadUrl.artifactpath)
assert ReleaseArtifacts.read_downloads(MockDownloadUrl.pageurl) == {
'ice34': [fullpath + MockDownloadUrl.artifactnames[0]],
'ice35': [fullpath + MockDownloadUrl.artifactnames[1]]
}
self.mox.VerifyAll()
def test_read_downloads(self):
self.mox.StubOutWithMock(fileutils, 'open_url')
fileutils.open_url(
MockDownloadUrl.pageurl).AndReturn(
MockDownloadUrl(True))
self.mox.ReplayAll()
fullpath = '%s%s' % (
MockDownloadUrl.pageurl, MockDownloadUrl.artifactpath)
assert ReleaseArtifacts.read_downloads(MockDownloadUrl.pageurl) == {
'ice34': [fullpath + MockDownloadUrl.artifactnames[0]],
'ice35': [fullpath + MockDownloadUrl.artifactnames[1]]
}
self.mox.VerifyAll()
def partial_mock_artifacts(self, matrix):
# JenkinsArtifacts.__init__ does a lot of work, so we can't just
# stubout methods after constructing it
self.mox.StubOutWithMock(fileutils, 'open_url')
if matrix:
fileutils.open_url(
MockUrl.unlabelledurl + 'api/xml',
httpuser=MockAuth.httpuser,
httppassword=MockAuth.httppassword).AndReturn(
MockUrl(True))
fileutils.open_url(
MockUrl.labelledurl + 'api/xml',
httpuser=MockAuth.httpuser,
httppassword=MockAuth.httppassword).AndReturn(
MockUrl(False))
self.mox.ReplayAll()
return JenkinsArtifacts(Args(matrix))
def partial_mock_artifacts(self, release):
# ReleaseArtifacts.__init__ does a lot of work, so we can't just
# stubout methods after constructing it
self.mox.StubOutWithMock(fileutils, 'dereference_url')
if release == 'latest':
fileutils.dereference_url(MockDownloadUrl.latesturl).AndReturn(
MockDownloadUrl.pageurl)
if release == '0.0':
fileutils.dereference_url(MockDownloadUrl.latesturl +
'0.0').AndReturn(MockDownloadUrl.pageurl)
self.mox.StubOutWithMock(fileutils, 'open_url')
fileutils.open_url(MockDownloadUrl.pageurl).AndReturn(
MockDownloadUrl(True))
self.mox.ReplayAll()
args = Args(False)
args.branch = release
return ReleaseArtifacts(args)
def partial_mock_artifacts(self, release):
# ReleaseArtifacts.__init__ does a lot of work, so we can't just
# stubout methods after constructing it
self.mox.StubOutWithMock(fileutils, 'dereference_url')
if release == 'latest':
fileutils.dereference_url(MockDownloadUrl.latesturl).AndReturn(
MockDownloadUrl.pageurl)
if release == '0.0':
fileutils.dereference_url(
MockDownloadUrl.latesturl + '0.0').AndReturn(
MockDownloadUrl.pageurl)
self.mox.StubOutWithMock(fileutils, 'open_url')
fileutils.open_url(
MockDownloadUrl.pageurl).AndReturn(
MockDownloadUrl(True))
self.mox.ReplayAll()
args = Args(False)
args.branch = release
return ReleaseArtifacts(args)
