From ebf4e88d7db86f8f9c038d23a15a33ccc60243c1 Mon Sep 17 00:00:00 2001 From: Yuming Zhu Date: Tue, 11 Oct 2016 19:14:02 +0000 Subject: [PATCH] ut: cli - test_build --- cli/koji | 1 - tests/test_cli/test_build.py | 731 +++++++++++++++++++++++++++++++++++ 2 files changed, 731 insertions(+), 1 deletion(-) create mode 100644 tests/test_cli/test_build.py diff --git a/cli/koji b/cli/koji index 34e4920e..7c08d074 100755 --- a/cli/koji +++ b/cli/koji @@ -962,7 +962,6 @@ def handle_build(options, session, args): parser.error(_("Destination tag %s is locked" % dest_tag['name'])) source = args[1] opts = {} - print build_opts if build_opts.arch_override: opts['arch_override'] = ' '.join(build_opts.arch_override.replace(',',' ').split()) for key in ('skip_tag', 'scratch', 'repo_id'): diff --git a/tests/test_cli/test_build.py b/tests/test_cli/test_build.py new file mode 100644 index 00000000..477839de --- /dev/null +++ b/tests/test_cli/test_build.py @@ -0,0 +1,731 @@ +import unittest + +import StringIO as stringio + +import os + +import sys + +import mock + +import loadcli + +cli = loadcli.cli + + +class TestBuild(unittest.TestCase): + # Show long diffs in error output... + maxDiff = None + + def setUp(self): + # Mock out the options parsed in main + self.options = mock.MagicMock() + self.options.quiet = None + self.options.weburl = 'weburl' + # Mock out the xmlrpc server + self.session = mock.MagicMock() + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_from_srpm(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, activate_session_mock, + stdout): + target = 'target' + dest_tag = 'dest_tag' + target_info = {'dest_tag': dest_tag} + dest_tag_info = {'locked': False} + source = 'srpm' + task_id = 1 + args = [target, source] + opts = {} + priority = None + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + self.session.build.return_value = task_id + # Run it and check immediate output + # args: target, srpm + # expected: success + rv = cli.handle_build(self.options, self.session, args) + actual = stdout.getvalue() + expected = """Uploading srpm: srpm + +Created task: 1 +Task info: weburl/taskinfo?taskID=1 +""" + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_called_once_with('cli-build') + self.assertEqual(running_in_bg_mock.call_count, 2) + self.session.uploadWrapper.assert_called_once_with(source, 'random_path', callback=cli._progress_callback) + self.session.build.assert_called_once_with('random_path/' + source, target, opts, priority=priority) + self.session.logout.assert_called() + watch_tasks_mock.assert_called_once_with(self.session, [task_id], quiet=self.options.quiet) + self.assertEqual(rv, 0) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_from_scm(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stdout): + target = 'target' + dest_tag = 'dest_tag' + target_info = {'dest_tag': dest_tag} + dest_tag_info = {'locked': False} + source = 'http://scm' + task_id = 1 + args = [target, source] + opts = {} + priority = None + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + self.session.build.return_value = task_id + # Run it and check immediate output + # args: target, http://scm + # expected: success + rv = cli.handle_build(self.options, self.session, args) + actual = stdout.getvalue() + expected = """Created task: 1 +Task info: weburl/taskinfo?taskID=1 +""" + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_not_called() + running_in_bg_mock.assert_called_once() + self.session.uploadWrapper.assert_not_called() + self.session.build.assert_called_once_with(source, target, opts, priority=priority) + self.session.logout.assert_called() + watch_tasks_mock.assert_called_once_with(self.session, [task_id], quiet=self.options.quiet) + self.assertEqual(rv, 0) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('sys.stderr', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_no_arg( + self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, activate_session_mock, stderr, stdout): + args = [] + progname = os.path.basename(sys.argv[0]) or 'koji' + + # Run it and check immediate output + with self.assertRaises(SystemExit) as cm: + cli.handle_build(self.options, self.session, args) + actual_stdout = stdout.getvalue() + actual_stderr = stderr.getvalue() + expected_stdout = '' + expected_stderr = """Usage: %s build [options] target +(Specify the --help global option for a list of other help options) + +%s: error: Exactly two arguments (a build target and a SCM URL or srpm file) are required +""" % (progname, progname) + self.assertMultiLineEqual(actual_stdout, expected_stdout) + self.assertMultiLineEqual(actual_stderr, expected_stderr) + + # Finally, assert that things were called as we expected. + activate_session_mock.assert_not_called() + self.session.getBuildTarget.assert_not_called() + self.session.getTag.assert_not_called() + unique_path_mock.assert_not_called() + running_in_bg_mock.assert_not_called() + self.session.uploadWrapper.assert_not_called() + self.session.build.assert_not_called() + self.session.logout.assert_not_called() + watch_tasks_mock.assert_not_called() + self.assertEqual(cm.exception.code, 2) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('sys.stderr', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_help( + self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, activate_session_mock, stderr, stdout): + args = ['--help'] + progname = os.path.basename(sys.argv[0]) or 'koji' + + # Run it and check immediate output + with self.assertRaises(SystemExit) as cm: + cli.handle_build(self.options, self.session, args) + actual_stdout = stdout.getvalue() + actual_stderr = stderr.getvalue() + expected_stdout = """Usage: %s build [options] target +(Specify the --help global option for a list of other help options) + +Options: + -h, --help show this help message and exit + --skip-tag Do not attempt to tag package + --scratch Perform a scratch build + --wait Wait on the build, even if running in the background + --nowait Don't wait on build + --quiet Do not print the task information + --arch-override=ARCH_OVERRIDE + Override build arches + --repo-id=REPO_ID Use a specific repo + --noprogress Do not display progress of the upload + --background Run the build at a lower priority +""" % progname + expected_stderr = '' + self.assertMultiLineEqual(actual_stdout, expected_stdout) + self.assertMultiLineEqual(actual_stderr, expected_stderr) + + # Finally, assert that things were called as we expected. + activate_session_mock.assert_not_called() + self.session.getBuildTarget.assert_not_called() + self.session.getTag.assert_not_called() + unique_path_mock.assert_not_called() + running_in_bg_mock.assert_not_called() + self.session.uploadWrapper.assert_not_called() + self.session.build.assert_not_called() + self.session.logout.assert_not_called() + watch_tasks_mock.assert_not_called() + self.assertEqual(cm.exception.code, 0) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('sys.stderr', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_arch_override_denied( + self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, activate_session_mock, stderr, stdout): + target = 'target' + source = 'http://scm' + arch_override = 'somearch' + args = [target, source, '--arch-override=' + arch_override] + progname = os.path.basename(sys.argv[0]) or 'koji' + + # Run it and check immediate output + with self.assertRaises(SystemExit) as cm: + cli.handle_build(self.options, self.session, args) + actual_stdout = stdout.getvalue() + actual_stderr = stderr.getvalue() + expected_stdout = '' + expected_stderr = """Usage: %s build [options] target +(Specify the --help global option for a list of other help options) + +%s: error: --arch_override is only allowed for --scratch builds +""" % (progname, progname) + self.assertMultiLineEqual(actual_stdout, expected_stdout) + self.assertMultiLineEqual(actual_stderr, expected_stderr) + + # Finally, assert that things were called as we expected. + activate_session_mock.assert_not_called() + self.session.getBuildTarget.assert_not_called() + self.session.getTag.assert_not_called() + unique_path_mock.assert_not_called() + running_in_bg_mock.assert_not_called() + self.session.uploadWrapper.assert_not_called() + self.session.build.assert_not_called() + self.session.logout.assert_not_called() + watch_tasks_mock.assert_not_called() + self.assertEqual(cm.exception.code, 2) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_none_tag(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stdout): + target = 'nOne' + source = 'http://scm' + task_id = 1 + repo_id = 2 + args = ['--repo-id=' + str(repo_id), target, source] + opts = {'repo_id': repo_id, 'skip_tag': True} + priority = None + + self.session.build.return_value = task_id + # Run it and check immediate output + # args: --repo-id=2, nOne, http://scm + # expected: success + rv = cli.handle_build(self.options, self.session, args) + actual = stdout.getvalue() + expected = """Created task: 1 +Task info: weburl/taskinfo?taskID=1 +""" + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_not_called() + self.session.getTag.assert_not_called() + unique_path_mock.assert_not_called() + running_in_bg_mock.assert_called_once() + self.session.uploadWrapper.assert_not_called() + # target==None, repo_id==2, skip_tag==True + self.session.build.assert_called_once_with(source, None, opts, priority=priority) + self.session.logout.assert_called() + watch_tasks_mock.assert_called_once_with(self.session, [task_id], quiet=self.options.quiet) + self.assertEqual(rv, 0) + + @mock.patch('sys.stderr', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_target_not_found(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stderr): + target = 'target' + target_info = None + source = 'http://scm' + args = [target, source] + + progname = os.path.basename(sys.argv[0]) or 'koji' + + self.session.getBuildTarget.return_value = target_info + # Run it and check immediate output + # args: target, http://scm + # expected: failed, target not found + with self.assertRaises(SystemExit) as cm: + cli.handle_build(self.options, self.session, args) + actual = stderr.getvalue() + expected = """Usage: %s build [options] target +(Specify the --help global option for a list of other help options) + +%s: error: Unknown build target: target +""" % (progname, progname) + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_not_called() + unique_path_mock.assert_not_called() + running_in_bg_mock.assert_not_called() + self.session.uploadWrapper.assert_not_called() + self.session.build.assert_not_called() + self.session.logout.assert_not_called() + watch_tasks_mock.assert_not_called() + self.assertEqual(cm.exception.code, 2) + + @mock.patch('sys.stderr', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_dest_tag_not_found(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stderr): + target = 'target' + dest_tag = 'dest_tag' + dest_tag_name = 'dest_tag_name' + target_info = {'dest_tag': dest_tag, 'dest_tag_name': dest_tag_name} + dest_tag_info = None + source = 'http://scm' + args = [target, source] + + progname = os.path.basename(sys.argv[0]) or 'koji' + + # Mock out the xmlrpc server + self.session = mock.MagicMock() + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + # Run it and check immediate output + # args: target, http://scm + # expected: failed, dest_tag not found + with self.assertRaises(SystemExit) as cm: + cli.handle_build(self.options, self.session, args) + actual = stderr.getvalue() + expected = """Usage: %s build [options] target +(Specify the --help global option for a list of other help options) + +%s: error: Unknown destination tag: dest_tag_name +""" % (progname, progname) + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_not_called() + running_in_bg_mock.assert_not_called() + self.session.uploadWrapper.assert_not_called() + self.session.build.assert_not_called() + self.session.logout.assert_not_called() + watch_tasks_mock.assert_not_called() + self.assertEqual(cm.exception.code, 2) + + @mock.patch('sys.stderr', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_dest_tag_locked(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stderr): + target = 'target' + dest_tag = 'dest_tag' + dest_tag_name = 'dest_tag_name' + target_info = {'dest_tag': dest_tag, 'dest_tag_name': dest_tag_name} + dest_tag_info = {'name': 'dest_tag_name', 'locked': True} + source = 'http://scm' + args = [target, source] + + progname = os.path.basename(sys.argv[0]) or 'koji' + + # Mock out the xmlrpc server + self.session = mock.MagicMock() + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + # Run it and check immediate output + # args: target, http://scm + # expected: failed, dest_tag is locked + with self.assertRaises(SystemExit) as cm: + cli.handle_build(self.options, self.session, args) + actual = stderr.getvalue() + expected = """Usage: %s build [options] target +(Specify the --help global option for a list of other help options) + +%s: error: Destination tag dest_tag_name is locked +""" % (progname, progname) + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_not_called() + running_in_bg_mock.assert_not_called() + self.session.uploadWrapper.assert_not_called() + self.session.build.assert_not_called() + self.session.logout.assert_not_called() + watch_tasks_mock.assert_not_called() + self.assertEqual(cm.exception.code, 2) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_arch_override(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stdout): + target = 'target' + dest_tag = 'dest_tag' + target_info = {'dest_tag': dest_tag} + dest_tag_info = {'locked': False} + source = 'http://scm' + task_id = 1 + arch_override = 'somearch' + args = ['--arch-override=' + arch_override, '--scratch', target, source] + opts = {'arch_override': arch_override, 'scratch': True} + priority = None + + # Mock out the xmlrpc server + self.session = mock.MagicMock() + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + self.session.build.return_value = task_id + # Run it and check immediate output + # args: --arch-override=somearch, --scratch, target, http://scm + # expected: success + rv = cli.handle_build(self.options, self.session, args) + actual = stdout.getvalue() + expected = """Created task: 1 +Task info: weburl/taskinfo?taskID=1 +""" + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_not_called() + running_in_bg_mock.assert_called_once() + self.session.uploadWrapper.assert_not_called() + # arch-override=='somearch', scratch==True + self.session.build.assert_called_once_with(source, target, opts, priority=priority) + self.session.logout.assert_called() + watch_tasks_mock.assert_called_once_with(self.session, [task_id], quiet=self.options.quiet) + self.assertEqual(rv, 0) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_background(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stdout): + target = 'target' + dest_tag = 'dest_tag' + target_info = {'dest_tag': dest_tag} + dest_tag_info = {'locked': False} + source = 'http://scm' + task_id = 1 + args = ['--background', target, source] + priority = 5 + opts = {} + + # Mock out the xmlrpc server + self.session = mock.MagicMock() + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + self.session.build.return_value = task_id + # Run it and check immediate output + # args: --background, target, http://scm + # expected: success + rv = cli.handle_build(self.options, self.session, args) + actual = stdout.getvalue() + expected = """Created task: 1 +Task info: weburl/taskinfo?taskID=1 +""" + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_not_called() + running_in_bg_mock.assert_called_once() + self.session.uploadWrapper.assert_not_called() + self.session.build.assert_called_once_with(source, target, opts, priority=priority) + self.session.logout.assert_called() + watch_tasks_mock.assert_called_once_with(self.session, [task_id], quiet=self.options.quiet) + self.assertEqual(rv, 0) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=True) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_running_in_bg(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stdout): + target = 'target' + dest_tag = 'dest_tag' + target_info = {'dest_tag': dest_tag} + dest_tag_info = {'locked': False} + source = 'srpm' + task_id = 1 + args = [target, source] + opts = {} + priority = None + + # Mock out the xmlrpc server + self.session = mock.MagicMock() + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + self.session.build.return_value = task_id + # Run it and check immediate output + # args: target, srpm + # expected: success + rv = cli.handle_build(self.options, self.session, args) + actual = stdout.getvalue() + expected = """Uploading srpm: srpm + +Created task: 1 +Task info: weburl/taskinfo?taskID=1 +""" + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_called_once_with('cli-build') + self.assertEqual(running_in_bg_mock.call_count, 2) + # callback==None + self.session.uploadWrapper.assert_called_once_with(source, 'random_path', callback=None) + self.session.build.assert_called_once_with('random_path/' + source, target, opts, priority=priority) + self.session.logout.assert_not_called() + watch_tasks_mock.assert_not_called() + self.assertIsNone(rv) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_noprogress(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stdout): + target = 'target' + dest_tag = 'dest_tag' + target_info = {'dest_tag': dest_tag} + dest_tag_info = {'locked': False} + source = 'srpm' + task_id = 1 + args = ['--noprogress', target, source] + opts = {} + priority = None + + # Mock out the xmlrpc server + self.session = mock.MagicMock() + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + self.session.build.return_value = task_id + # Run it and check immediate output + # args: --noprogress, target, srpm + # expected: success + rv = cli.handle_build(self.options, self.session, args) + actual = stdout.getvalue() + expected = """Uploading srpm: srpm + +Created task: 1 +Task info: weburl/taskinfo?taskID=1 +""" + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_called_once_with('cli-build') + self.assertEqual(running_in_bg_mock.call_count, 2) + # callback==None + self.session.uploadWrapper.assert_called_once_with(source, 'random_path', callback=None) + self.session.build.assert_called_once_with('random_path/' + source, target, opts, priority=priority) + self.session.logout.assert_called_once() + watch_tasks_mock.assert_called_once_with(self.session, [task_id], quiet=self.options.quiet) + self.assertEqual(rv, 0) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_quiet(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stdout): + target = 'target' + dest_tag = 'dest_tag' + target_info = {'dest_tag': dest_tag} + dest_tag_info = {'locked': False} + source = 'srpm' + task_id = 1 + quiet = True + args = ['--quiet', target, source] + opts = {} + priority = None + + # Mock out the xmlrpc server + self.session = mock.MagicMock() + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + self.session.build.return_value = task_id + # Run it and check immediate output + # args: --quiet, target, srpm + # expected: success + rv = cli.handle_build(self.options, self.session, args) + actual = stdout.getvalue() + expected = '\n' + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_called_once_with('cli-build') + self.assertEqual(running_in_bg_mock.call_count, 2) + # callback==None + self.session.uploadWrapper.assert_called_once_with(source, 'random_path', callback=None) + self.session.build.assert_called_once_with('random_path/' + source, target, opts, priority=priority) + self.session.logout.assert_called_once() + watch_tasks_mock.assert_called_once_with(self.session, [task_id], quiet=quiet) + self.assertEqual(rv, 0) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_wait(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stdout): + target = 'target' + dest_tag = 'dest_tag' + target_info = {'dest_tag': dest_tag} + dest_tag_info = {'locked': False} + source = 'srpm' + task_id = 1 + quiet = None + args = ['--wait', target, source] + opts = {} + priority = None + + # Mock out the xmlrpc server + self.session = mock.MagicMock() + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + self.session.build.return_value = task_id + # Run it and check immediate output + # args: --wait, target, srpm + # expected: success + rv = cli.handle_build(self.options, self.session, args) + actual = stdout.getvalue() + expected = """Uploading srpm: srpm + +Created task: 1 +Task info: weburl/taskinfo?taskID=1 +""" + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_called_once_with('cli-build') + # the second one won't be executed when wait==False + self.assertEqual(running_in_bg_mock.call_count, 1) + self.session.uploadWrapper.assert_called_once_with(source, 'random_path', callback=cli._progress_callback) + self.session.build.assert_called_once_with('random_path/' + source, target, opts, priority=priority) + self.session.logout.assert_called_once() + watch_tasks_mock.assert_called_once_with(self.session, [task_id], quiet=self.options.quiet) + self.assertEqual(rv, 0) + + @mock.patch('sys.stdout', new_callable=stringio.StringIO) + @mock.patch('koji_cli.activate_session') + @mock.patch('koji_cli._unique_path', return_value='random_path') + @mock.patch('koji_cli._running_in_bg', return_value=False) + @mock.patch('koji_cli.watch_tasks', return_value=0) + def test_handle_build_nowait(self, watch_tasks_mock, running_in_bg_mock, unique_path_mock, + activate_session_mock, stdout): + target = 'target' + dest_tag = 'dest_tag' + target_info = {'dest_tag': dest_tag} + dest_tag_info = {'locked': False} + source = 'srpm' + task_id = 1 + args = ['--nowait', target, source] + opts = {} + priority = None + + # Mock out the xmlrpc server + self.session = mock.MagicMock() + + self.session.getBuildTarget.return_value = target_info + self.session.getTag.return_value = dest_tag_info + self.session.build.return_value = task_id + # Run it and check immediate output + # args: --nowait, target, srpm + # expected: success + rv = cli.handle_build(self.options, self.session, args) + actual = stdout.getvalue() + expected = """Uploading srpm: srpm + +Created task: 1 +Task info: weburl/taskinfo?taskID=1 +""" + self.assertMultiLineEqual(actual, expected) + # Finally, assert that things were called as we expected. + activate_session_mock.assert_called_once_with(self.session) + self.session.getBuildTarget.assert_called_once_with(target) + self.session.getTag.assert_called_once_with(dest_tag) + unique_path_mock.assert_called_once_with('cli-build') + # the second one won't be executed when wait==False + self.assertEqual(running_in_bg_mock.call_count, 1) + self.session.uploadWrapper.assert_called_once_with(source, 'random_path', callback=cli._progress_callback) + self.session.build.assert_called_once_with('random_path/' + source, target, opts, priority=priority) + self.session.logout.assert_not_called() + watch_tasks_mock.assert_not_called() + self.assertIsNone(rv) + + +if __name__ == '__main__': + unittest.main()