199 lines
8.4 KiB
Python
199 lines
8.4 KiB
Python
from __future__ import absolute_import
|
|
import koji
|
|
import locale
|
|
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
import os
|
|
import time
|
|
from six.moves import StringIO
|
|
|
|
from koji_cli.commands import anon_handle_rpminfo
|
|
from . import utils
|
|
|
|
|
|
class TestRpminfo(utils.CliTestCase):
|
|
def setUp(self):
|
|
# force locale to compare 'expect' value
|
|
locale.setlocale(locale.LC_ALL, ('en_US', 'UTF-8'))
|
|
self.maxDiff = None
|
|
self.options = mock.MagicMock()
|
|
self.options.quiet = True
|
|
self.options.debug = False
|
|
self.session = mock.MagicMock()
|
|
self.session.getAPIVersion.return_value = koji.API_VERSION
|
|
self.original_timezone = os.environ.get('TZ')
|
|
os.environ['TZ'] = 'UTC'
|
|
time.tzset()
|
|
self.buildroot_info = {'arch': 'x86_64',
|
|
'br_type': 0,
|
|
'host_id': 1,
|
|
'host_name': 'kojibuilder',
|
|
'id': 3,
|
|
'repo_id': 2,
|
|
'tag_id': 4,
|
|
'tag_name': 'test-tag',
|
|
'task_id': 10}
|
|
self.buildinfo = {'build_id': 1,
|
|
'epoch': 7,
|
|
'id': 1,
|
|
'name': 'test-rpm',
|
|
'nvr': 'test-rpm-1.1-11',
|
|
'package_id': 2,
|
|
'package_name': 'test-rpm',
|
|
'release': '11',
|
|
'task_id': 8,
|
|
'version': '1.1'}
|
|
self.getrpminfo = {'arch': 'noarch',
|
|
'build_id': 1,
|
|
'buildroot_id': 3,
|
|
'buildtime': 1615877809,
|
|
'epoch': 7,
|
|
'id': 294,
|
|
'name': 'test-rpm',
|
|
'release': '11',
|
|
'version': '1.1',
|
|
'payloadhash': 'b2b95550390e5f213fc25f33822425f7',
|
|
'size': 7030}
|
|
self.listrpminfos = [{'arch': 'src',
|
|
'build_id': 1,
|
|
'buildroot_id': 3,
|
|
'buildtime': 1615877809,
|
|
'epoch': 7,
|
|
'id': 290,
|
|
'name': 'test-rpm',
|
|
'release': '11',
|
|
'version': '1.1',
|
|
'payloadhash': 'b2b95550390e5f213fc25f33822425f7',
|
|
'size': 7030}]
|
|
self.error_format = """Usage: %s rpminfo [options] <n-v-r.a|id> [<n-v-r.a|id> ...]
|
|
(Specify the --help global option for a list of other help options)
|
|
|
|
%s: error: {message}
|
|
""" % (self.progname, self.progname)
|
|
|
|
def tearDown(self):
|
|
locale.setlocale(locale.LC_ALL, "")
|
|
if self.original_timezone is None:
|
|
del os.environ['TZ']
|
|
else:
|
|
os.environ['TZ'] = self.original_timezone
|
|
time.tzset()
|
|
|
|
@mock.patch('sys.stdout', new_callable=StringIO)
|
|
def test_handle_rpminfo_valid(self, stdout):
|
|
rpm_nvra = 'test-rpm-1.1-11.noarch'
|
|
self.session.getBuildroot.return_value = self.buildroot_info
|
|
self.session.listBuildroots.return_value = [self.buildroot_info]
|
|
self.session.getBuild.return_value = self.buildinfo
|
|
self.session.getRPM.return_value = self.getrpminfo
|
|
self.session.listRPMs.return_value = self.listrpminfos
|
|
expected_output = """RPM: 7:test-rpm-1.1-11.noarch [294]
|
|
Build: test-rpm-1.1-11 [1]
|
|
RPM Path: /mnt/koji/packages/test-rpm/1.1/11/noarch/test-rpm-1.1-11.noarch.rpm
|
|
SRPM: 7:test-rpm-1.1-11 [290]
|
|
SRPM Path: /mnt/koji/packages/test-rpm/1.1/11/src/test-rpm-1.1-11.src.rpm
|
|
Built: Tue, 16 Mar 2021 06:56:49 UTC
|
|
SIGMD5: b2b95550390e5f213fc25f33822425f7
|
|
Size: 7030
|
|
Build ID: 1
|
|
Buildroot: 3 (tag test-tag, arch x86_64, repo 2)
|
|
Build Host: kojibuilder
|
|
Build Task: 10
|
|
Used in 1 buildroots:
|
|
id build tag arch build host
|
|
-------- ---------------------------- -------- -----------------------------
|
|
3 test-tag x86_64 kojibuilder
|
|
"""
|
|
|
|
anon_handle_rpminfo(self.options, self.session, ['--buildroot', rpm_nvra])
|
|
self.assert_console_message(stdout, expected_output)
|
|
self.session.getBuildroot.assert_called_once_with(self.getrpminfo['buildroot_id'])
|
|
self.session.listBuildroots.assert_called_once_with(queryOpts={'order': 'buildroot.id'},
|
|
rpmID=self.getrpminfo['id'])
|
|
self.session.getBuild.assert_called_once_with(self.getrpminfo['build_id'])
|
|
self.session.getRPM.assert_called_once_with(rpm_nvra)
|
|
self.session.listRPMs.assert_called_once_with(buildID=self.getrpminfo['build_id'],
|
|
arches='src')
|
|
|
|
def test_handle_rpminfo_non_exist_nvra(self):
|
|
rpm_nvra = 'test-rpm-nvra.arch'
|
|
self.session.getRPM.return_value = None
|
|
arguments = ['--buildroot', rpm_nvra]
|
|
expected = "No such rpm: %s\n" % rpm_nvra + "\n"
|
|
self.assert_system_exit(
|
|
anon_handle_rpminfo,
|
|
self.options, self.session, arguments,
|
|
stdout='',
|
|
stderr=expected,
|
|
exit_code=1,
|
|
activate_session=None)
|
|
|
|
def test_handle_rpminfo_more_nvra_non_exist_nvra(self):
|
|
rpm_nvra = 'test-rpm-1.1-11.noarch'
|
|
non_exist_rpm_nvra = 'not-test-rpm-11-112.arch'
|
|
self.session.getBuildroot.return_value = self.buildroot_info
|
|
self.session.listBuildroots.return_value = [self.buildroot_info]
|
|
self.session.getBuild.return_value = self.buildinfo
|
|
self.session.getRPM.side_effect = [None, self.getrpminfo]
|
|
self.session.listRPMs.return_value = self.listrpminfos
|
|
expected_output = """RPM: 7:test-rpm-1.1-11.noarch [294]
|
|
Build: test-rpm-1.1-11 [1]
|
|
RPM Path: /mnt/koji/packages/test-rpm/1.1/11/noarch/test-rpm-1.1-11.noarch.rpm
|
|
SRPM: 7:test-rpm-1.1-11 [290]
|
|
SRPM Path: /mnt/koji/packages/test-rpm/1.1/11/src/test-rpm-1.1-11.src.rpm
|
|
Built: Tue, 16 Mar 2021 06:56:49 UTC
|
|
SIGMD5: b2b95550390e5f213fc25f33822425f7
|
|
Size: 7030
|
|
Build ID: 1
|
|
Buildroot: 3 (tag test-tag, arch x86_64, repo 2)
|
|
Build Host: kojibuilder
|
|
Build Task: 10
|
|
Used in 1 buildroots:
|
|
id build tag arch build host
|
|
-------- ---------------------------- -------- -----------------------------
|
|
3 test-tag x86_64 kojibuilder
|
|
"""
|
|
|
|
arguments = ['--buildroot', non_exist_rpm_nvra, rpm_nvra]
|
|
expected_error = "No such rpm: %s\n" % non_exist_rpm_nvra + "\n"
|
|
self.assert_system_exit(
|
|
anon_handle_rpminfo,
|
|
self.options, self.session, arguments,
|
|
stdout=expected_output,
|
|
stderr=expected_error,
|
|
exit_code=1,
|
|
activate_session=None)
|
|
self.session.getBuildroot.assert_called_once_with(self.getrpminfo['buildroot_id'])
|
|
self.session.listBuildroots.assert_called_once_with(queryOpts={'order': 'buildroot.id'},
|
|
rpmID=self.getrpminfo['id'])
|
|
self.session.getBuild.assert_called_once_with(self.getrpminfo['build_id'])
|
|
self.assertEqual(self.session.getRPM.call_count, 2)
|
|
self.session.listRPMs.assert_called_once_with(buildID=self.getrpminfo['build_id'],
|
|
arches='src')
|
|
|
|
|
|
def test_rpminfo_without_option(self):
|
|
arguments = []
|
|
self.assert_system_exit(
|
|
anon_handle_rpminfo,
|
|
self.options, self.session, arguments,
|
|
stdout='',
|
|
stderr=self.format_error_message("Please specify an RPM"),
|
|
exit_code=2,
|
|
activate_session=None)
|
|
self.session.getBuildroot.assert_not_called()
|
|
self.session.listBuildroots.assert_not_called()
|
|
self.session.getBuild.assert_not_called()
|
|
|
|
def test_rpminfo_help(self):
|
|
self.assert_help(
|
|
anon_handle_rpminfo,
|
|
"""Usage: %s rpminfo [options] <n-v-r.a|id> [<n-v-r.a|id> ...]
|
|
(Specify the --help global option for a list of other help options)
|
|
|
|
Options:
|
|
-h, --help show this help message and exit
|
|
--buildroots show buildroots the rpm was used in
|
|
""" % self.progname)
|