def test_handle_remove_pkg(self, activate_session_mock, stderr): tag = 'tag' dsttag = {'name': tag, 'id': 1} package = 'package' args = [tag, package] kwargs = {'force': None} options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getTag.return_value = dsttag session.listPackages.return_value = [ {'package_name': package, 'package_id': 1}] # Run it and check immediate output # args: tag, package # expected: success handle_remove_pkg(options, session, args) actual = stderr.getvalue() expected = '' self.assertMultiLineEqual(actual, expected) # Finally, assert that things were called as we expected. activate_session_mock.assert_called_once_with(session, options) session.getTag.assert_called_once_with(tag) session.listPackages.assert_called_once_with( tagID=dsttag['id']) session.packageListRemove.assert_called_once_with( tag, package, **kwargs) session.multiCall.assert_called_once_with(strict=True)
def test_handle_remove_pkg_help( self, activate_session_mock, stderr, stdout): args = [] options = mock.MagicMock() progname = os.path.basename(sys.argv[0]) or 'koji' # Mock out the xmlrpc server session = mock.MagicMock() # Run it and check immediate output with self.assertRaises(SystemExit) as ex: handle_remove_pkg(options, session, args) self.assertExitCode(ex, 2) actual_stdout = stdout.getvalue() actual_stderr = stderr.getvalue() expected_stdout = '' expected_stderr = """Usage: %s remove-pkg [options] <tag> <package> [<package> ...] (Specify the --help global option for a list of other help options) %s: error: Please specify a tag and at least one package """ % (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() session.getTag.assert_not_called() session.listPackages.assert_not_called() session.packageListRemove.assert_not_called()
def test_handle_remove_pkg_no_package(self, activate_session_mock, stderr): tag = 'tag' dsttag = {'name': tag, 'id': 1} packages = ['package1', 'package2', 'package3'] args = [tag] + packages options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getTag.return_value = dsttag session.listPackages.return_value = [ {'package_name': 'package1', 'package_id': 1}, {'package_name': 'package3', 'package_id': 3}, {'package_name': 'other_package', 'package_id': 4}] # Run it and check immediate output # args: tag, package1, package2, package3 # expected: failed: can not find package2 under tag with self.assertRaises(SystemExit) as ex: handle_remove_pkg(options, session, args) self.assertExitCode(ex, 1) actual = stderr.getvalue() expected = 'Package package2 is not in tag tag\n' self.assertMultiLineEqual(actual, expected) # Finally, assert that things were called as we expected. activate_session_mock.assert_called_once_with(session, options) session.getTag.assert_called_once_with(tag) session.listPackages.assert_called_once_with( tagID=dsttag['id']) session.packageListRemove.assert_not_called() session.multiCall.assert_not_called()
def test_handle_remove_pkg_tag_no_exists( self, activate_session_mock, stderr): tag = 'tag' dsttag = None packages = ['package1', 'package2', 'package3'] args = [tag] + packages options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getTag.return_value = dsttag # Run it and check immediate output # args: tag, package1, package2, package3 # expected: failed: tag does not exist with self.assertRaises(SystemExit) as ex: handle_remove_pkg(options, session, args) self.assertExitCode(ex, 1) actual = stderr.getvalue() expected = 'No such tag: tag\n' self.assertMultiLineEqual(actual, expected) # Finally, assert that things were called as we expected. activate_session_mock.assert_called_once_with(session, options) session.getTag.assert_called_once_with(tag) session.listPackages.assert_not_called() session.packageListRemove.assert_not_called()
def test_handle_remove_pkg_parameter_error(self, stderr): tag = 'tag' dsttag = {'name': tag, 'id': 1} package = 'package' args = [tag, package] kwargs = {'force': None} self.session.getTag.return_value = dsttag self.session.listPackages.side_effect = [ koji.ParameterError, [{ 'package_name': package, 'package_id': 1 }] ] # Run it and check immediate output # args: tag, package # expected: success handle_remove_pkg(self.options, self.session, args) actual = stderr.getvalue() expected = '' self.assertMultiLineEqual(actual, expected) # Finally, assert that things were called as we expected. self.activate_session_mock.assert_called_once_with( self.session, self.options) self.session.getTag.assert_called_once_with(tag) self.session.listPackages.assert_has_calls([ call(tagID=dsttag['id'], with_owners=False), call(tagID=dsttag['id']) ]) self.session.packageListRemove.assert_called_once_with( tag, package, **kwargs) self.session.multiCall.assert_called_once_with(strict=True)
def test_handle_remove_pkg_help( self, activate_session_mock, stderr, stdout): args = [] options = mock.MagicMock() progname = os.path.basename(sys.argv[0]) or 'koji' # Mock out the xmlrpc server session = mock.MagicMock() # Run it and check immediate output with self.assertRaises(SystemExit) as cm: handle_remove_pkg(options, session, args) actual_stdout = stdout.getvalue() actual_stderr = stderr.getvalue() expected_stdout = '' expected_stderr = """Usage: %s remove-pkg [options] tag package [package2 ...] (Specify the --help global option for a list of other help options) %s: error: Please specify a tag and at least one package """ % (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() session.getTag.assert_not_called() session.listPackages.assert_not_called() session.packageListRemove.assert_not_called() if isinstance(cm.exception, int): self.assertEqual(cm.exception, 2) else: self.assertEqual(cm.exception.code, 2)
def test_handle_remove_pkg(self, activate_session_mock, stdout): tag = 'tag' dsttag = {'name': tag, 'id': 1} package = 'package' args = [tag, package] kwargs = {'force': None} options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getTag.return_value = dsttag session.listPackages.return_value = [ {'package_name': package, 'package_id': 1}] # Run it and check immediate output # args: tag, package # expected: success rv = handle_remove_pkg(options, session, args) actual = stdout.getvalue() expected = '' self.assertMultiLineEqual(actual, expected) # Finally, assert that things were called as we expected. activate_session_mock.assert_called_once_with(session, options) session.getTag.assert_called_once_with(tag) session.listPackages.assert_called_once_with( tagID=dsttag['id']) session.packageListRemove.assert_called_once_with( tag, package, **kwargs) session.multiCall.assert_called_once_with(strict=True) self.assertNotEqual(rv, 1)
def test_handle_remove_pkg_tag_no_exists( self, activate_session_mock, stdout): tag = 'tag' dsttag = None packages = ['package1', 'package2', 'package3'] args = [tag] + packages options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getTag.return_value = dsttag # Run it and check immediate output # args: tag, package1, package2, package3 # expected: failed: tag does not exist rv = handle_remove_pkg(options, session, args) actual = stdout.getvalue() expected = 'No such tag: tag\n' self.assertMultiLineEqual(actual, expected) # Finally, assert that things were called as we expected. activate_session_mock.assert_called_once_with(session, options) session.getTag.assert_called_once_with(tag) session.listPackages.assert_not_called() session.packageListRemove.assert_not_called() self.assertEqual(rv, 1)
def test_handle_remove_pkg_no_package(self, activate_session_mock, stdout): tag = 'tag' dsttag = {'name': tag, 'id': 1} packages = ['package1', 'package2', 'package3'] args = [tag] + packages options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getTag.return_value = dsttag session.listPackages.return_value = [ {'package_name': 'package1', 'package_id': 1}, {'package_name': 'package3', 'package_id': 3}, {'package_name': 'other_package', 'package_id': 4}] # Run it and check immediate output # args: tag, package1, package2, package3 # expected: failed: can not find package2 under tag rv = handle_remove_pkg(options, session, args) actual = stdout.getvalue() expected = 'Package package2 is not in tag tag\n' self.assertMultiLineEqual(actual, expected) # Finally, assert that things were called as we expected. activate_session_mock.assert_called_once_with(session, options) session.getTag.assert_called_once_with(tag) session.listPackages.assert_called_once_with( tagID=dsttag['id']) session.packageListRemove.assert_not_called() session.multiCall.assert_not_called() self.assertEqual(rv, 1)
def test_handle_remove_pkg_force(self, activate_session_mock, stdout): tag = 'tag' dsttag = {'name': tag, 'id': 1} packages = ['package1', 'package2', 'package3'] args = ['--force', tag] + packages kwargs = {'force': True} options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getTag.return_value = dsttag session.listPackages.return_value = [{ 'package_name': 'package1', 'package_id': 1 }, { 'package_name': 'package2', 'package_id': 2 }, { 'package_name': 'package3', 'package_id': 3 }, { 'package_name': 'other_package', 'package_id': 4 }] # Run it and check immediate output # args: --force, tag, package1, package2, package3 # expected: success rv = handle_remove_pkg(options, session, args) actual = stdout.getvalue() expected = '' self.assertMultiLineEqual(actual, expected) # Finally, assert that things were called as we expected. activate_session_mock.assert_called_once_with(session, options) self.assertEqual(session.mock_calls, [ call.getTag(tag), call.listPackages(tagID=dsttag['id']), call.packageListRemove(tag, packages[0], **kwargs), call.packageListRemove(tag, packages[1], **kwargs), call.packageListRemove(tag, packages[2], **kwargs), call.multiCall(strict=True) ]) self.assertNotEqual(rv, 1)
def test_handle_remove_pkg_force(self, activate_session_mock, stdout): tag = 'tag' dsttag = {'name': tag, 'id': 1} packages = ['package1', 'package2', 'package3'] args = ['--force', tag] + packages kwargs = {'force': True} options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getTag.return_value = dsttag session.listPackages.return_value = [ {'package_name': 'package1', 'package_id': 1}, {'package_name': 'package2', 'package_id': 2}, {'package_name': 'package3', 'package_id': 3}, {'package_name': 'other_package', 'package_id': 4} ] # Run it and check immediate output # args: --force, tag, package1, package2, package3 # expected: success rv = handle_remove_pkg(options, session, args) actual = stdout.getvalue() expected = '' self.assertMultiLineEqual(actual, expected) # Finally, assert that things were called as we expected. activate_session_mock.assert_called_once_with(session, options) self.assertEqual( session.mock_calls, [ call.getTag(tag), call.listPackages( tagID=dsttag['id']), call.packageListRemove( tag, packages[0], **kwargs), call.packageListRemove( tag, packages[1], **kwargs), call.packageListRemove( tag, packages[2], **kwargs), call.multiCall( strict=True)]) self.assertNotEqual(rv, 1)