particle-os/debian-koji
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

use multicall in a few more places in the cli

Browse source
This commit is contained in:
Mike McLean 2019-05-04 15:56:32 -04:00
parent 29f62a090b
commit 14b25b1f4b
4 changed files with 21 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/test_cli/test_set_pkg_arches.py
View file

@ -47,10 +47,13 @@ class TestSetPkgArches(utils.CliTestCase):
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)
session.packageListSetArches.assert_has_calls(calls)
multicall.packageListSetArches.assert_has_calls(calls)
self.assert_console_message(stdout, '')
def test_handle_set_pkg_arches_help(self):

5
tests/test_cli/test_set_pkg_owner.py
View file

@ -47,10 +47,13 @@ class TestSetPkgOwner(utils.CliTestCase):
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)
session.packageListSetOwner.assert_has_calls(calls)
multicall.packageListSetOwner.assert_has_calls(calls)
self.assert_console_message(stdout, '')
def test_handle_set_pkg_owner_help(self):

5
tests/test_cli/test_unblock_pkg.py
View file

@ -48,10 +48,13 @@ class TestUnblockPkg(utils.CliTestCase):
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)
session.packageListUnblock.assert_has_calls(calls)
multicall.packageListUnblock.assert_has_calls(calls)
self.assert_console_message(stdout, '')
def test_handle_unblock_pkg_help(self):