scripts/slave/recipe_modules/service_account/example.py

# Copyright 2016 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Small example of using the service_account api."""

from recipe_engine.recipe_api import Property
from recipe_engine.types import freeze

DEPS = [
  'recipe_engine/platform',
  'recipe_engine/properties',
  'recipe_engine/raw_io',
  'service_account',
]

PROPERTIES = {
  'scopes': Property(),
  'lifetime_sec': Property(),
}

BUILDERS = freeze({
  'windows': {
    'platform': ('win', 32),
  },
  'linux': {
    'platform': ('linux', 64),
  },
})


def RunSteps(api, scopes, lifetime_sec):
  _ = api.service_account.get_token('fake-account', scopes=scopes, lifetime_sec=lifetime_sec)


def GenTests(api):
  def props(scopes=None, lifetime_sec=None):
    return api.properties.generic(scopes=scopes, lifetime_sec=lifetime_sec)

  for buildername in BUILDERS:
    platform = BUILDERS[buildername].get('platform')
    yield (api.test(buildername) +
           props() +
           api.step_data('get access token',
                         stdout=api.raw_io.output('MockTokenValueThing')) +
           api.platform(*platform))
    yield (api.test('%s_no_authutil' % buildername) +
           props() +
           api.step_data('get access token', retcode=1) +
           api.platform(*platform))
    yield (api.test(buildername + '_with_scopes') +
           props(scopes=['fake-scope']) +
           api.step_data('get access token',
                         stdout=api.raw_io.output('MockTokenValueThing')) +
           api.platform(*platform))
    yield (api.test(buildername + '_with_lifetime_sec') +
           props(lifetime_sec=60) +
           api.step_data('get access token',
                         stdout=api.raw_io.output('MockTokenValueThing')) +
           api.platform(*platform))