def test_handle_add_pkg_owner_no_exists(self, activate_session_mock,
stdout):
tag = 'tag'
packages = ['package1', 'package2', 'package3']
owner = 'owner'
owner_info = None
extra_arches = 'arch1,arch2 arch3, arch4'
args = ['--owner=' + owner, '--extra-arches=' + extra_arches, tag
] + packages
options = mock.MagicMock()
# Mock out the xmlrpc server
session = mock.MagicMock()
session.getUser.return_value = owner_info
# Run it and check immediate output
# args: --owner, --extra-arches='arch1,arch2 arch3, arch4',
# tag, package1, package2, package3
# expected: failed: owner does not exist
rv = handle_add_pkg(options, session, args)
actual = stdout.getvalue()
expected = 'User owner does not exist\n'
self.assertMultiLineEqual(actual, expected)
# Finally, assert that things were called as we expected.
activate_session_mock.assert_not_called()
self.assertEqual(session.mock_calls, [call.getUser(owner)])
self.assertEqual(rv, 1)
def test_handle_add_pkg_tag_no_exists(self, activate_session_mock, stdout):
tag = 'tag'
dsttag = None
packages = ['package1', 'package2', 'package3']
owner = 'owner'
owner_info = mock.ANY
extra_arches = 'arch1,arch2 arch3, arch4'
args = ['--owner=' + owner, '--extra-arches=' + extra_arches, tag
] + packages
options = mock.MagicMock()
# Mock out the xmlrpc server
session = mock.MagicMock()
session.getUser.return_value = owner_info
session.getTag.return_value = dsttag
# Run it and check immediate output
# args: --owner, --extra-arches='arch1,arch2 arch3, arch4',
# tag, package1, package2, package3
# expected: failed: tag does not exist
with self.assertRaises(SystemExit) as cm:
handle_add_pkg(options, session, args)
actual = stdout.getvalue()
expected = 'No such tag: tag\n'
self.assertMultiLineEqual(actual, expected)
# Finally, assert that things were called as we expected.
activate_session_mock.assert_called_once_with(session, options)
self.assertEqual(
session.mock_calls,
[call.getUser(owner), call.getTag(tag)])
if isinstance(cm.exception, int):
self.assertEqual(cm.exception, 1)
else:
self.assertEqual(cm.exception.code, 1)
def test_handle_add_pkg_tag_no_exists(self):
tag = 'tag'
dsttag = None
packages = ['package1', 'package2', 'package3']
owner = 'owner'
owner_info = mock.ANY
extra_arches = 'arch1,arch2 arch3, arch4'
arguments = [
'--owner=' + owner, '--extra-arches=' + extra_arches, tag
] + packages
self.session.getUser.return_value = owner_info
self.session.getTag.return_value = dsttag
# Run it and check immediate output
# args: --owner, --extra-arches='arch1,arch2 arch3, arch4',
# tag, package1, package2, package3
# expected: failed: tag does not exist
self.assert_system_exit(handle_add_pkg,
self.options,
self.session,
arguments,
stdout='',
stderr='No such tag: %s\n' % tag,
exit_code=1,
activate_session=None)
# Finally, assert that things were called as we expected.
self.activate_session_mock.assert_called_once_with(
self.session, self.options)
self.assertEqual(
self.session.mock_calls,
[call.getUser(owner), call.getTag(tag)])
def test_handle_add_pkg_multi_pkg(self, activate_session_mock, stdout):
tag = 'tag'
dsttag = {'name': tag, 'id': 1}
packages = ['package1', 'package2', 'package3']
owner = 'owner'
owner_info = mock.ANY
extra_arches = 'arch1,arch2 arch3, arch4'
args = ['--owner=' + owner, '--extra-arches=' + extra_arches, tag
] + packages
kwargs = {
'force': None,
'block': False,
'extra_arches': 'arch1 arch2 arch3 arch4'
}
options = mock.MagicMock()
# Mock out the xmlrpc server
session = mock.MagicMock()
session.getUser.return_value = owner_info
session.getTag.return_value = dsttag
session.listPackages.return_value = [{
'package_name': 'package2',
'package_id': 2
}]
# Run it and check immediate output
# args: --owner, --extra-arches='arch1,arch2 arch3, arch4',
# tag, package1, package2, package3
# expected: success
rv = handle_add_pkg(options, session, args)
actual = stdout.getvalue()
expected = 'Package package2 already exists in tag tag\nAdding 2 packages to tag tag\n'
self.assertMultiLineEqual(actual, expected)
# Finally, assert that things were called as we expected.
activate_session_mock.assert_called_once_with(session, options)
self.assertEqual(session.mock_calls, [
call.getUser(owner),
call.getTag(tag),
call.listPackages(tagID=dsttag['id']),
call.packageListAdd(tag, packages[0], owner, **kwargs),
call.packageListAdd(tag, packages[2], owner, **kwargs),
call.multiCall(strict=True)
])
self.assertNotEqual(rv, 1)
