def test_handle_set_pkg_arches(self, activate_session_mock, stdout):
"""Test handle_set_pkg_arches function"""
session = mock.MagicMock()
options = mock.MagicMock()
arguments = ['x86_64', 'tag', '--force', 'bash', 'less', 'sed']
expected = self.format_error_message(
"Please specify an archlist, a tag, and at least one package")
# Case 1. argument error
self.assert_system_exit(handle_set_pkg_arches,
options,
session, [],
stderr=expected,
activate_session=None)
activate_session_mock.assert_not_called()
# Case 2. run set arch to x86_64
multicall = mock.MagicMock()
multicall.__enter__.return_value = multicall
session.multicall.return_value = multicall
calls = [
mock.call('tag', pkg, 'x86_64', force=True)
for pkg in arguments[3:]
]
handle_set_pkg_arches(options, session, arguments)
activate_session_mock.assert_called_with(session, options)
multicall.packageListSetArches.assert_has_calls(calls)
self.assert_console_message(stdout, '')
def test_handle_set_pkg_arches(
self,
activate_session_mock,
stdout):
"""Test handle_set_pkg_arches function"""
session = mock.MagicMock()
options = mock.MagicMock()
arguments = ['x86_64', 'tag', '--force', 'bash', 'less', 'sed']
expected = self.format_error_message(
"Please specify an archlist, a tag, and at least one package")
# Case 1. argument error
self.assert_system_exit(
handle_set_pkg_arches,
options,
session,
[],
stderr=expected,
activate_session=None)
activate_session_mock.assert_not_called()
# Case 2. run set arch to x86_64
calls = [mock.call('tag', pkg, 'x86_64', force=True) for pkg in arguments[3:]]
handle_set_pkg_arches(options, session, arguments)
activate_session_mock.assert_called_with(session, options)
session.packageListSetArches.assert_has_calls(calls)
self.assert_console_message(stdout, '')