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