# Copyright (C) 2015 Midokura SARL.
# All Rights Reserved.
#
#    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
import testtools

from oslo_utils import uuidutils

import neutron.common.rpc as n_rpc
from neutron import context
from neutron.tests.unit import testlib_api

import neutron_taas.db.taas_db  # noqa
import neutron_taas.extensions.taas as taas_ext
from neutron_taas.services.taas import taas_plugin


class TestTaasPlugin(testlib_api.SqlTestCaseLight):
    def setUp(self):
        super(TestTaasPlugin, self).setUp()
        mock.patch.object(n_rpc, 'create_connection', auto_spec=True).start()
        mock.patch.object(taas_plugin, 'TaasCallbacks', auto_spec=True).start()
        mock.patch.object(taas_plugin, 'TaasAgentApi', auto_spec=True).start()
        self._plugin = taas_plugin.TaasPlugin()
        self._context = context.get_admin_context()

    def test_create_tap_service(self):
        tenant_id = 'tenant-X'
        network_id = uuidutils.generate_uuid()
        host_id = 'host-A'
        port_id = uuidutils.generate_uuid()
        port_details = {
            'tenant_id': tenant_id,
            'binding:host_id': host_id,
        }
        tap_service = {
            'tenant_id': tenant_id,
            'name': 'MyTap',
            'description': 'This is my tap service',
            'port_id': port_id,
            'network_id': network_id,
        }
        req = {
            'tap_service': tap_service,
        }
        with mock.patch.object(self._plugin, '_get_port_details',
                               return_value=port_details):
            self._plugin.create_tap_service(self._context, req)
        tap_service['id'] = mock.ANY
        expected_msg = {
            'tap_service': tap_service,
            'taas_id': mock.ANY,
            'port': port_details,
        }
        self.assertEqual(
            [
                mock.call.create_tap_service(self._context, expected_msg,
                                             host_id)
            ], self._plugin.agent_rpc.mock_calls)

    def test_create_tap_service_wrong_tenant_id(self):
        tenant_id = 'tenant-X'
        network_id = uuidutils.generate_uuid()
        host_id = 'host-A'
        port_id = uuidutils.generate_uuid()
        port_details = {
            'tenant_id': 'other-tenant',
            'binding:host_id': host_id,
        }
        tap_service = {
            'tenant_id': tenant_id,
            'name': 'MyTap',
            'description': 'This is my tap service',
            'port_id': port_id,
            'network_id': network_id,
        }
        req = {
            'tap_service': tap_service,
        }
        with mock.patch.object(self._plugin, '_get_port_details',
                               return_value=port_details), \
            testtools.ExpectedException(taas_ext.PortDoesNotBelongToTenant):
            self._plugin.create_tap_service(self._context, req)
        self.assertEqual([], self._plugin.agent_rpc.mock_calls)