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     def setUp(self):
  19         self.dependency = {'installer': 'test_installer',
  20                            'scenario': 'test_scenario'}
  21
  22         self.testcase = {'dependencies': self.dependency,
  23                          'enabled': 'true',
  24                          'case_name': 'test_name',
  25                          'criteria': 'test_criteria',
  26                          'blocking': 'test_blocking',
  27                          'clean_flag': 'test_clean_flag',
  28                          'description': 'test_desc'}
  29
  30         self.dic_tier = {'name': 'test_tier',
  31                          'order': 'test_order',
  32                          'ci_loop': 'test_ci_loop',
  33                          'description': 'test_desc',
  34                          'testcases': [self.testcase]}
  35
  36         self.mock_yaml = mock.Mock()
  37         attrs = {'get.return_value': [self.dic_tier]}
  38         self.mock_yaml.configure_mock(**attrs)
  39
  40         with mock.patch('functest.ci.tier_builder.yaml.safe_load',
  41                         return_value=self.mock_yaml), \
  42                 mock.patch('__builtin__.open', mock.mock_open()):
  43             self.tierbuilder = tier_builder.TierBuilder('test_installer',
  44                                                         'test_scenario',
  45                                                         'testcases_file')
  46         self.tier_obj = self.tierbuilder.tier_objects[0]
  47
  48     def test_get_tiers(self):
  49         self.assertEqual(self.tierbuilder.get_tiers(),
  50                          [self.tier_obj])
  51
  52     def test_get_tier_names(self):
  53         self.assertEqual(self.tierbuilder.get_tier_names(),
  54                          ['test_tier'])
  55
  56     def test_get_tier_present_tier(self):
  57         self.assertEqual(self.tierbuilder.get_tier('test_tier'),
  58                          self.tier_obj)
  59
  60     def test_get_tier_missing_tier(self):
  61         self.assertEqual(self.tierbuilder.get_tier('test_tier2'),
  62                          None)
  63
  64     def test_get_test_present_test(self):
  65         self.assertEqual(self.tierbuilder.get_test('test_name'),
  66                          self.tier_obj.get_test('test_name'))
  67
  68     def test_get_test_missing_test(self):
  69         self.assertEqual(self.tierbuilder.get_test('test_name2'),
  70                          None)
  71
  72     def test_get_tests_present_tier(self):
  73         self.assertEqual(self.tierbuilder.get_tests('test_tier'),
  74                          self.tier_obj.tests_array)
  75
  76     def test_get_tests_missing_tier(self):
  77         self.assertEqual(self.tierbuilder.get_tests('test_tier2'),
  78                          None)
  79
  80     def test_get_tier_name_ok(self):
  81         self.assertEqual(self.tierbuilder.get_tier_name('test_name'),
  82                          'test_tier')
  83
  84     def test_get_tier_name_ko(self):
  85         self.assertEqual(self.tierbuilder.get_tier_name('test_name2'), None)
  86
  87
  88 if __name__ == "__main__":
  89     logging.disable(logging.CRITICAL)
  90     unittest.main(verbosity=2)