blob: 2d2649f7ab11e81d2d7b0fc1c6fb95b815bd4d25 [file] [log] [blame]
# Copyright 2017-present Open Networking Foundation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import unittest
from mock import patch, MagicMock
from tosca.parser import TOSCA_Parser
from grpc_client.resources import RESOURCES
class FakeObj:
new = None
filter = None
class FakeModel:
save = None
delete = None
is_new = False
id = 1
class FakeGuiExt:
objects = FakeObj
class FakeSite:
objects = FakeObj
class FakeInstance:
objects = FakeObj
class FakeUser:
objects = FakeObj
class FakeNode:
objects = FakeObj
USERNAME = "username"
PASSWORD = "pass"
mock_resources = {}
mock_resources["%s~%s" % (USERNAME, PASSWORD)] = {
'XOSGuiExtension': FakeGuiExt,
'Site': FakeSite,
'User': FakeUser,
'Instance': FakeInstance,
'Node': FakeNode
}
class TOSCA_Parser_E2E(unittest.TestCase):
@patch.dict(RESOURCES, mock_resources, clear=True)
@patch.object(FakeGuiExt.objects, 'filter', MagicMock(return_value=[FakeModel]))
@patch.object(FakeModel, 'save')
def test_basic_creation(self, mock_save):
"""
[TOSCA_Parser] Should save models defined in a TOSCA recipe
"""
recipe = """
tosca_definitions_version: tosca_simple_yaml_1_0
description: Persist xos-sample-gui-extension
imports:
- custom_types/xosguiextension.yaml
topology_template:
node_templates:
# UI Extension
test:
type: tosca.nodes.XOSGuiExtension
properties:
name: test
files: /spa/extensions/test/vendor.js, /spa/extensions/test/app.js
"""
parser = TOSCA_Parser(recipe, USERNAME, PASSWORD)
parser.execute()
# checking that the model has been saved
mock_save.assert_called()
self.assertIsNotNone(parser.templates_by_model_name['test'])
self.assertEqual(parser.ordered_models_name, ['test'])
# check that the model was saved with the expected values
saved_model = parser.saved_model_by_name['test']
self.assertEqual(saved_model.name, 'test')
self.assertEqual(saved_model.files, '/spa/extensions/test/vendor.js, /spa/extensions/test/app.js')
@patch.dict(RESOURCES, mock_resources, clear=True)
@patch.object(FakeGuiExt.objects, 'filter', MagicMock(return_value=[FakeModel]))
@patch.object(FakeNode.objects, 'filter', MagicMock(return_value=[FakeModel]))
@patch.object(FakeModel, 'delete')
def test_basic_deletion(self, mock_model):
"""
[TOSCA_Parser] Should delete models defined in a TOSCA recipe
"""
recipe = """
tosca_definitions_version: tosca_simple_yaml_1_0
description: Persist xos-sample-gui-extension
imports:
- custom_types/node.yaml
- custom_types/xosguiextension.yaml
topology_template:
node_templates:
should_stay:
type: tosca.nodes.Node
properties:
name: should_stay
must-exist: true
test:
type: tosca.nodes.XOSGuiExtension
properties:
name: test
files: /spa/extensions/test/vendor.js, /spa/extensions/test/app.js
"""
parser = TOSCA_Parser(recipe, USERNAME, PASSWORD, delete=True)
parser.execute()
# checking that the model has been saved
mock_model.assert_called_once()
self.assertIsNotNone(parser.templates_by_model_name['test'])
self.assertEqual(parser.ordered_models_name, ['test', 'should_stay'])
@patch.dict(RESOURCES, mock_resources, clear=True)
@patch.object(FakeSite.objects, 'filter', MagicMock(return_value=[FakeModel]))
@patch.object(FakeUser.objects, 'filter', MagicMock(return_value=[FakeModel]))
@patch.object(FakeModel, 'save')
def test_related_models_creation(self, mock_save):
"""
[TOSCA_Parser] Should save related models defined in a TOSCA recipe
"""
recipe = """
tosca_definitions_version: tosca_simple_yaml_1_0
description: Create a new site with one user
imports:
- custom_types/user.yaml
- custom_types/site.yaml
topology_template:
node_templates:
# Site
site_onlab:
type: tosca.nodes.Site
properties:
name: Open Networking Lab
site_url: http://onlab.us/
hosts_nodes: True
# User
usertest:
type: tosca.nodes.User
properties:
username: test@opencord.org
email: test@opencord.org
password: mypwd
firstname: User
lastname: Test
is_admin: True
requirements:
- site:
node: site_onlab
relationship: tosca.relationships.BelongsToOne
"""
parser = TOSCA_Parser(recipe, USERNAME, PASSWORD)
parser.execute()
self.assertEqual(mock_save.call_count, 2)
self.assertIsNotNone(parser.templates_by_model_name['site_onlab'])
self.assertIsNotNone(parser.templates_by_model_name['usertest'])
self.assertEqual(parser.ordered_models_name, ['site_onlab', 'usertest'])
# check that the model was saved with the expected values
saved_site = parser.saved_model_by_name['site_onlab']
self.assertEqual(saved_site.name, 'Open Networking Lab')
saved_user = parser.saved_model_by_name['usertest']
self.assertEqual(saved_user.firstname, 'User')
self.assertEqual(saved_user.site_id, 1)
@patch.dict(RESOURCES, mock_resources, clear=True)
@patch.object(FakeSite.objects, 'filter', MagicMock(return_value=[]))
def test_must_exist_fail(self):
"""
[TOSCA_Parser] Should throw an error if an object with 'must_exist' does not exist
"""
recipe = """
tosca_definitions_version: tosca_simple_yaml_1_0
description: Create a new site with one user
imports:
- custom_types/site.yaml
topology_template:
node_templates:
# Site
site_onlab:
type: tosca.nodes.Site
properties:
name: Open Networking Lab
must-exist: True
"""
parser = TOSCA_Parser(recipe, USERNAME, PASSWORD)
with self.assertRaises(Exception) as e:
parser.execute()
self.assertEqual(e.exception.message.message, "[XOS-TOSCA] Model of class Site and properties {'name': 'Open Networking Lab'} has property 'must-exist' but cannot be found")
@patch.dict(RESOURCES, mock_resources, clear=True)
@patch.object(FakeInstance.objects, 'filter', MagicMock(return_value=[FakeModel]))
@patch.object(FakeModel, 'save')
def test_number_param(self, mock_save):
"""
[TOSCA_Parser] Should correctly parse number parameters
"""
recipe = """
tosca_definitions_version: tosca_simple_yaml_1_0
description: Create a new site with one user
imports:
- custom_types/instance.yaml
topology_template:
node_templates:
# Site
instance#test_instance:
type: tosca.nodes.Instance
properties:
name: test_instance
numberCores: 10
"""
parser = TOSCA_Parser(recipe, USERNAME, PASSWORD)
parser.execute()
# checking that the model has been saved
mock_save.assert_called()
# check that the model was saved with the expected values
saved_model = parser.saved_model_by_name['instance#test_instance']
self.assertEqual(saved_model.name, 'test_instance')
self.assertEqual(saved_model.numberCores, 10)