1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
import unittest
from tests.utils.test_utils import init_tosca_builder_with_schema_and_spec, \
init_tosca_builder_with_policy_schema_and_spec, init_tosca_builder_with_hello_world_spec_k8
from toscalib.tosca_workbook import ToscaWorkBook
class TestToscaBuilderMethods(unittest.TestCase):
def test_create_node_type(self):
builder = init_tosca_builder_with_schema_and_spec()
self.assertIsNotNone(builder.spec_import.type)
self.assertEqual(builder.spec_import.type, 'docker')
self.assertTrue(builder._using_dmaap())
self.assertIsNone(builder.cloudify_type)
self.assertIsNone(builder.new_type_name)
self.assertNotIn('tosca.dcae.nodes.dockerApp.test_spec_ss', builder.db.NODE_TYPES)
self.assertNotIn('tosca.dcae.nodes.dockerApp.docker_flag', builder.db.NODE_TYPES)
builder.create_node_type()
self.assertIn('tosca.dcae.nodes.dockerApp.test_spec_ss', builder.db.NODE_TYPES)
self.assertIsNotNone(builder.new_type_name)
self.assertEqual(builder.new_type_name, 'tosca.dcae.nodes.dockerApp.test_spec_ss')
self.assertIsNotNone(builder.cloudify_type)
# TODO uncomment after K8 support integration
# self.assertEqual(builder.cloudify_type, builder.db.NODE_TYPES['dcae.nodes.ContainerizedServiceComponentUsingDmaap'])
# builder.create_node_type('docker_flag', True)
# self.assertIn('tosca.dcae.nodes.dockerApp.docker_flag', builder.db.NODE_TYPES)
# self.assertEqual(builder.new_type_name, 'tosca.dcae.nodes.dockerApp.docker_flag')
# self.assertEqual(builder.cloudify_type, builder.db.NODE_TYPES['dcae.nodes.DockerContainerForComponentsUsingDmaap'])
def test_create_model(self):
builder = init_tosca_builder_with_schema_and_spec()
builder.create_node_type()
builder.create_model('test_ss')
self.assertIsNotNone(builder.template)
self.assertIsNotNone(builder.template.db)
self.assertEqual(builder.db, builder.template.db)
self.assertIn('test_ss', builder.template.node_dict)
self.assertIn('topic0', builder.template.node_dict)
def test_create_translate(self):
builder = init_tosca_builder_with_schema_and_spec()
builder.create_node_type()
builder.create_translate('test_ss')
self.assertIsNotNone(builder.template)
self.assertEqual(len(builder.template.sub_rules), 2)
self.assertEqual(builder.template.metadata, {'template_name': 'test_ss_translate'})
self.assertEqual(builder.db, builder.template.db)
def test_create_policy(self):
builder = init_tosca_builder_with_policy_schema_and_spec()
self.assertEqual(len(builder.db.DATA_TYPES), 0)
self.assertEqual(len(builder.db.NODE_TYPES), 1)
builder.create_policy()
self.assertEqual(len(builder.db.DATA_TYPES), 3)
self.assertEqual(len(builder.db.NODE_TYPES), 2)
def test_spec_to_model_to_blueprint_create(self):
#TODO imports section
#TODO assertions
builder = init_tosca_builder_with_hello_world_spec_k8()
name = builder.spec_import.name
filename = 'WEB'
builder.create_node_type(name)
schema = builder.export_schema(filename)
builder.create_model(name)
template = builder.export_model(filename)
builder.create_translate(name)
translate = builder.export_translation(filename)
workbook = ToscaWorkBook()
workbook._import_dir('../data/shared_model/')
workbook._import_yml_str(schema)
workbook._import_yml_str(template)
workbook._import_yml_str(translate)
workbook._translate_template_yaml_str(template)
workbook._add_shared_node(
[{'dcae.capabilities.cdapHost': 'cdap_host'}, {'dcae.capabilities.dockerHost': 'docker_host'},
{'dcae.capabilities.composition.host': 'composition_virtual'}])
bp = workbook._export_yaml_web('cloudify,main')
|
