tests/unit/common/test_template_format.py

   1 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
   2 #    not use this file except in compliance with the License. You may obtain
   3 #    a copy of the License at
   4 #
   5 #         http://www.apache.org/licenses/LICENSE-2.0
   6 #
   7 #    Unless required by applicable law or agreed to in writing, software
   8 #    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
   9 #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  10 #    License for the specific language governing permissions and limitations
  11 #    under the License.
  12
  13 # yardstick: this file is copied from python-heatclient and slightly modified
  14
  15 from __future__ import absolute_import
  16 import mock
  17 import unittest
  18 import yaml
  19
  20 from yardstick.common import template_format
  21
  22
  23 class TemplateFormatTestCase(unittest.TestCase):
  24
  25     def test_parse_to_value_exception(self):
  26
  27         with mock.patch.object(yaml, 'load') as yaml_loader:
  28             yaml_loader.side_effect = yaml.scanner.ScannerError()
  29             self.assertRaises(ValueError, template_format.parse, 'FOOBAR')
  30             yaml_loader.side_effect = yaml.parser.ParserError()
  31             self.assertRaises(ValueError, template_format.parse, 'FOOBAR')
  32             yaml_loader.side_effect = \
  33                 yaml.reader.ReaderError('', '', '', '', '')
  34             self.assertRaises(ValueError, template_format.parse, 'FOOBAR')
  35
  36     def test_parse_no_version_format(self):
  37
  38         yaml = ''
  39         self.assertRaises(ValueError, template_format.parse, yaml)
  40         yaml2 = "Parameters: {}\n" \
  41                 "Mappings: {}\n" \
  42                 "Resources: {}\n" \
  43                 "Outputs: {}"
  44         self.assertRaises(ValueError, template_format.parse, yaml2)
  45
  46
  47 def main():
  48     unittest.main()
  49
  50 if __name__ == '__main__':
  51     main()