particle-os/debian-koji
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fall back to task files for non-complete builds

Browse source
This commit is contained in:
Mike McLean 2024-08-13 13:59:14 -04:00 • committed by Tomas Kopecek
parent 32a609925d
commit 8b5b516227
2 changed files with 41 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
cli/koji_cli/commands.py
View file

@ -7115,12 +7115,18 @@ def anon_handle_download_logs(options, session, args):
binfo = session.getBuild(arg)
if binfo is None:
error("There is no build with n-v-r: %s" % arg)
if binfo.get('build_id'):
b_state = koji.BUILD_STATES[binfo['state']]
# getBuildLogs will only work for complete builds
if b_state == 'COMPLETE':
build_id = binfo['build_id']
sys.stdout.write("Using build ID: %s\n" % build_id)
elif binfo.get('task_id'):
# try falling back to the task logs
task_id = binfo['task_id']
sys.stdout.write("Build %s is %s\n" % (arg, b_state))
sys.stdout.write("Using task ID: %s\n" % task_id)
else:
sys.stdout.write("Unable to download build %s (state=%s)\n" % (arg, b_state))
else:
try:
task_id = int(arg)

38
tests/test_cli/test_download_logs.py
View file

@ -6,6 +6,8 @@ except ImportError:
import mock
import six
import koji
from koji_cli.commands import anon_handle_download_logs
from . import utils
@ -41,6 +43,7 @@ Note this command only downloads task logs, not build logs.
""" % (self.progname, self.progname, self.progname)
self.nvr = 'bash-1.2.3-f26'
self.task_id = 123456
self.build_id = 232323
self.builtin_open = None
if six.PY2:
@ -102,7 +105,31 @@ Note this command only downloads task logs, not build logs.
self.session.getTaskChildren.assert_not_called()
def test_anon_handle_download_logs_nvr(self):
self.session.getBuild.return_value = {'task_id': self.task_id}
self.session.getBuild.return_value = {'task_id': self.task_id,
'build_id': self.build_id,
'state': koji.BUILD_STATES['COMPLETE']}
self.session.getBuildLogs.return_value = [
{'dir': 'noarch', 'name': 'test.log', 'path': 'path/to/test.log'},
]
rv = anon_handle_download_logs(self.options, self.session, ['--nvr', self.nvr])
actual = self.stdout.getvalue()
expected = 'Using build ID: %s\n' % self.build_id
self.assertMultiLineEqual(actual, expected)
self.assertIsNone(rv)
self.session.getBuild.assert_called_once_with(self.nvr)
self.session.getBuildLogs.assert_called_once_with(self.build_id)
self.session.getTaskInfo.assert_not_called()
self.session.downloadTaskOutput.assert_not_called()
self.session.getTaskChildren.assert_not_called()
self.download_file.assert_called_once()
def test_anon_handle_download_logs_nvr_with_task_id(self):
# for a failed build with a task id we should fall back to fetching task logs
self.session.getBuild.return_value = {'build_id': self.build_id,
'task_id': self.task_id,
'nvr': self.nvr,
'state': koji.BUILD_STATES['FAILED']}
self.session.getTaskInfo.return_value = {
'arch': 'x86_64',
'state': 'CLOSED',
@ -112,7 +139,7 @@ Note this command only downloads task logs, not build logs.
rv = anon_handle_download_logs(self.options, self.session, ['--nvr', self.nvr])
actual = self.stdout.getvalue()
expected = 'Using task ID: %s\n' % self.task_id
expected = 'Build %s is FAILED\nUsing task ID: %s\n' % (self.nvr, self.task_id)
self.assertMultiLineEqual(actual, expected)
self.assertIsNone(rv)
@ -121,11 +148,14 @@ Note this command only downloads task logs, not build logs.
self.session.downloadTaskOutput.assert_not_called()
self.session.getTaskChildren.assert_has_calls([mock.call(self.task_id), mock.call(23), ])
def test_anon_handle_download_logs_nvr_without_task_id(self):
self.session.getBuild.return_value = {'build_id': 1, 'nvr': self.nvr}
self.session.getBuild.return_value = {'build_id': self.build_id,
'nvr': self.nvr,
'state': koji.BUILD_STATES['FAILED']}
rv = anon_handle_download_logs(self.options, self.session, ['--nvr', self.nvr])
actual = self.stdout.getvalue()
expected = 'Using build ID: 1\n'
expected = 'Unable to download build %s (state=FAILED)\n' % self.nvr
self.assertMultiLineEqual(actual, expected)
self.assertIsNone(rv)