functest/cli/commands/cli_tier.py

   1 #!/usr/bin/env python
   2 #
   3 # jose.lausuch@ericsson.com
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9
  10 """ global variables """
  11
  12 import pkg_resources
  13
  14 import click
  15
  16 import functest.ci.tier_builder as tb
  17 from functest.utils.constants import CONST
  18 import functest.utils.functest_utils as ft_utils
  19
  20
  21 class Tier(object):
  22
  23     def __init__(self):
  24         self.tiers = tb.TierBuilder(
  25             CONST.__getattribute__('INSTALLER_TYPE'),
  26             CONST.__getattribute__('DEPLOY_SCENARIO'),
  27             pkg_resources.resource_filename('functest', 'ci/testcases.yaml'))
  28
  29     def list(self):
  30         summary = ""
  31         for tier in self.tiers.get_tiers():
  32             summary += ("    - %s. %s:\n\t   %s\n"
  33                         % (tier.get_order(),
  34                            tier.get_name(),
  35                            tier.get_test_names()))
  36         return summary
  37
  38     def show(self, tiername):
  39         tier = self.tiers.get_tier(tiername)
  40         if tier is None:
  41             return None
  42         else:
  43             tier_info = self.tiers.get_tier(tiername)
  44             return tier_info
  45
  46     def gettests(self, tiername):
  47         tier = self.tiers.get_tier(tiername)
  48         if tier is None:
  49             return None
  50         else:
  51             tests = tier.get_test_names()
  52             return tests
  53
  54     @staticmethod
  55     def run(tiername, noclean=False, report=False):
  56         flags = ""
  57         if noclean:
  58             flags += "-n "
  59         if report:
  60             flags += "-r "
  61
  62         cmd = "run_tests {}-t {}".format(flags, tiername)
  63         ft_utils.execute_command(cmd)
  64
  65
  66 class CliTier(Tier):
  67
  68     def __init__(self):
  69         super(CliTier, self).__init__()
  70
  71     def list(self):
  72         click.echo(super(CliTier, self).list())
  73
  74     def show(self, tiername):
  75         tier_info = super(CliTier, self).show(tiername)
  76         if tier_info:
  77             click.echo(tier_info)
  78         else:
  79             tier_names = self.tiers.get_tier_names()
  80             click.echo("The tier with name '%s' does not exist. "
  81                        "Available tiers are:\n  %s\n" % (tiername, tier_names))
  82
  83     def gettests(self, tiername):
  84         tests = super(CliTier, self).gettests(tiername)
  85         if tests:
  86             click.echo("Test cases in tier '%s':\n %s\n" % (tiername, tests))
  87         else:
  88             tier_names = self.tiers.get_tier_names()
  89             click.echo("The tier with name '%s' does not exist. "
  90                        "Available tiers are:\n  %s\n" % (tiername, tier_names))