yardstick/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_scanner(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
  31     def test_parse_parser(self):
  32
  33         with mock.patch.object(yaml, 'load') as yaml_loader:
  34             yaml_loader.side_effect = yaml.parser.ParserError()
  35             self.assertRaises(ValueError, template_format.parse, 'FOOBAR')
  36
  37     def test_parse_reader(self):
  38
  39         with mock.patch.object(yaml, 'load') as yaml_loader:
  40             yaml_loader.side_effect = \
  41                 yaml.reader.ReaderError('', '', '', '', '')
  42             self.assertRaises(ValueError, template_format.parse, 'FOOBAR')
  43
  44     def test_parse_no_version_format(self):
  45
  46         yaml = ''
  47         self.assertRaises(ValueError, template_format.parse, yaml)
  48         yaml2 = "Parameters: {}\n" \
  49                 "Mappings: {}\n" \
  50                 "Resources: {}\n" \
  51                 "Outputs: {}"
  52         self.assertRaises(ValueError, template_format.parse, yaml2)