271 lines
11 KiB
Python
271 lines
11 KiB
Python
# Copyright 2014 Rackspace Australia
|
|
#
|
|
# 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 sqlalchemy as sa
|
|
|
|
from tests.base import ZuulTestCase, ZuulDBTestCase
|
|
|
|
|
|
def _get_reporter_from_connection_name(reporters, connection_name):
|
|
# Reporters are placed into lists for each action they may exist in.
|
|
# Search through the given list for the correct reporter by its conncetion
|
|
# name
|
|
for r in reporters:
|
|
if r.connection.connection_name == connection_name:
|
|
return r
|
|
|
|
|
|
class TestConnections(ZuulTestCase):
|
|
config_file = 'zuul-connections-same-gerrit.conf'
|
|
tenant_config_file = 'config/zuul-connections-same-gerrit/main.yaml'
|
|
|
|
def test_multiple_gerrit_connections(self):
|
|
"Test multiple connections to the one gerrit"
|
|
|
|
A = self.fake_review_gerrit.addFakeChange('org/project', 'master', 'A')
|
|
self.addEvent('review_gerrit', A.getPatchsetCreatedEvent(1))
|
|
|
|
self.waitUntilSettled()
|
|
|
|
self.assertEqual(len(A.patchsets[-1]['approvals']), 1)
|
|
self.assertEqual(A.patchsets[-1]['approvals'][0]['type'], 'verified')
|
|
self.assertEqual(A.patchsets[-1]['approvals'][0]['value'], '1')
|
|
self.assertEqual(A.patchsets[-1]['approvals'][0]['by']['username'],
|
|
'jenkins')
|
|
|
|
B = self.fake_review_gerrit.addFakeChange('org/project', 'master', 'B')
|
|
self.executor_server.failJob('project-test2', B)
|
|
self.addEvent('review_gerrit', B.getPatchsetCreatedEvent(1))
|
|
|
|
self.waitUntilSettled()
|
|
|
|
self.assertEqual(len(B.patchsets[-1]['approvals']), 1)
|
|
self.assertEqual(B.patchsets[-1]['approvals'][0]['type'], 'verified')
|
|
self.assertEqual(B.patchsets[-1]['approvals'][0]['value'], '-1')
|
|
self.assertEqual(B.patchsets[-1]['approvals'][0]['by']['username'],
|
|
'civoter')
|
|
|
|
|
|
class TestSQLConnection(ZuulDBTestCase):
|
|
config_file = 'zuul-sql-driver.conf'
|
|
tenant_config_file = 'config/sql-driver/main.yaml'
|
|
|
|
def test_sql_tables_created(self, metadata_table=None):
|
|
"Test the tables for storing results are created properly"
|
|
buildset_table = 'zuul_buildset'
|
|
build_table = 'zuul_build'
|
|
|
|
insp = sa.engine.reflection.Inspector(
|
|
self.connections.connections['resultsdb'].engine)
|
|
|
|
self.assertEqual(9, len(insp.get_columns(buildset_table)))
|
|
self.assertEqual(10, len(insp.get_columns(build_table)))
|
|
|
|
def test_sql_results(self):
|
|
"Test results are entered into an sql table"
|
|
# Grab the sa tables
|
|
tenant = self.sched.abide.tenants.get('tenant-one')
|
|
reporter = _get_reporter_from_connection_name(
|
|
tenant.layout.pipelines['check'].success_actions,
|
|
'resultsdb'
|
|
)
|
|
|
|
# Add a success result
|
|
A = self.fake_gerrit.addFakeChange('org/project', 'master', 'A')
|
|
self.fake_gerrit.addEvent(A.getPatchsetCreatedEvent(1))
|
|
self.waitUntilSettled()
|
|
|
|
# Add a failed result for a negative score
|
|
B = self.fake_gerrit.addFakeChange('org/project', 'master', 'B')
|
|
|
|
self.executor_server.failJob('project-test1', B)
|
|
self.fake_gerrit.addEvent(B.getPatchsetCreatedEvent(1))
|
|
self.waitUntilSettled()
|
|
|
|
conn = self.connections.connections['resultsdb'].engine.connect()
|
|
result = conn.execute(
|
|
sa.sql.select([reporter.connection.zuul_buildset_table]))
|
|
|
|
buildsets = result.fetchall()
|
|
self.assertEqual(2, len(buildsets))
|
|
buildset0 = buildsets[0]
|
|
buildset1 = buildsets[1]
|
|
|
|
self.assertEqual('check', buildset0['pipeline'])
|
|
self.assertEqual('org/project', buildset0['project'])
|
|
self.assertEqual(1, buildset0['change'])
|
|
self.assertEqual(1, buildset0['patchset'])
|
|
self.assertEqual(1, buildset0['score'])
|
|
self.assertEqual('Build succeeded.', buildset0['message'])
|
|
|
|
buildset0_builds = conn.execute(
|
|
sa.sql.select([reporter.connection.zuul_build_table]).
|
|
where(
|
|
reporter.connection.zuul_build_table.c.buildset_id ==
|
|
buildset0['id']
|
|
)
|
|
).fetchall()
|
|
|
|
# Check the first result, which should be the project-merge job
|
|
self.assertEqual('project-merge', buildset0_builds[0]['job_name'])
|
|
self.assertEqual("SUCCESS", buildset0_builds[0]['result'])
|
|
self.assertEqual('https://server/job/project-merge/0/',
|
|
buildset0_builds[0]['log_url'])
|
|
|
|
self.assertEqual('check', buildset1['pipeline'])
|
|
self.assertEqual('org/project', buildset1['project'])
|
|
self.assertEqual(2, buildset1['change'])
|
|
self.assertEqual(1, buildset1['patchset'])
|
|
self.assertEqual(-1, buildset1['score'])
|
|
self.assertEqual('Build failed.', buildset1['message'])
|
|
|
|
buildset1_builds = conn.execute(
|
|
sa.sql.select([reporter.connection.zuul_build_table]).
|
|
where(
|
|
reporter.connection.zuul_build_table.c.buildset_id ==
|
|
buildset1['id']
|
|
)
|
|
).fetchall()
|
|
|
|
# Check the second last result, which should be the project-test1 job
|
|
# which failed
|
|
self.assertEqual('project-test1', buildset1_builds[-2]['job_name'])
|
|
self.assertEqual("FAILURE", buildset1_builds[-2]['result'])
|
|
self.assertEqual('https://server/job/project-test1/0/',
|
|
buildset1_builds[-2]['log_url'])
|
|
|
|
def test_multiple_sql_connections(self):
|
|
"Test putting results in different databases"
|
|
self.updateConfigLayout(
|
|
'tests/fixtures/layout-sql-reporter.yaml')
|
|
|
|
# Add a successful result
|
|
A = self.fake_gerrit.addFakeChange('org/project', 'master', 'A')
|
|
self.fake_gerrit.addEvent(A.getPatchsetCreatedEvent(1))
|
|
self.waitUntilSettled()
|
|
|
|
# Add a failed result
|
|
B = self.fake_gerrit.addFakeChange('org/project', 'master', 'B')
|
|
self.executor_server.failJob('project-test1', B)
|
|
self.fake_gerrit.addEvent(B.getPatchsetCreatedEvent(1))
|
|
self.waitUntilSettled()
|
|
|
|
# Grab the sa tables for resultsdb
|
|
tenant = self.sched.abide.tenants.get('tenant-one')
|
|
reporter1 = _get_reporter_from_connection_name(
|
|
tenant.layout.pipelines['check'].success_actions,
|
|
'resultsdb'
|
|
)
|
|
|
|
conn = self.connections.connections['resultsdb'].engine.connect()
|
|
buildsets_resultsdb = conn.execute(sa.sql.select(
|
|
[reporter1.connection.zuul_buildset_table])).fetchall()
|
|
# Should have been 2 buildset reported to the resultsdb (both success
|
|
# and failure report)
|
|
self.assertEqual(2, len(buildsets_resultsdb))
|
|
|
|
# The first one should have passed
|
|
self.assertEqual('check', buildsets_resultsdb[0]['pipeline'])
|
|
self.assertEqual('org/project', buildsets_resultsdb[0]['project'])
|
|
self.assertEqual(1, buildsets_resultsdb[0]['change'])
|
|
self.assertEqual(1, buildsets_resultsdb[0]['patchset'])
|
|
self.assertEqual(1, buildsets_resultsdb[0]['score'])
|
|
self.assertEqual('Build succeeded.', buildsets_resultsdb[0]['message'])
|
|
|
|
# Grab the sa tables for resultsdb_failures
|
|
reporter2 = _get_reporter_from_connection_name(
|
|
tenant.layout.pipelines['check'].failure_actions,
|
|
'resultsdb_failures'
|
|
)
|
|
|
|
conn = self.connections.connections['resultsdb_failures'].\
|
|
engine.connect()
|
|
buildsets_resultsdb_failures = conn.execute(sa.sql.select(
|
|
[reporter2.connection.zuul_buildset_table])).fetchall()
|
|
# The failure db should only have 1 buildset failed
|
|
self.assertEqual(1, len(buildsets_resultsdb_failures))
|
|
|
|
self.assertEqual('check', buildsets_resultsdb_failures[0]['pipeline'])
|
|
self.assertEqual(
|
|
'org/project', buildsets_resultsdb_failures[0]['project'])
|
|
self.assertEqual(2, buildsets_resultsdb_failures[0]['change'])
|
|
self.assertEqual(1, buildsets_resultsdb_failures[0]['patchset'])
|
|
self.assertEqual(-1, buildsets_resultsdb_failures[0]['score'])
|
|
self.assertEqual(
|
|
'Build failed.', buildsets_resultsdb_failures[0]['message'])
|
|
|
|
|
|
class TestConnectionsBadSQL(ZuulDBTestCase):
|
|
config_file = 'zuul-sql-driver-bad.conf'
|
|
tenant_config_file = 'config/sql-driver/main.yaml'
|
|
|
|
def test_unable_to_connect(self):
|
|
"Test the SQL reporter fails gracefully when unable to connect"
|
|
self.config.set('zuul', 'layout_config',
|
|
'tests/fixtures/layout-sql-reporter.yaml')
|
|
self.sched.reconfigure(self.config)
|
|
|
|
# Trigger a reporter. If no errors are raised, the reporter has been
|
|
# disabled correctly
|
|
A = self.fake_gerrit.addFakeChange('org/project', 'master', 'A')
|
|
self.fake_gerrit.addEvent(A.getPatchsetCreatedEvent(1))
|
|
self.waitUntilSettled()
|
|
|
|
|
|
class TestMultipleGerrits(ZuulTestCase):
|
|
config_file = 'zuul-connections-multiple-gerrits.conf'
|
|
tenant_config_file = 'config/zuul-connections-multiple-gerrits/main.yaml'
|
|
|
|
def test_multiple_project_separate_gerrits(self):
|
|
self.executor_server.hold_jobs_in_build = True
|
|
|
|
A = self.fake_another_gerrit.addFakeChange(
|
|
'org/project1', 'master', 'A')
|
|
self.fake_another_gerrit.addEvent(A.getPatchsetCreatedEvent(1))
|
|
|
|
self.waitUntilSettled()
|
|
|
|
self.assertBuilds([dict(name='project-test2',
|
|
changes='1,1',
|
|
project='org/project1',
|
|
pipeline='another_check')])
|
|
|
|
# NOTE(jamielennox): the tests back the git repo for both connections
|
|
# onto the same git repo on the file system. If we just create another
|
|
# fake change the fake_review_gerrit will try to create another 1,1
|
|
# change and git will fail to create the ref. Arbitrarily set it to get
|
|
# around the problem.
|
|
self.fake_review_gerrit.change_number = 50
|
|
|
|
B = self.fake_review_gerrit.addFakeChange(
|
|
'org/project1', 'master', 'B')
|
|
self.fake_review_gerrit.addEvent(B.getPatchsetCreatedEvent(1))
|
|
|
|
self.waitUntilSettled()
|
|
|
|
self.assertBuilds([
|
|
dict(name='project-test2',
|
|
changes='1,1',
|
|
project='org/project1',
|
|
pipeline='another_check'),
|
|
dict(name='project-test1',
|
|
changes='51,1',
|
|
project='org/project1',
|
|
pipeline='review_check'),
|
|
])
|
|
|
|
self.executor_server.hold_jobs_in_build = False
|
|
self.executor_server.release()
|
|
self.waitUntilSettled()
|