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 os
  13 import pkg_resources
  14
  15 import click
  16
  17 import functest.ci.tier_builder as tb
  18 from functest.utils.constants import CONST
  19 import functest.utils.functest_utils as ft_utils
  20
  21
  22 class Tier(object):
  23
  24     def __init__(self):
  25         self.tiers = tb.TierBuilder(
  26             CONST.__getattribute__('INSTALLER_TYPE'),
  27             CONST.__getattribute__('DEPLOY_SCENARIO'),
  28             pkg_resources.resource_filename('functest', 'ci/testcases.yaml'))
  29
  30     def list(self):
  31         summary = ""
  32         for tier in self.tiers.get_tiers():
  33             summary += ("    - %s. %s:\n\t   %s\n"
  34                         % (tier.get_order(),
  35                            tier.get_name(),
  36                            tier.get_test_names()))
  37         return summary
  38
  39     def show(self, tiername):
  40         tier = self.tiers.get_tier(tiername)
  41         if tier is None:
  42             return None
  43         else:
  44             tier_info = self.tiers.get_tier(tiername)
  45             return tier_info
  46
  47     def gettests(self, tiername):
  48         tier = self.tiers.get_tier(tiername)
  49         if tier is None:
  50             return None
  51         else:
  52             tests = tier.get_test_names()
  53             return tests
  54
  55     @staticmethod
  56     def run(tiername, noclean=False, report=False):
  57
  58         flags = ""
  59         if noclean:
  60             flags += "-n "
  61         if report:
  62             flags += "-r "
  63
  64         if not os.path.isfile(CONST.__getattribute__('env_active')):
  65             click.echo("Functest environment is not ready. "
  66                        "Run first 'functest env prepare'")
  67         else:
  68             cmd = "run_tests {}-t {}".format(flags, tiername)
  69             ft_utils.execute_command(cmd)
  70
  71
  72 class CliTier(Tier):
  73
  74     def __init__(self):
  75         super(CliTier, self).__init__()
  76
  77     def list(self):
  78         click.echo(super(CliTier, self).list())
  79
  80     def show(self, tiername):
  81         tier_info = super(CliTier, self).show(tiername)
  82         if tier_info:
  83             click.echo(tier_info)
  84         else:
  85             tier_names = self.tiers.get_tier_names()
  86             click.echo("The tier with name '%s' does not exist. "
  87                        "Available tiers are:\n  %s\n" % (tiername, tier_names))
  88
  89     def gettests(self, tiername):
  90         tests = super(CliTier, self).gettests(tiername)
  91         if tests:
  92             click.echo("Test cases in tier '%s':\n %s\n" % (tiername, tests))
  93         else:
  94             tier_names = self.tiers.get_tier_names()
  95             click.echo("The tier with name '%s' does not exist. "
  96                        "Available tiers are:\n  %s\n" % (tiername, tier_names))