7dc7c7cdb0
Also, sync requirements with Neutron. Partial-Bug: #1501792 Change-Id: Ib5f70646ddcbc807d45c8d352219029a8d9530d1
97 lines
3.4 KiB
Python
97 lines
3.4 KiB
Python
# 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)
|