dsttag = session.getTag(tag)

      if not dsttag:

      groups = dict([(p['name'], p['group_id']) for p in session.getTagGroups(tag, inherit=False)])

      group_id = groups.get(group, None)

      if group_id is not None:

      session.groupListAdd(tag, group)

      dsttag = session.getTag(tag)

      if not dsttag:

      groups = dict([(p['name'], p['group_id']) for p in session.getTagGroups(tag, inherit=False)])

      group_id = groups.get(group, None)

      if group_id is None:

      session.groupListBlock(tag, group)

      session.multicall = True

      for host in args:

          session.getHost(host)

      for host, [info] in zip(args, session.multiCall(strict=True)):

          if not info:

      session.multicall = True

      for host in args:

      if not options.new:

          channelinfo = session.getChannel(channel)

          if not channelinfo:

      host = args[0]

      hostinfo = session.getHost(host)

      if not hostinfo:

      kwargs = {}

      if options.new:

          kwargs['create'] = True

      activate_session(session, goptions)

      hostinfo = session.getHost(host)

      if not hostinfo:

      hostchannels = [c['name'] for c in session.listChannels(hostinfo['id'])]

      channel = args[1]

      if channel not in hostchannels:

      session.removeHostFromChannel(host, channel)

      activate_session(session, goptions)

      cinfo = session.getChannel(args[0])

      if not cinfo:

      session.removeChannel(args[0], force=options.force)

      activate_session(session, goptions)

      cinfo = session.getChannel(args[0])

      if not cinfo:

      session.renameChannel(args[0], args[1])

      if not options.owner:

          parser.error(_("Please specify an owner for the package(s)"))

      if not session.getUser(options.owner):

      activate_session(session, goptions)

      tag = args[0]

      opts = {}

      # check if list of packages exists for that tag already

      dsttag = session.getTag(tag)

      if dsttag is None:

      pkglist = dict([(p['package_name'], p['package_id'])

                      for p in session.listPackages(tagID=dsttag['id'], inherited=True)])

      ret = 0

      for package in args[1:]:

          package_id = pkglist.get(package, None)

          if package_id is None:

              ret = 1

      if ret:

      session.multicall = True

      for package in args[1:]:

          # force is not supported on older hub, so use it only explicitly

      # check if list of packages exists for that tag already

      dsttag = session.getTag(tag)

      if dsttag is None:

      pkglist = dict([(p['package_name'], p['package_id'])

                      for p in session.listPackages(tagID=dsttag['id'])])

      ret = 0

      for package in args[1:]:

          package_id = pkglist.get(package, None)

          if package_id is None:

              ret = 1

      if ret:

      session.multicall = True

      for package in args[1:]:

          session.packageListRemove(tag, package, **opts)

      ancestors = session.getFullInheritance(build_target['build_tag'])

      if dest_tag['id'] not in [build_target['build_tag']] + \

              [ancestor['parent_id'] for ancestor in ancestors]:

      sources = args[1:]

      src_list = []

              # quick check that it looks like a N-V-R

              build_level.append(src)

          else:

      if build_level:

          src_list.append(build_level)

              parser.error(_("Task id must be an integer"))

          broots = session.listBuildroots(taskID=task_id)

          if not broots:

          if len(broots) > 1:

              print(_("Multiple buildroots found: %s" % [br['id'] for br in broots]))

          brootinfo = broots[-1]

              options.name = "%s-task_%i" % (opts['tag_name'], task_id)

      elif options.tag:

          if not options.arch:

          tag = session.getTag(options.tag)

          if not tag:

              parser.error(_("Invalid tag: %s" % options.tag))

          arch = options.arch

          config = session.getBuildConfig(tag['id'])

          if not config:

          opts['tag_name'] = tag['name']

          if options.latest:

              opts['repoid'] = 'latest'

          else:

              repo = session.getRepo(config['id'])

              if not repo:

              opts['repoid'] = repo['id']

      elif options.target:

          if not options.arch:

          arch = options.arch

          target = session.getBuildTarget(options.target)

          if not target:

          else:

              repo = session.getRepo(target['build_tag'])

              if not repo:

              opts['repoid'] = repo['id']

      else:

          parser.error(_("Please specify one of: --tag, --target, --task, --buildroot"))

      session.multicall = True

      for host in args:

          session.getHost(host)

      for host, [id] in zip(args, session.multiCall(strict=True)):

          if not id:

              print("Host %s does not exist" % host)

      session.multicall = True

      for host in args:

          session.disableHost(host)

      session.multicall = True

      for host in args:

          session.getHost(host)

      for host, [id] in zip(args, session.multiCall(strict=True)):

          if not id:

              print("Host %s does not exist" % host)

      session.multicall = True

      for host in args:

          session.enableHost(host)

          }

          others = session.listTasks(query)

          if others:

      callopts = {}

      if my_opts.channel:

      # check if the tag exists

      dsttag = session.getTag(args[1])

      if dsttag is None:

      if libcomps is not None:

          _import_comps(session, args[0], args[1], local_options)

      elif yumcomps is not None:

          _import_comps_alt(session, args[0], args[1], local_options)

      else:

  def _import_comps(session, filename, tag, options):

          parser.error(_("You must provide a volume and at least one build"))

      volinfo = session.getVolume(args[0])

      if not volinfo:

      activate_session(session, goptions)

      builds = []

      for nvr in args[1:]:

          else:

              builds.append(binfo)

      if not builds:

      for binfo in builds:

          session.changeBuildVolume(binfo['id'], volinfo['id'])

          if options.verbose:

      name = args[0]

      volinfo = session.getVolume(name)

      if volinfo:

      activate_session(session, goptions)

      volinfo = session.addVolume(name)

      print("Added volume %(name)s with id %(id)i" % volinfo)

      if options.user:

          user = session.getUser(options.user)

          if not user:

          perms = session.getUserPerms(user['id'])

      elif options.mine:

          perms = session.getPerms()

      data = session.listPackages(**opts)

      if not data:

      if not options.quiet:

          if allpkgs:

              print("Package")

      chkbuildtag = session.getTag(build_tag)

      chkdesttag = session.getTag(dest_tag)

      if not chkbuildtag:

      if not chkbuildtag.get("arches", None):

      if not chkdesttag:

      session.createBuildTarget(name, build_tag, dest_tag)

          targetInfo['build_tag_name'] = options.build_tag

          chkbuildtag = session.getTag(options.build_tag)

          if not chkbuildtag:

          if not chkbuildtag.get("arches", None):

      if options.dest_tag:

          chkdesttag = session.getTag(options.dest_tag)

          if not chkdesttag:

          targetInfo['dest_tag_name'] = options.dest_tag

      session.editBuildTarget(targetInfo['orig_name'], targetInfo['name'],

      target = args[0]

      target_info = session.getBuildTarget(target)

      if not target_info:

      session.deleteBuildTarget(target_info['id'])

      tag = args[0]

      tag_info = session.getTag(tag)

      if not tag_info:

      session.deleteTag(tag_info['id'])

          data = [datum for datum in data if datum['priority'] == priority]

      if len(data) == 0:

      elif len(data) > 1:

          print(_("Multiple matches for tag."))

          if not parent:

          if not priority:

      # len(data) == 1

      data = data[0]

      inheritanceData = session.getInheritanceData(tag['id'])

      samePriority = [datum for datum in inheritanceData if datum['priority'] == options.priority]

      if samePriority:

                  "please specify a different priority with --priority.") % tag['name'])

      new_data = data.copy()

      if options.priority is not None and options.priority.isdigit():

          elif options.maxdepth.lower() == "none":

              new_data['maxdepth'] = None

          else:

      if options.intransitive:

          new_data['intransitive'] = options.intransitive

      if options.noconfig:

      if delete:

          # removing entirely

          if current_tags and not options.force:

          session.deleteExternalRepo(args[0])

      else:

          for tag in tags:

      packages = args[1:]

      user = session.getUser(owner)

      if not user:

      opts = {'with_dups': True}

      old_user = None

      if options.old_user:

          old_user = session.getUser(options.old_user)

          if not old_user:

          opts['userID'] = old_user['id']

      to_change = []

      for package in packages:

      if not packages and options.old_user:

          entries = session.listPackages(**opts)

          if not entries:

          to_change.extend(entries)

      for entry in to_change:

          if user['id'] == entry['owner_id']:

                  # not in tag, see if it even exists

                  binfo = session.getBuild(nvr)

                  if not binfo:

                  else:

                  if not options.force:

      builds.reverse()

      for binfo in builds:

          if options.test:

      if build.isdigit():

          if suboptions.latestfrom:

          build = int(build)

          if suboptions.task_id:

              builds = session.listBuilds(taskID=build)

              parser.error(_("Invalid tag: %s") % tag)

          targets = session.getBuildTargets(buildTagID=tag_info['id'])

          if not targets:

              targets = session.getBuildTargets(destTagID=tag_info['id'])

              if targets:

                  maybe = {}.fromkeys([t['build_tag_name'] for t in targets])

                  maybe = sorted(maybe.keys())

          tag_id = tag_info['id']

      for nvr in builds:

          if (time.time() - start) >= (suboptions.timeout * 60.0):

              if not suboptions.quiet:

                  if builds:

                            (koji.util.duration(start), koji.util.printList(suboptions.builds), tag))

                  else:

                            (koji.util.duration(start), tag))

          time.sleep(options.poll_interval)

          last_repo = repo

      if options.user:

          user = session.getUser(options.user)

          if not user:

          user_id = user['id']

      else:

          user_id = None

          session.groupListAdd.assert_called_once_with(tag, group)

          self.assertNotEqual(rv, 1)

      @mock.patch('koji_cli.commands.activate_session')

          tag = 'tag'

          group = 'group'

          arguments = [tag, group]

              {'name': 'group', 'group_id': 'groupId'}]

          # Run it and check immediate output

          expected = 'Group group already exists for tag tag\n'

          self.assertMultiLineEqual(actual, expected)

          session.getTag.assert_called_once_with(tag)

          session.getTagGroups.assert_called_once_with(tag, inherit=False)

          session.groupListAdd.assert_not_called()

      @mock.patch('sys.stdout', new_callable=six.StringIO)

      @mock.patch('sys.stderr', new_callable=six.StringIO)

          session.getTagGroups.assert_not_called()

          session.groupListAdd.assert_not_called()

      @mock.patch('koji_cli.commands.activate_session')

          tag = 'tag'

          group = 'group'

          arguments = [tag, group]

          session.getTag.return_value = None

          # Run it and check immediate output

          expected = 'Unknown tag: tag\n'

          self.assertMultiLineEqual(actual, expected)

          session.getTag.assert_called_once_with(tag)

          session.getTagGroups.assert_not_called()

          session.groupListAdd.assert_not_called()

  if __name__ == '__main__':

  import os

  import six

  import sys

  import koji

  from koji_cli.commands import handle_add_host

      # Show long diffs in error output...

      maxDiff = None

          # Run it and check immediate output

          # args: host, arch1, arch2, --krb-principal=krb

          # expected: failed, host already exists

              handle_add_host(options, session, arguments)

          actual = stderr.getvalue()

          expected = 'host is already in the database\n'

          self.assertMultiLineEqual(actual, expected)

          session = mock.MagicMock()

          # Run it and check immediate output

              handle_add_host(options, session, arguments)

          actual_stdout = stdout.getvalue()

          actual_stderr = stderr.getvalue()

          expected_stdout = ''

          activate_session_mock.assert_not_called()

          session.hasHost.assert_not_called()

          session.addHost.assert_not_called()

      @mock.patch('koji_cli.commands.activate_session')

          host = 'host'

          arches = ['arch1', 'arch2']

          krb_principal = '--krb-principal=krb'

          # Run it and check immediate output

          # args: host, arch1, arch2, --krb-principal=krb

          # expected: failed

              handle_add_host(options, session, arguments)

          expected = ''

          self.assertMultiLineEqual(actual, expected)

          # Finally, assert that things were called as we expected.

          activate_session_mock.assert_called_once_with(session, options)

          session.getHost.assert_called_once_with(host)

          session.addHost.assert_called_once_with(host, arches, **kwargs)

      import unittest

  from koji_cli.commands import handle_add_host_to_channel

      # Show long diffs in error output...

      maxDiff = None

              host, channel, **kwargs)

          self.assertNotEqual(rv, 1)

      @mock.patch('koji_cli.commands.activate_session')

      def test_handle_add_host_to_channel_no_channel(

          host = 'host'

          channel = 'channel'

          channel_info = None

          # Run it and check immediate output

          # args: host, channel

          # expected: failed, channel not found

          expected = 'No such channel: channel\n'

          self.assertMultiLineEqual(actual, expected)

          # Finally, assert that things were called as we expected.

          session.getChannel.assert_called_once_with(channel)

          session.getHost.assert_not_called()

          session.addHostToChannel.assert_not_called()

      @mock.patch('koji_cli.commands.activate_session')

      def test_handle_add_host_to_channel_no_host(

          host = 'host'

          host_info = None

          channel = 'channel'

          # Run it and check immediate output

          # args: host, channel

          # expected: success

          expected = 'No such host: host\n'

          self.assertMultiLineEqual(actual, expected)

          # Finally, assert that things were called as we expected.

          session.getChannel.assert_called_once_with(channel)

          session.getHost.assert_called_once_with(host)

          session.addHostToChannel.assert_not_called()

      @mock.patch('sys.stdout', new_callable=six.StringIO)

      @mock.patch('sys.stderr', new_callable=six.StringIO)

          # Run it and check immediate output

          # args: _empty_

          # expected: failed, help msg shows

              handle_add_host_to_channel(options, session, args)

          actual_stdout = stdout.getvalue()

          actual_stderr = stderr.getvalue()

          expected_stdout = ''

          session.getChannel.assert_not_called()

          session.listChannels.assert_not_called()

          session.addHostToChannel.assert_not_called()

  if __name__ == '__main__':

  from mock import call

  from koji_cli.commands import handle_add_pkg

      # Show long diffs in error output...

      maxDiff = None

                            call.multiCall(strict=True)])

          self.assertNotEqual(rv, 1)

      @mock.patch('koji_cli.commands.activate_session')

      def test_handle_add_pkg_owner_no_exists(

          tag = 'tag'

          packages = ['package1', 'package2', 'package3']

          owner = 'owner'

          # args: --owner, --extra-arches='arch1,arch2 arch3, arch4',

          #       tag, package1, package2, package3

          # expected: failed: owner does not exist

          expected = 'User owner does not exist\n'

          self.assertMultiLineEqual(actual, expected)

          # Finally, assert that things were called as we expected.

          activate_session_mock.assert_not_called()

          self.assertEqual(session.mock_calls,

                           [call.getUser(owner)])

      @mock.patch('sys.stdout', new_callable=six.StringIO)

      @mock.patch('koji_cli.commands.activate_session')

          # args: --owner, --extra-arches='arch1,arch2 arch3, arch4',

          #       tag, package1, package2, package3

          # expected: failed: tag does not exist

              handle_add_pkg(options, session, args)

          actual = stdout.getvalue()

          expected = 'No such tag: tag\n'

          self.assertMultiLineEqual(actual, expected)