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

Add unit tests for koji commands

Browse source
This commit is contained in:
Franz Chih-Ping Hsieh 2017-11-20 11:24:22 -05:00 • committed by Mike McLean
parent d331fdfa11
commit f16f26be63
17 changed files with 3502 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

125
tests/test_cli/test_list_tasks.py
View file

@ -1,8 +1,12 @@
import mock
import six
import unittest
import koji
from koji_cli.lib import _list_tasks
from koji_cli.commands import handle_list_tasks
from . import utils
class TestListTasks(unittest.TestCase):
def setUp(self):
@ -29,7 +33,7 @@ class TestListTasks(unittest.TestCase):
'state': [koji.TASK_STATES[s] for s in ('FREE', 'OPEN', 'ASSIGNED')],
'decode': True,
'owner': 1,
}, {'order' : 'priority,create_time'})
}, {'order': 'priority,create_time'})
# invalid me
session.getLoggedInUser.return_value = None
@ -50,7 +54,7 @@ class TestListTasks(unittest.TestCase):
'state': [koji.TASK_STATES[s] for s in ('FREE', 'OPEN', 'ASSIGNED')],
'decode': True,
'owner': 2,
}, {'order' : 'priority,create_time'})
}, {'order': 'priority,create_time'})
# invalid user
session.getUser.return_value = None
@ -66,7 +70,7 @@ class TestListTasks(unittest.TestCase):
'state': [koji.TASK_STATES[s] for s in ('FREE', 'OPEN', 'ASSIGNED')],
'decode': True,
'arch': ['x86_64', 'i386'],
}, {'order' : 'priority,create_time'})
}, {'order': 'priority,create_time'})
# only method
session.listTasks.reset_mock()
@ -77,7 +81,7 @@ class TestListTasks(unittest.TestCase):
'state': [koji.TASK_STATES[s] for s in ('FREE', 'OPEN', 'ASSIGNED')],
'decode': True,
'method': 'method',
}, {'order' : 'priority,create_time'})
}, {'order': 'priority,create_time'})
# only channel
session.listTasks.reset_mock()
@ -89,7 +93,7 @@ class TestListTasks(unittest.TestCase):
'state': [koji.TASK_STATES[s] for s in ('FREE', 'OPEN', 'ASSIGNED')],
'decode': True,
'channel_id': 123,
}, {'order' : 'priority,create_time'})
}, {'order': 'priority,create_time'})
session.getChannel.assert_called_once_with('channel')
# invalid channel
@ -107,7 +111,7 @@ class TestListTasks(unittest.TestCase):
'state': [koji.TASK_STATES[s] for s in ('FREE', 'OPEN', 'ASSIGNED')],
'decode': True,
'host_id': 234,
}, {'order' : 'priority,create_time'})
}, {'order': 'priority,create_time'})
session.getHost.assert_called_once_with('host')
# invalid host
@ -127,7 +131,7 @@ class TestListTasks(unittest.TestCase):
{
'children': [
{
'children': [ {'id': 3, 'parent': 2, 'sub': True}],
'children': [{'id': 3, 'parent': 2, 'sub': True}],
'id': 2,
'parent': 1,
'sub': True
@ -148,3 +152,110 @@ class TestListTasks(unittest.TestCase):
'sub': True}
])
class TestCliListTasks(utils.CliTestCase):
# Show long diffs in error output...
maxDiff = None
def setUp(self):
self.error_format = """Usage: %s list-tasks [options]
(Specify the --help global option for a list of other help options)
%s: error: {message}
""" % (self.progname, self.progname)
@mock.patch('sys.stdout', new_callable=six.StringIO)
@mock.patch('koji_cli.commands._list_tasks')
@mock.patch('koji_cli.commands.activate_session')
def test_handle_list_tasks(
self,
activate_session_mock,
list_tasks_mock,
stdout):
"""Test handle_list_tasks function"""
session = mock.MagicMock()
options = mock.MagicMock(quiet=False)
tasks = [
{
'id': 5,
'sub': True,
'state': 0,
'owner_name': 'kojiadmin',
'method': 'createrepo',
'parent': 3,
'priority': 14,
'arch': 'noarch'
},
{
'children':
[
{
'id': 5,
'sub': True,
'state': 0,
'owner_name': 'kojiadmin',
'method': 'createrepo',
'parent': 3,
'priority': 14,
'arch': 'noarch'
},
],
'id': 3,
'state': 0,
'owner_name': 'kojiadmin',
'method': 'newRepo',
'priority': 15,
'arch': 'noarch',
}
]
header = \
"ID Pri Owner State Arch Name\n"
task_output = \
"3 15 kojiadmin FREE noarch newRepo (noarch)\n" + \
"5 14 kojiadmin FREE noarch +createrepo (noarch)\n"
expected = self.format_error_message(
"This command takes no arguments")
# Case 1, argument error test.
self.assert_system_exit(
handle_list_tasks,
options,
session,
['test'],
stderr=expected,
activate_session=None)
# Case 2, no tasks
list_tasks_mock.return_value = None
handle_list_tasks(options, session, [])
self.assert_console_message(stdout, '(no tasks)\n')
# Case 3, show tasks with header
list_tasks_mock.return_value = tasks
handle_list_tasks(options, session, [])
self.assert_console_message(stdout, header + task_output)
# Case 4. show task without header
handle_list_tasks(options, session, ['--quiet'])
self.assert_console_message(stdout, task_output)
def test_handle_list_tasks_help(self):
self.assert_help(
handle_list_tasks,
"""Usage: %s list-tasks [options]
(Specify the --help global option for a list of other help options)
Options:
-h, --help show this help message and exit
--mine Just print your tasks
--user=USER Only tasks for this user
--arch=ARCH Only tasks for this architecture
--method=METHOD Only tasks of this method
--channel=CHANNEL Only tasks in this channel
--host=HOST Only tasks for this host
--quiet Do not display the column headers
""" % self.progname)