tosca2heat/tosca-parser/toscaparser/tests/test_custom_relationships.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 import os
  14
  15 from toscaparser.tests.base import TestCase
  16 from toscaparser.tosca_template import ToscaTemplate
  17
  18
  19 class CustomRelationshipTypesTest(TestCase):
  20
  21     '''TOSCA template.'''
  22     tosca_tpl = os.path.join(
  23         os.path.dirname(os.path.abspath(__file__)),
  24         "data/relationship/test_custom_relationship.yaml")
  25     tosca = ToscaTemplate(tosca_tpl)
  26
  27     def test_version(self):
  28         self.assertEqual(self.tosca.version, "tosca_simple_yaml_1_0")
  29
  30     def test_custom_types(self):
  31         expected_custom_types = ['tosca.capabilities.HA',
  32                                  'tosca.nodes.HACompute',
  33                                  'tosca.relationships.HA']
  34         self.assertItemsEqual(self.tosca.topology_template.custom_defs.keys(),
  35                               expected_custom_types)