Remove main() and __main__ from tests.
[yardstick.git] / 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_to_value_exception(self):
26
27         # TODO(elfoley): Don't hide the error that occurs in
28         # template_format.parse
29         # TODO(elfoley): Separate these tests; one per error type
30         with mock.patch.object(yaml, 'load') as yaml_loader:
31             yaml_loader.side_effect = yaml.scanner.ScannerError()
32             self.assertRaises(ValueError, template_format.parse, 'FOOBAR')
33             yaml_loader.side_effect = yaml.parser.ParserError()
34             self.assertRaises(ValueError, template_format.parse, 'FOOBAR')
35             yaml_loader.side_effect = \
36                 yaml.reader.ReaderError('', '', '', '', '')
37             self.assertRaises(ValueError, template_format.parse, 'FOOBAR')
38
39     def test_parse_no_version_format(self):
40
41         yaml = ''
42         self.assertRaises(ValueError, template_format.parse, yaml)
43         yaml2 = "Parameters: {}\n" \
44                 "Mappings: {}\n" \
45                 "Resources: {}\n" \
46                 "Outputs: {}"
47         self.assertRaises(ValueError, template_format.parse, yaml2)