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