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