# Copyright (c) 2016 Red Hat, Inc. # # Permission is hereby granted, free of charge, to any person obtaining a copy # of this software and associated documentation files (the "Software"), to deal # in the Software without restriction, including without limitation the rights # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell # copies of the Software, and to permit persons to whom the Software is # furnished to do so, subject to the following conditions: # # The above copyright notice and this permission notice shall be included in all # copies or substantial portions of the Software. # # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE # SOFTWARE. # # Written by Jan Kaluza import unittest import mock import koji import xmlrpclib import module_build_service.messaging import module_build_service.scheduler.handlers.repos import module_build_service.models import module_build_service.builder from mock import patch, MagicMock from tests import conf, init_data from module_build_service.builder.KojiModuleBuilder import KojiModuleBuilder class FakeKojiModuleBuilder(KojiModuleBuilder): @module_build_service.utils.retry(wait_on=(xmlrpclib.ProtocolError, koji.GenericError)) def get_session(self, config, owner): koji_session = MagicMock() koji_session.getRepo.return_value = {'create_event': 'fake event'} def _get_tag(name): _id = 2 if name.endswith("build") else 1 return {"name": name, "id": _id} koji_session.getTag = _get_tag return koji_session class TestKojiBuilder(unittest.TestCase): def setUp(self): init_data() self.config = mock.Mock() self.config.koji_profile = conf.koji_profile self.config.koji_repository_url = conf.koji_repository_url self.module = module_build_service.models.ModuleBuild.query.filter_by(id=1).one() def test_tag_to_repo(self): """ Test that when a repo msg hits us and we have no match, that we do nothing gracefully. """ repo = module_build_service.builder.GenericBuilder.tag_to_repo( "koji", self.config, "module-base-runtime-0.25-9", "x86_64") self.assertEquals(repo, "https://kojipkgs.stg.fedoraproject.org/repos" "/module-base-runtime-0.25-9/latest/x86_64") def test_recover_orphaned_artifact_when_tagged(self): """ Test recover_orphaned_artifact when the artifact is found and tagged in both tags """ builder = FakeKojiModuleBuilder(owner=self.module.owner, module=self.module, config=conf, tag_name='module-foo', components=[]) builder.module_tag = {"name": "module-foo", "id": 1} builder.module_build_tag = {"name": "module-foo-build", "id": 2} # Set listTagged to return test data build_tagged = [{"nvr": "foo-1.0-1.module+e0095747", "task_id": 12345, 'build_id': 91}] dest_tagged = [{"nvr": "foo-1.0-1.module+e0095747", "task_id": 12345, 'build_id': 91}] builder.koji_session.listTagged.side_effect = [build_tagged, dest_tagged] module_build = module_build_service.models.ModuleBuild.query.get(30) component_build = module_build.component_builds[0] component_build.task_id = None component_build.state = None component_build.nvr = None actual = builder.recover_orphaned_artifact(component_build) self.assertEqual(len(actual), 3) self.assertEquals(type(actual[0]), module_build_service.messaging.KojiBuildChange) self.assertEquals(actual[0].build_id, 91) self.assertEquals(actual[0].task_id, 12345) self.assertEquals(actual[0].build_new_state, koji.BUILD_STATES['COMPLETE']) self.assertEquals(actual[0].build_name, 'rubygem-rails') self.assertEquals(actual[0].build_version, '1.0') self.assertEquals(actual[0].build_release, '1.module+e0095747') self.assertEquals(actual[0].module_build_id, 30) self.assertEquals(type(actual[1]), module_build_service.messaging.KojiTagChange) self.assertEquals(actual[1].tag, 'module-foo-build') self.assertEquals(actual[1].artifact, 'rubygem-rails') self.assertEquals(type(actual[2]), module_build_service.messaging.KojiTagChange) self.assertEquals(actual[2].tag, 'module-foo') self.assertEquals(actual[2].artifact, 'rubygem-rails') self.assertEqual(component_build.state, koji.BUILD_STATES['COMPLETE']) self.assertEqual(component_build.task_id, 12345) self.assertEqual(component_build.state_reason, 'Found existing build') self.assertEquals(builder.koji_session.tagBuild.call_count, 0) def test_recover_orphaned_artifact_when_untagged(self): """ Tests recover_orphaned_artifact when the build is found but untagged """ builder = FakeKojiModuleBuilder(owner=self.module.owner, module=self.module, config=conf, tag_name='module-foo', components=[]) builder.module_tag = {"name": "module-foo", "id": 1} builder.module_build_tag = {"name": "module-foo-build", "id": 2} # Set listTagged to return test data builder.koji_session.listTagged.side_effect = [[], [], []] untagged = [{ "id": 9000, "name": "foo", "version": "1.0", "release": "1.module+e0095747", }] builder.koji_session.untaggedBuilds.return_value = untagged build_info = { 'nvr': 'foo-1.0-1.module+e0095747', 'task_id': 12345, 'build_id': 91 } builder.koji_session.getBuild.return_value = build_info module_build = module_build_service.models.ModuleBuild.query.get(30) component_build = module_build.component_builds[0] component_build.task_id = None component_build.nvr = None component_build.state = None actual = builder.recover_orphaned_artifact(component_build) self.assertEqual(len(actual), 1) self.assertEquals(type(actual[0]), module_build_service.messaging.KojiBuildChange) self.assertEquals(actual[0].build_id, 91) self.assertEquals(actual[0].task_id, 12345) self.assertEquals(actual[0].build_new_state, koji.BUILD_STATES['COMPLETE']) self.assertEquals(actual[0].build_name, 'rubygem-rails') self.assertEquals(actual[0].build_version, '1.0') self.assertEquals(actual[0].build_release, '1.module+e0095747') self.assertEquals(actual[0].module_build_id, 30) self.assertEqual(component_build.state, koji.BUILD_STATES['COMPLETE']) self.assertEqual(component_build.task_id, 12345) self.assertEqual(component_build.state_reason, 'Found existing build') builder.koji_session.tagBuild.assert_called_once_with(2, 'foo-1.0-1.module+e0095747') def test_recover_orphaned_artifact_when_nothing_exists(self): """ Test recover_orphaned_artifact when the build is not found """ builder = FakeKojiModuleBuilder(owner=self.module.owner, module=self.module, config=conf, tag_name='module-foo', components=[]) builder.module_tag = {"name": "module-foo", "id": 1} builder.module_build_tag = {"name": "module-foo-build", "id": 2} # Set listTagged to return nothing... tagged = [] builder.koji_session.listTagged.return_value = tagged untagged = [{ "nvr": "foo-1.0-1.nope", "release": "nope", }] builder.koji_session.untaggedBuilds.return_value = untagged module_build = module_build_service.models.ModuleBuild.query.get(30) component_build = module_build.component_builds[0] component_build.task_id = None component_build.nvr = None component_build.state = None actual = builder.recover_orphaned_artifact(component_build) self.assertEqual(actual, []) # Make sure nothing erroneous gets tag self.assertEquals(builder.koji_session.tagBuild.call_count, 0) @patch('koji.util') def test_buildroot_ready(self, mocked_kojiutil): attrs = {'checkForBuilds.return_value': None, 'checkForBuilds.side_effect': IOError} mocked_kojiutil.configure_mock(**attrs) fake_kmb = FakeKojiModuleBuilder(owner=self.module.owner, module=self.module, config=conf, tag_name='module-nginx-1.2', components=[]) fake_kmb.module_target = {'build_tag': 'module-fake_tag'} with self.assertRaises(IOError): fake_kmb.buildroot_ready() self.assertEquals(mocked_kojiutil.checkForBuilds.call_count, 3) def test_tagging_already_tagged_artifacts(self): """ Tests that buildroot_add_artifacts and tag_artifacts do not try to tag already tagged artifacts """ builder = FakeKojiModuleBuilder(owner=self.module.owner, module=self.module, config=conf, tag_name='module-nginx-1.2', components=[]) builder.module_tag = {"name": "module-foo", "id": 1} builder.module_build_tag = {"name": "module-foo-build", "id": 2} # Set listTagged to return test data tagged = [{"nvr": "foo-1.0-1.module_e0095747"}, {"nvr": "bar-1.0-1.module_e0095747"}] builder.koji_session.listTagged.return_value = tagged # Try to tag one artifact which is already tagged and one new ... to_tag = ["foo-1.0-1.module_e0095747", "new-1.0-1.module_e0095747"] builder.buildroot_add_artifacts(to_tag) # ... only new one should be added. builder.koji_session.tagBuild.assert_called_once_with( builder.module_build_tag["id"], "new-1.0-1.module_e0095747") # Try the same for tag_artifacts(...). builder.koji_session.tagBuild.reset_mock() builder.tag_artifacts(to_tag) builder.koji_session.tagBuild.assert_called_once_with( builder.module_tag["id"], "new-1.0-1.module_e0095747") class TestGetKojiClientSession(unittest.TestCase): def setUp(self): init_data() self.config = mock.Mock() self.config.koji_profile = conf.koji_profile self.config.koji_config = conf.koji_config self.module = module_build_service.models.ModuleBuild.query.filter_by(id=1).one() self.tag_name = 'module-fool-1.2' @patch.object(koji.ClientSession, 'krb_login') def test_proxyuser(self, mocked_krb_login): KojiModuleBuilder(owner=self.module.owner, module=self.module, config=self.config, tag_name=self.tag_name, components=[]) args, kwargs = mocked_krb_login.call_args self.assertTrue(set([('proxyuser', self.module.owner)]).issubset(set(kwargs.items())))