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

Add and upfate CLI unit tests

Browse source 
Fixes: https://pagure.io/koji/issue/729
This commit is contained in:
Jana Cupova 2021-11-04 10:00:29 +01:00 • committed by Tomas Kopecek
parent ae15fa34ba
commit c3e89500f2
26 changed files with 1573 additions and 1537 deletions
Download patch file Download diff file Expand all files Collapse all files

62
tests/test_cli/test_add_volume.py
View file

@ -2,6 +2,7 @@ from __future__ import absolute_import
import mock
import six
import unittest
import koji
from koji_cli.commands import handle_add_volume
from . import utils
@ -9,60 +10,59 @@ from . import utils
class TestAddVolume(utils.CliTestCase):
# Show long diffs in error output...
maxDiff = None
def setUp(self):
# Show long diffs in error output...
self.maxDiff = None
self.options = mock.MagicMock()
self.options.debug = False
self.session = mock.MagicMock()
self.session.getAPIVersion.return_value = koji.API_VERSION
self.activate_session_mock = mock.patch('koji_cli.commands.activate_session').start()
self.error_format = """Usage: %s add-volume <volume-name>
(Specify the --help global option for a list of other help options)
%s: error: {message}
""" % (self.progname, self.progname)
@mock.patch('sys.stderr', new_callable=six.StringIO)
@mock.patch('sys.stdout', new_callable=six.StringIO)
@mock.patch('koji_cli.commands.activate_session')
def test_handle_add_volume(
self,
activate_session_mock,
stdout,
stderr):
def test_handle_add_volume(self, stdout):
"""Test handle_add_volume function"""
session = mock.MagicMock()
options = mock.MagicMock()
vol_name = 'vol-test-01'
vol_info = {'id': 1, 'name': vol_name}
# Case 1. argument error
expected = self.format_error_message(
"Command requires exactly one volume-name.")
for arg in [[], ['test-1', 'test-2']]:
self.assert_system_exit(
handle_add_volume,
options,
session,
arg,
stderr=expected,
self.options, self.session, arg,
stdout='',
stderr=self.format_error_message("Command requires exactly one volume-name."),
exit_code=2,
activate_session=None)
self.activate_session_mock.assert_not_called()
self.activate_session_mock.reset_mock()
# Case 2. volume already exists
expected = "Volume %s already exists" % vol_name + "\n"
session.getVolume.return_value = vol_info
with self.assertRaises(SystemExit) as ex:
handle_add_volume(options, session, [vol_name])
self.assertExitCode(ex, 1)
self.assert_console_message(stderr, expected)
session.getVolume.assert_called_with(vol_name)
activate_session_mock.assert_not_called()
self.session.getVolume.return_value = vol_info
self.assert_system_exit(
handle_add_volume,
self.options, self.session, [vol_name],
stdout='',
stderr="Volume %s already exists" % vol_name + "\n",
exit_code=1,
activate_session=None)
self.session.getVolume.assert_called_with(vol_name)
self.activate_session_mock.assert_not_called()
self.activate_session_mock.reset_mock()
# Case 3. Add volume
expected = "Added volume %(name)s with id %(id)i" % vol_info + "\n"
session.getVolume.return_value = {}
session.addVolume.return_value = vol_info
handle_add_volume(options, session, [vol_name])
self.session.getVolume.return_value = {}
self.session.addVolume.return_value = vol_info
handle_add_volume(self.options, self.session, [vol_name])
self.assert_console_message(stdout, expected)
session.addVolume(vol_name)
activate_session_mock.assert_called_with(session, options)
self.session.addVolume(vol_name)
self.activate_session_mock.assert_called_with(self.session, self.options)
def test_handle_add_volume_help(self):
self.assert_help(