recipe_modules/tricium/examples/wrapper.py

# Copyright 2020 The LUCI Authors. All rights reserved.
# Use of this source code is governed under the Apache License, Version 2.0
# that can be found in the LICENSE file.
"""An example of a recipe wrapping legacy analyzers."""

from google.protobuf import json_format

from recipe_engine import post_process

from PB.tricium.data import Data

PYTHON_VERSION_COMPATIBILITY = 'PY2+3'

DEPS = [
    'file',
    'path',
    'tricium',
]


def RunSteps(api):
  checkout_base = api.path['cleanup'].join('checkout')
  api.file.write_text('one', checkout_base.join('one.txt'), 'one')
  api.file.write_text('two', checkout_base.join('foo', 'two.txt'), 'two')

  analyzers = [
      api.tricium.analyzers.SPACEY,
      api.tricium.analyzers.PYLINT,
      api.tricium.analyzers.CPPLINT,
      api.tricium.analyzers.COMMITCHECK,
  ]
  # Analyzers can also be added via their names:
  analyzers.append(api.tricium.analyzers.by_name()['Eslint'])

  api.tricium.run_legacy(
      analyzers, checkout_base, ['one.py', 'foo/two.py'], commit_message='msg')


def GenTests(api):

  def results_json(num_comments):
    results = Data.Results()
    results.comments.extend([
        Data.Comment(category='X', message=str(i), path='x')
        for i in range(num_comments)
    ])
    return json_format.MessageToJson(results)

  yield (api.test('success') + api.step_data(
      'Spacey.read results', api.file.read_text(results_json(num_comments=1))) +
         api.post_check(post_process.StatusSuccess) +
         api.post_process(post_process.DropExpectation))

  yield (api.test('with_failure') +
         api.step_data('Spacey.run analyzer', retcode=1) +
         api.post_process(post_process.DropExpectation))

  yield (api.test('many_comments') +
         api.step_data('Pylint.read results',
                       api.file.read_text(results_json(num_comments=51))) +
         api.post_process(post_process.DropExpectation))