xtesting/ci/tier_builder.py

   1 #!/usr/bin/env python
   2
   3 # Copyright (c) 2016 Ericsson AB and others.
   4 #
   5 # All rights reserved. This program and the accompanying materials
   6 # are made available under the terms of the Apache License, Version 2.0
   7 # which accompanies this distribution, and is available at
   8 # http://www.apache.org/licenses/LICENSE-2.0
   9
  10 """TierBuilder class to parse testcases config file"""
  11
  12 import re
  13 import yaml
  14
  15 from xtesting.ci import tier_handler
  16 from xtesting.utils import env
  17
  18
  19 class TierBuilder(object):
  20     # pylint: disable=missing-docstring
  21
  22     def __init__(self, testcases_file):
  23         self.ci_installer = env.get('INSTALLER_TYPE')
  24         self.ci_scenario = env.get('DEPLOY_SCENARIO')
  25         self.testcases_file = testcases_file
  26         self.dic_tier_array = None
  27         self.tier_objects = []
  28         self.testcases_yaml = None
  29         self.generate_tiers()
  30
  31     def read_test_yaml(self):
  32         with open(self.testcases_file) as tc_file:
  33             self.testcases_yaml = yaml.safe_load(tc_file)
  34
  35         self.dic_tier_array = []
  36         for tier in self.testcases_yaml.get("tiers"):
  37             self.dic_tier_array.append(tier)
  38
  39     def generate_tiers(self):
  40         if self.dic_tier_array is None:
  41             self.read_test_yaml()
  42
  43         del self.tier_objects[:]
  44         for dic_tier in self.dic_tier_array:
  45             tier = tier_handler.Tier(
  46                 name=dic_tier['name'], order=dic_tier['order'],
  47                 description=dic_tier['description'])
  48             for dic_testcase in dic_tier['testcases']:
  49                 testcase = tier_handler.TestCase(
  50                     name=dic_testcase['case_name'],
  51                     enabled=dic_testcase.get('enabled', True),
  52                     skipped=False,
  53                     criteria=dic_testcase['criteria'],
  54                     blocking=dic_testcase['blocking'],
  55                     description=dic_testcase['description'],
  56                     project=dic_testcase['project_name'])
  57                 if not dic_testcase.get('dependencies'):
  58                     if testcase.is_enabled():
  59                         tier.add_test(testcase)
  60                     else:
  61                         testcase.skipped = True
  62                         tier.skip_test(testcase)
  63                 else:
  64                     for dependency in dic_testcase['dependencies']:
  65                         kenv = dependency.keys()[0]
  66                         if not re.search(dependency[kenv], env.get(kenv)):
  67                             testcase.skipped = True
  68                             tier.skip_test(testcase)
  69                             break
  70                     else:
  71                         if testcase.is_enabled():
  72                             tier.add_test(testcase)
  73                         else:
  74                             testcase.skipped = True
  75                             tier.skip_test(testcase)
  76             self.tier_objects.append(tier)
  77
  78     def get_tiers(self):
  79         return self.tier_objects
  80
  81     def get_tier_names(self):
  82         tier_names = []
  83         for tier in self.tier_objects:
  84             tier_names.append(tier.get_name())
  85         return tier_names
  86
  87     def get_tier(self, tier_name):
  88         for i in range(0, len(self.tier_objects)):
  89             if self.tier_objects[i].get_name() == tier_name:
  90                 return self.tier_objects[i]
  91         return None
  92
  93     def get_tier_name(self, test_name):
  94         for i in range(0, len(self.tier_objects)):
  95             if self.tier_objects[i].is_test(test_name):
  96                 return self.tier_objects[i].name
  97         return None
  98
  99     def get_test(self, test_name):
 100         for i in range(0, len(self.tier_objects)):
 101             if self.tier_objects[i].is_test(test_name):
 102                 return self.tier_objects[i].get_test(test_name)
 103         return None
 104
 105     def get_tests(self, tier_name):
 106         for i in range(0, len(self.tier_objects)):
 107             if self.tier_objects[i].get_name() == tier_name:
 108                 return self.tier_objects[i].get_tests()
 109         return None
 110
 111     def __str__(self):
 112         output = ""
 113         for i in range(0, len(self.tier_objects)):
 114             output += str(self.tier_objects[i]) + "\n"
 115         return output