def CreateGenericQuerier(suite=None,
project=None,
num_samples=None,
large_query_mode=None):
suite = suite or 'pixel'
project = project or 'project'
num_samples = num_samples or 5
large_query_mode = large_query_mode or False
return queries.BigQueryQuerier(suite, project, num_samples,
large_query_mode)
def main():
args = ParseArgs()
test_expectation_map = expectations.CreateTestExpectationMap(
args.expectation_file, args.tests)
ci_builders = builders.GetCiBuilders(
SUITE_TO_TELEMETRY_SUITE_MAP.get(args.suite, args.suite))
querier = queries.BigQueryQuerier(args.suite, args.project,
args.num_samples, args.large_query_mode)
# Unmatched results are mainly useful for script maintainers, as they don't
# provide any additional information for the purposes of finding unexpectedly
# passing tests or unused expectations.
unmatched = querier.FillExpectationMapForCiBuilders(
test_expectation_map, ci_builders)
try_builders = builders.GetTryBuilders(ci_builders)
unmatched.update(
querier.FillExpectationMapForTryBuilders(test_expectation_map,
try_builders))
unused_expectations = expectations.FilterOutUnusedExpectations(
test_expectation_map)
stale, semi_stale, active = expectations.SplitExpectationsByStaleness(
test_expectation_map)
result_output.OutputResults(stale, semi_stale, active, unmatched,
unused_expectations, args.output_format)
affected_urls = set()
stale_message = ''
if args.remove_stale_expectations:
stale_expectations = []
for _, expectation_map in stale.iteritems():
stale_expectations.extend(expectation_map.keys())
stale_expectations.extend(unused_expectations)
affected_urls |= expectations.RemoveExpectationsFromFile(
stale_expectations, args.expectation_file)
stale_message += (
'Stale expectations removed from %s. Stale comments, '
'etc. may still need to be removed.\n' % args.expectation_file)
if args.modify_semi_stale_expectations:
affected_urls |= expectations.ModifySemiStaleExpectations(
semi_stale, args.expectation_file)
stale_message += ('Semi-stale expectations modified in %s. Stale '
'comments, etc. may still need to be removed.\n' %
args.expectation_file)
if stale_message:
print(stale_message)
if affected_urls:
result_output.OutputAffectedUrls(affected_urls)