| #!/usr/bin/env python |
| |
| # Copyright 2012 Hewlett-Packard Development Company, L.P. |
| # Copyright 2014 Wikimedia Foundation Inc. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); you may |
| # not use this file except in compliance with the License. You may obtain |
| # a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| # License for the specific language governing permissions and limitations |
| # under the License. |
| |
| import logging |
| import os |
| import shutil |
| |
| import git |
| |
| import zuul.lib.cloner |
| |
| from tests.base import ZuulTestCase |
| from tests.base import FIXTURE_DIR |
| |
| logging.basicConfig(level=logging.DEBUG, |
| format='%(asctime)s %(name)-32s ' |
| '%(levelname)-8s %(message)s') |
| |
| |
| class TestCloner(ZuulTestCase): |
| |
| log = logging.getLogger("zuul.test.cloner") |
| workspace_root = None |
| |
| def setUp(self): |
| super(TestCloner, self).setUp() |
| self.workspace_root = os.path.join(self.test_root, 'workspace') |
| |
| self.config.set('zuul', 'layout_config', |
| 'tests/fixtures/layout-cloner.yaml') |
| self.sched.reconfigure(self.config) |
| self.registerJobs() |
| |
| def getWorkspaceRepos(self, projects): |
| repos = {} |
| for project in projects: |
| repos[project] = git.Repo( |
| os.path.join(self.workspace_root, project)) |
| return repos |
| |
| def getUpstreamRepos(self, projects): |
| repos = {} |
| for project in projects: |
| repos[project] = git.Repo( |
| os.path.join(self.upstream_root, project)) |
| return repos |
| |
| def test_one_branch(self): |
| self.worker.hold_jobs_in_build = True |
| |
| projects = ['org/project1', 'org/project2'] |
| A = self.fake_gerrit.addFakeChange('org/project1', 'master', 'A') |
| B = self.fake_gerrit.addFakeChange('org/project2', 'master', 'B') |
| A.addApproval('CRVW', 2) |
| B.addApproval('CRVW', 2) |
| self.fake_gerrit.addEvent(A.addApproval('APRV', 1)) |
| self.fake_gerrit.addEvent(B.addApproval('APRV', 1)) |
| |
| self.waitUntilSettled() |
| |
| self.assertEquals(2, len(self.builds), "Two builds are running") |
| |
| upstream = self.getUpstreamRepos(projects) |
| states = [ |
| {'org/project1': self.builds[0].parameters['ZUUL_COMMIT'], |
| 'org/project2': str(upstream['org/project2'].commit('master')), |
| }, |
| {'org/project1': self.builds[0].parameters['ZUUL_COMMIT'], |
| 'org/project2': self.builds[1].parameters['ZUUL_COMMIT'], |
| }, |
| ] |
| |
| for number, build in enumerate(self.builds): |
| self.log.debug("Build parameters: %s", build.parameters) |
| cloner = zuul.lib.cloner.Cloner( |
| git_base_url=self.upstream_root, |
| projects=projects, |
| workspace=self.workspace_root, |
| zuul_branch=build.parameters['ZUUL_BRANCH'], |
| zuul_ref=build.parameters['ZUUL_REF'], |
| zuul_url=self.git_root, |
| ) |
| cloner.execute() |
| work = self.getWorkspaceRepos(projects) |
| state = states[number] |
| |
| for project in projects: |
| self.assertEquals(state[project], |
| str(work[project].commit('HEAD')), |
| 'Project %s commit for build %s should ' |
| 'be correct' % (project, number)) |
| |
| shutil.rmtree(self.workspace_root) |
| |
| self.worker.hold_jobs_in_build = False |
| self.worker.release() |
| self.waitUntilSettled() |
| |
| def test_multi_branch(self): |
| self.worker.hold_jobs_in_build = True |
| projects = ['org/project1', 'org/project2', |
| 'org/project3', 'org/project4'] |
| |
| self.create_branch('org/project2', 'stable/havana') |
| self.create_branch('org/project4', 'stable/havana') |
| A = self.fake_gerrit.addFakeChange('org/project1', 'master', 'A') |
| B = self.fake_gerrit.addFakeChange('org/project2', 'stable/havana', 'B') |
| C = self.fake_gerrit.addFakeChange('org/project3', 'master', 'C') |
| A.addApproval('CRVW', 2) |
| B.addApproval('CRVW', 2) |
| C.addApproval('CRVW', 2) |
| self.fake_gerrit.addEvent(A.addApproval('APRV', 1)) |
| self.fake_gerrit.addEvent(B.addApproval('APRV', 1)) |
| self.fake_gerrit.addEvent(C.addApproval('APRV', 1)) |
| |
| self.waitUntilSettled() |
| |
| self.assertEquals(3, len(self.builds), "Three builds are running") |
| |
| upstream = self.getUpstreamRepos(projects) |
| states = [ |
| {'org/project1': self.builds[0].parameters['ZUUL_COMMIT'], |
| 'org/project2': str(upstream['org/project2'].commit('master')), |
| 'org/project3': str(upstream['org/project3'].commit('master')), |
| 'org/project4': str(upstream['org/project4']. |
| commit('master')), |
| }, |
| {'org/project1': self.builds[0].parameters['ZUUL_COMMIT'], |
| 'org/project2': self.builds[1].parameters['ZUUL_COMMIT'], |
| 'org/project3': str(upstream['org/project3'].commit('master')), |
| 'org/project4': str(upstream['org/project4']. |
| commit('stable/havana')), |
| }, |
| {'org/project1': self.builds[0].parameters['ZUUL_COMMIT'], |
| 'org/project2': str(upstream['org/project2'].commit('master')), |
| 'org/project3': self.builds[2].parameters['ZUUL_COMMIT'], |
| 'org/project4': str(upstream['org/project4']. |
| commit('master')), |
| }, |
| ] |
| |
| for number, build in enumerate(self.builds): |
| self.log.debug("Build parameters: %s", build.parameters) |
| cloner = zuul.lib.cloner.Cloner( |
| git_base_url=self.upstream_root, |
| projects=projects, |
| workspace=self.workspace_root, |
| zuul_branch=build.parameters['ZUUL_BRANCH'], |
| zuul_ref=build.parameters['ZUUL_REF'], |
| zuul_url=self.git_root, |
| ) |
| cloner.execute() |
| work = self.getWorkspaceRepos(projects) |
| state = states[number] |
| |
| for project in projects: |
| self.assertEquals(state[project], |
| str(work[project].commit('HEAD')), |
| 'Project %s commit for build %s should ' |
| 'be correct' % (project, number)) |
| shutil.rmtree(self.workspace_root) |
| |
| self.worker.hold_jobs_in_build = False |
| self.worker.release() |
| self.waitUntilSettled() |