heat/heat/tests/openstack/neutron/test_sfc/test_port_pair.py

#
#    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 mock

from heat.engine.resources.openstack.neutron.sfc import port_pair
from heat.engine import stack
from heat.engine import template
from heat.tests import common
from heat.tests import utils

sample_template = {
    'heat_template_version': '2016-04-08',
    'resources': {
        'test_resource': {
            'type': 'OS::Neutron::PortPair',
            'properties': {
                'name': 'test_port_pair',
                'description': 'desc',
                'ingress':  '6af055d3-26f6-48dd-a597-7611d7e58d35',
                'egress':  '6af055d3-26f6-48dd-a597-7611d7e58d35',
                'service_function_parameters': {'correlation': None}
            }
        }
    }
}


class PortPairTest(common.HeatTestCase):

    def setUp(self):
        super(PortPairTest, self).setUp()

        self.ctx = utils.dummy_context()

        self.stack = stack.Stack(
            self.ctx, 'test_stack',
            template.Template(sample_template)
        )
        self.test_resource = self.stack['test_resource']

        self.test_client_plugin = mock.MagicMock()
        self.test_resource.client_plugin = mock.MagicMock(
            return_value=self.test_client_plugin)

        self.test_client = mock.MagicMock()
        self.test_resource.client = mock.MagicMock(
            return_value=self.test_client)

        self.test_client_plugin.get_notification = mock.MagicMock(
            return_value='sample_notification')

    def test_resource_mapping(self):
        mapping = port_pair.resource_mapping()
        self.assertEqual(port_pair.PortPair, mapping['OS::Neutron::PortPair'])

    def _get_mock_resource(self):
        value = mock.MagicMock()
        value.id = '477e8273-60a7-4c41-b683-fdb0bc7cd152'
        return value

    def test_resource_handle_create(self):
        mock_port_pair_create = self.test_client_plugin.create_sfc_resource
        mock_resource = self._get_mock_resource()
        mock_port_pair_create.return_value = mock_resource

    # validate the properties
        self.assertEqual(
            'test_port_pair',
            self.test_resource.properties.get(
                port_pair.PortPair.NAME))
        self.assertEqual(
            'desc',
            self.test_resource.properties.get(
                port_pair.PortPair.DESCRIPTION))
        self.assertEqual(
            '6af055d3-26f6-48dd-a597-7611d7e58d35',
            self.test_resource.properties.get(
                port_pair.PortPair.INGRESS))
        self.assertEqual(
            '6af055d3-26f6-48dd-a597-7611d7e58d35',
            self.test_resource.properties.get(
                port_pair.PortPair.EGRESS))
        self.assertEqual(
            {'correlation': None},
            self.test_resource.properties.get(
                port_pair.PortPair.SERVICE_FUNCTION_PARAMETERS))

        self.test_resource.data_set = mock.Mock()
        self.test_resource.handle_create()
        mock_port_pair_create.assert_called_once_with(
            'port_pair',
            {
                'name': 'test_port_pair',
                'description': 'desc',
                'ingress': '6af055d3-26f6-48dd-a597-7611d7e58d35',
                'egress': '6af055d3-26f6-48dd-a597-7611d7e58d35',
                'service_function_parameters': {'correlation': None},
            }
        )

    def test_resource_handle_delete(self):
        mock_port_pair_delete = self.test_client_plugin.delete_sfc_resource
        self.test_resource.resource_id = '477e8273-60a7-4c41-b683-fdb0bc7cd151'
        mock_port_pair_delete.return_value = None
        self.assertIsNone(self.test_resource.handle_delete())
        mock_port_pair_delete.assert_called_once_with(
            'port_pair', self.test_resource.resource_id)

    def test_resource_handle_delete_resource_id_is_none(self):
        self.test_resource.resource_id = None
        self.assertIsNone(self.test_resource.handle_delete())
        self.assertEqual(0, self.test_client_plugin.
                         delete_sfc_resource.call_count)

    def test_resource_handle_delete_not_found(self):
        self.test_resource.resource_id = '477e8273-60a7-4c41-b683-fdb0bc7cd151'
        mock_port_pair_delete = self.test_client_plugin.delete_sfc_resource
        mock_port_pair_delete.side_effect = self.test_client_plugin.NotFound
        self.assertIsNone(self.test_resource.handle_delete())

    def test_resource_show_resource(self):
        mock_port_pair_get = self.test_client_plugin.show_sfc_resource
        mock_port_pair_get.return_value = {}
        self.assertEqual({},
                         self.test_resource._show_resource(),
                         'Failed to show resource')

    def test_resource_handle_update(self):
        mock_port_pair_patch = self.test_client_plugin.update_sfc_resource
        self.test_resource.resource_id = '477e8273-60a7-4c41-b683-fdb0bc7cd151'

        prop_diff = {
            port_pair.PortPair.NAME:
                'name-updated',
            port_pair.PortPair.DESCRIPTION:
                'description-updated',
            port_pair.PortPair.INGRESS:
                '6af055d3-26f6-48dd-a597-7611d7e58d35',
            port_pair.PortPair.EGRESS:
                '6af055d3-26f6-48dd-a597-7611d7e58d35',
            port_pair.PortPair.SERVICE_FUNCTION_PARAMETERS:
                {'correlation': None}
            }
        self.test_resource.handle_update(json_snippet=None,
                                         tmpl_diff=None,
                                         prop_diff=prop_diff)

        mock_port_pair_patch.assert_called_once_with(
            'port_pair',
            {
                'name': 'name-updated',
                'description': 'description-updated',
                'ingress': '6af055d3-26f6-48dd-a597-7611d7e58d35',
                'egress': '6af055d3-26f6-48dd-a597-7611d7e58d35',
                'service_function_parameters': {'correlation': None},
            },  self.test_resource.resource_id)