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 <srpm path or scm url>
(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 <srpm path or scm url>
(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 <srpm path or scm url>
(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 <srpm path or scm url>
(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'

        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 <srpm path or scm url>
(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'

        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 <srpm path or scm url>
(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

        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 = {}

        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

        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

        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

        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

        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

        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()