functest/tests/unit/ci/test_tier_builder.py

   1 #!/usr/bin/env python
   2
   3 # All rights reserved. This program and the accompanying materials
   4 # are made available under the terms of the Apache License, Version 2.0
   5 # which accompanies this distribution, and is available at
   6 # http://www.apache.org/licenses/LICENSE-2.0
   7
   8 import logging
   9 import unittest
  10
  11 import mock
  12
  13 from functest.ci import tier_builder
  14
  15
  16 class TierBuilderTesting(unittest.TestCase):
  17
  18     logging.disable(logging.CRITICAL)
  19
  20     def setUp(self):
  21         self.dependency = {'installer': 'test_installer',
  22                            'scenario': 'test_scenario'}
  23
  24         self.testcase = {'dependencies': self.dependency,
  25                          'name': 'test_name',
  26                          'criteria': 'test_criteria',
  27                          'blocking': 'test_blocking',
  28                          'clean_flag': 'test_clean_flag',
  29                          'description': 'test_desc'}
  30
  31         self.dic_tier = {'name': 'test_tier',
  32                          'order': 'test_order',
  33                          'ci_loop': 'test_ci_loop',
  34                          'description': 'test_desc',
  35                          'testcases': [self.testcase]}
  36
  37         self.mock_yaml = mock.Mock()
  38         attrs = {'get.return_value': [self.dic_tier]}
  39         self.mock_yaml.configure_mock(**attrs)
  40
  41         with mock.patch('functest.ci.tier_builder.yaml.safe_load',
  42                         return_value=self.mock_yaml), \
  43                 mock.patch('__builtin__.open', mock.mock_open()):
  44             self.tierbuilder = tier_builder.TierBuilder('test_installer',
  45                                                         'test_scenario',
  46                                                         'testcases_file')
  47         self.tier_obj = self.tierbuilder.tier_objects[0]
  48
  49     def test_get_tiers(self):
  50         self.assertEqual(self.tierbuilder.get_tiers(),
  51                          [self.tier_obj])
  52
  53     def test_get_tier_names(self):
  54         self.assertEqual(self.tierbuilder.get_tier_names(),
  55                          ['test_tier'])
  56
  57     def test_get_tier_present_tier(self):
  58         self.assertEqual(self.tierbuilder.get_tier('test_tier'),
  59                          self.tier_obj)
  60
  61     def test_get_tier_missing_tier(self):
  62         self.assertEqual(self.tierbuilder.get_tier('test_tier2'),
  63                          None)
  64
  65     def test_get_test_present_test(self):
  66         self.assertEqual(self.tierbuilder.get_test('test_name'),
  67                          self.tier_obj.get_test('test_name'))
  68
  69     def test_get_test_missing_test(self):
  70         self.assertEqual(self.tierbuilder.get_test('test_name2'),
  71                          None)
  72
  73     def test_get_tests_present_tier(self):
  74         self.assertEqual(self.tierbuilder.get_tests('test_tier'),
  75                          self.tier_obj.tests_array)
  76
  77     def test_get_tests_missing_tier(self):
  78         self.assertEqual(self.tierbuilder.get_tests('test_tier2'),
  79                          None)
  80
  81
  82 if __name__ == "__main__":
  83     unittest.main(verbosity=2)