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
  49             for dic_testcase in dic_tier['testcases']:
  50                 testcase = tier_handler.TestCase(
  51                     name=dic_testcase['case_name'],
  52                     enabled=dic_testcase.get('enabled', True),
  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                         tier.skip_test(testcase)
  62                 else:
  63                     for dependency in dic_testcase['dependencies']:
  64                         kenv = dependency.keys()[0]
  65                         if not re.search(dependency[kenv], env.get(kenv)):
  66                             tier.skip_test(testcase)
  67                             break
  68                     else:
  69                         if testcase.is_enabled():
  70                             tier.add_test(testcase)
  71                         else:
  72                             tier.skip_test(testcase)
  73             self.tier_objects.append(tier)
  74
  75     def get_tiers(self):
  76         return self.tier_objects
  77
  78     def get_tier_names(self):
  79         tier_names = []
  80         for tier in self.tier_objects:
  81             tier_names.append(tier.get_name())
  82         return tier_names
  83
  84     def get_tier(self, tier_name):
  85         for i in range(0, len(self.tier_objects)):
  86             if self.tier_objects[i].get_name() == tier_name:
  87                 return self.tier_objects[i]
  88         return None
  89
  90     def get_tier_name(self, test_name):
  91         for i in range(0, len(self.tier_objects)):
  92             if self.tier_objects[i].is_test(test_name):
  93                 return self.tier_objects[i].name
  94         return None
  95
  96     def get_test(self, test_name):
  97         for i in range(0, len(self.tier_objects)):
  98             if self.tier_objects[i].is_test(test_name):
  99                 return self.tier_objects[i].get_test(test_name)
 100         return None
 101
 102     def get_tests(self, tier_name):
 103         for i in range(0, len(self.tier_objects)):
 104             if self.tier_objects[i].get_name() == tier_name:
 105                 return self.tier_objects[i].get_tests()
 106         return None
 107
 108     def __str__(self):
 109         output = ""
 110         for i in range(0, len(self.tier_objects)):
 111             output += str(self.tier_objects[i]) + "\n"
 112         return output