def test_handle_add_pkg_owner_no_exists(self, activate_session_mock, stdout): tag = 'tag' packages = ['package1', 'package2', 'package3'] owner = 'owner' owner_info = None extra_arches = 'arch1,arch2 arch3, arch4' args = ['--owner=' + owner, '--extra-arches=' + extra_arches, tag ] + packages options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getUser.return_value = owner_info # Run it and check immediate output # args: --owner, --extra-arches='arch1,arch2 arch3, arch4', # tag, package1, package2, package3 # expected: failed: owner does not exist rv = handle_add_pkg(options, session, args) actual = stdout.getvalue() 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)]) self.assertEqual(rv, 1)
def test_handle_add_pkg_tag_no_exists(self, activate_session_mock, stdout): tag = 'tag' dsttag = None packages = ['package1', 'package2', 'package3'] owner = 'owner' owner_info = mock.ANY extra_arches = 'arch1,arch2 arch3, arch4' args = ['--owner=' + owner, '--extra-arches=' + extra_arches, tag ] + packages options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getUser.return_value = owner_info session.getTag.return_value = dsttag # Run it and check immediate output # args: --owner, --extra-arches='arch1,arch2 arch3, arch4', # tag, package1, package2, package3 # expected: failed: tag does not exist with self.assertRaises(SystemExit) as cm: handle_add_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) self.assertEqual( session.mock_calls, [call.getUser(owner), call.getTag(tag)]) if isinstance(cm.exception, int): self.assertEqual(cm.exception, 1) else: self.assertEqual(cm.exception.code, 1)
def test_handle_add_pkg_tag_no_exists(self): tag = 'tag' dsttag = None packages = ['package1', 'package2', 'package3'] owner = 'owner' owner_info = mock.ANY extra_arches = 'arch1,arch2 arch3, arch4' arguments = [ '--owner=' + owner, '--extra-arches=' + extra_arches, tag ] + packages self.session.getUser.return_value = owner_info self.session.getTag.return_value = dsttag # Run it and check immediate output # args: --owner, --extra-arches='arch1,arch2 arch3, arch4', # tag, package1, package2, package3 # expected: failed: tag does not exist self.assert_system_exit(handle_add_pkg, self.options, self.session, arguments, stdout='', stderr='No such tag: %s\n' % tag, exit_code=1, activate_session=None) # Finally, assert that things were called as we expected. self.activate_session_mock.assert_called_once_with( self.session, self.options) self.assertEqual( self.session.mock_calls, [call.getUser(owner), call.getTag(tag)])
def test_handle_add_pkg_multi_pkg(self, activate_session_mock, stdout): tag = 'tag' dsttag = {'name': tag, 'id': 1} packages = ['package1', 'package2', 'package3'] owner = 'owner' owner_info = mock.ANY extra_arches = 'arch1,arch2 arch3, arch4' args = ['--owner=' + owner, '--extra-arches=' + extra_arches, tag ] + packages kwargs = { 'force': None, 'block': False, 'extra_arches': 'arch1 arch2 arch3 arch4' } options = mock.MagicMock() # Mock out the xmlrpc server session = mock.MagicMock() session.getUser.return_value = owner_info session.getTag.return_value = dsttag session.listPackages.return_value = [{ 'package_name': 'package2', 'package_id': 2 }] # Run it and check immediate output # args: --owner, --extra-arches='arch1,arch2 arch3, arch4', # tag, package1, package2, package3 # expected: success rv = handle_add_pkg(options, session, args) actual = stdout.getvalue() expected = 'Package package2 already exists in tag tag\nAdding 2 packages to 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) self.assertEqual(session.mock_calls, [ call.getUser(owner), call.getTag(tag), call.listPackages(tagID=dsttag['id']), call.packageListAdd(tag, packages[0], owner, **kwargs), call.packageListAdd(tag, packages[2], owner, **kwargs), call.multiCall(strict=True) ]) self.assertNotEqual(rv, 1)