from oslo_log import log as logging
+from neutron.core_extensions import base as base_core
+from neutron.core_extensions import qos as qos_core
from neutron.plugins.ml2 import driver_api as api
-from neutron.services.qos import qos_extension
LOG = logging.getLogger(__name__)
class QosExtensionDriver(api.ExtensionDriver):
def initialize(self):
- self.qos_ext_handler = qos_extension.QosResourceExtensionHandler()
+ self.core_ext_handler = qos_core.QosCoreResourceExtension()
LOG.debug("QosExtensionDriver initialization complete")
def process_create_network(self, context, data, result):
- self.qos_ext_handler.process_resource(
- context, qos_extension.NETWORK, data, result)
+ self.core_ext_handler.process_fields(
+ context, base_core.NETWORK, data, result)
process_update_network = process_create_network
def process_create_port(self, context, data, result):
- self.qos_ext_handler.process_resource(
- context, qos_extension.PORT, data, result)
+ self.core_ext_handler.process_fields(
+ context, base_core.PORT, data, result)
process_update_port = process_create_port
def extend_network_dict(self, session, db_data, result):
result.update(
- self.qos_ext_handler.extract_resource_fields(qos_extension.NETWORK,
- db_data))
+ self.core_ext_handler.extract_fields(
+ base_core.NETWORK, db_data))
def extend_port_dict(self, session, db_data, result):
result.update(
- self.qos_ext_handler.extract_resource_fields(qos_extension.PORT,
- db_data))
+ self.core_ext_handler.extract_fields(base_core.PORT, db_data))
import mock
from neutron import context
+from neutron.core_extensions import base as base_core
+from neutron.core_extensions import qos as qos_core
from neutron.plugins.common import constants as plugin_constants
from neutron.services.qos import qos_consts
-from neutron.services.qos import qos_extension
from neutron.tests import base
'network_id': 'fake_net_id'}}
-class QosResourceExtensionHandlerTestCase(base.BaseTestCase):
+class QosCoreResourceExtensionTestCase(base.BaseTestCase):
def setUp(self):
- super(QosResourceExtensionHandlerTestCase, self).setUp()
- self.ext_handler = qos_extension.QosResourceExtensionHandler()
+ super(QosCoreResourceExtensionTestCase, self).setUp()
+ self.core_extension = qos_core.QosCoreResourceExtension()
policy_p = mock.patch('neutron.objects.qos.policy.QosPolicy')
self.policy_m = policy_p.start()
self.context = context.get_admin_context()
- def test_process_resource_no_qos_policy_id(self):
- self.ext_handler.process_resource(
- self.context, qos_extension.PORT, {}, None)
+ def test_process_fields_no_qos_policy_id(self):
+ self.core_extension.process_fields(
+ self.context, base_core.PORT, {}, None)
self.assertFalse(self.policy_m.called)
def _mock_plugin_loaded(self, plugin_loaded):
return mock.patch('neutron.manager.NeutronManager.get_service_plugins',
return_value=plugins)
- def test_process_resource_no_qos_plugin_loaded(self):
+ def test_process_fields_no_qos_plugin_loaded(self):
with self._mock_plugin_loaded(False):
- self.ext_handler.process_resource(
- self.context, qos_extension.PORT,
+ self.core_extension.process_fields(
+ self.context, base_core.PORT,
{qos_consts.QOS_POLICY_ID: None}, None)
self.assertFalse(self.policy_m.called)
- def test_process_resource_port_new_policy(self):
+ def test_process_fields_port_new_policy(self):
with self._mock_plugin_loaded(True):
qos_policy_id = mock.Mock()
actual_port = {'id': mock.Mock(),
qos_consts.QOS_POLICY_ID: qos_policy_id}
qos_policy = mock.MagicMock()
self.policy_m.get_by_id = mock.Mock(return_value=qos_policy)
- self.ext_handler.process_resource(
- self.context, qos_extension.PORT,
+ self.core_extension.process_fields(
+ self.context, base_core.PORT,
{qos_consts.QOS_POLICY_ID: qos_policy_id},
actual_port)
qos_policy.attach_port.assert_called_once_with(actual_port['id'])
- def test_process_resource_port_updated_policy(self):
+ def test_process_fields_port_updated_policy(self):
with self._mock_plugin_loaded(True):
qos_policy_id = mock.Mock()
port_id = mock.Mock()
return_value=old_qos_policy)
new_qos_policy = mock.MagicMock()
self.policy_m.get_by_id = mock.Mock(return_value=new_qos_policy)
- self.ext_handler.process_resource(
- self.context, qos_extension.PORT,
+ self.core_extension.process_fields(
+ self.context, base_core.PORT,
{qos_consts.QOS_POLICY_ID: qos_policy_id},
actual_port)
old_qos_policy.detach_port.assert_called_once_with(port_id)
new_qos_policy.attach_port.assert_called_once_with(port_id)
- def test_process_resource_network_new_policy(self):
+ def test_process_fields_network_new_policy(self):
with self._mock_plugin_loaded(True):
qos_policy_id = mock.Mock()
actual_network = {'id': mock.Mock(),
qos_consts.QOS_POLICY_ID: qos_policy_id}
qos_policy = mock.MagicMock()
self.policy_m.get_by_id = mock.Mock(return_value=qos_policy)
- self.ext_handler.process_resource(
- self.context, qos_extension.NETWORK,
+ self.core_extension.process_fields(
+ self.context, base_core.NETWORK,
{qos_consts.QOS_POLICY_ID: qos_policy_id}, actual_network)
qos_policy.attach_network.assert_called_once_with(
actual_network['id'])
- def test_process_resource_network_updated_policy(self):
+ def test_process_fields_network_updated_policy(self):
with self._mock_plugin_loaded(True):
qos_policy_id = mock.Mock()
network_id = mock.Mock()
return_value=old_qos_policy)
new_qos_policy = mock.MagicMock()
self.policy_m.get_by_id = mock.Mock(return_value=new_qos_policy)
- self.ext_handler.process_resource(
- self.context, qos_extension.NETWORK,
+ self.core_extension.process_fields(
+ self.context, base_core.NETWORK,
{qos_consts.QOS_POLICY_ID: qos_policy_id}, actual_network)
old_qos_policy.detach_network.assert_called_once_with(network_id)
new_qos_policy.attach_network.assert_called_once_with(network_id)
- def test_extract_resource_fields_plugin_not_loaded(self):
+ def test_extract_fields_plugin_not_loaded(self):
with self._mock_plugin_loaded(False):
- fields = self.ext_handler.extract_resource_fields(None, None)
+ fields = self.core_extension.extract_fields(None, None)
self.assertEqual({}, fields)
- def _test_extract_resource_fields_for_port(self, qos_policy_id):
+ def _test_extract_fields_for_port(self, qos_policy_id):
with self._mock_plugin_loaded(True):
- fields = self.ext_handler.extract_resource_fields(
- qos_extension.PORT, _get_test_dbdata(qos_policy_id))
+ fields = self.core_extension.extract_fields(
+ base_core.PORT, _get_test_dbdata(qos_policy_id))
self.assertEqual({qos_consts.QOS_POLICY_ID: qos_policy_id}, fields)
- def test_extract_resource_fields_no_port_policy(self):
- self._test_extract_resource_fields_for_port(None)
+ def test_extract_fields_no_port_policy(self):
+ self._test_extract_fields_for_port(None)
- def test_extract_resource_fields_port_policy_exists(self):
+ def test_extract_fields_port_policy_exists(self):
qos_policy_id = mock.Mock()
- self._test_extract_resource_fields_for_port(qos_policy_id)
+ self._test_extract_fields_for_port(qos_policy_id)
- def _test_extract_resource_fields_for_network(self, qos_policy_id):
+ def _test_extract_fields_for_network(self, qos_policy_id):
with self._mock_plugin_loaded(True):
- fields = self.ext_handler.extract_resource_fields(
- qos_extension.NETWORK, _get_test_dbdata(qos_policy_id))
+ fields = self.core_extension.extract_fields(
+ base_core.NETWORK, _get_test_dbdata(qos_policy_id))
self.assertEqual({qos_consts.QOS_POLICY_ID: qos_policy_id}, fields)
- def test_extract_resource_fields_no_network_policy(self):
- self._test_extract_resource_fields_for_network(None)
+ def test_extract_fields_no_network_policy(self):
+ self._test_extract_fields_for_network(None)
- def test_extract_resource_fields_network_policy_exists(self):
+ def test_extract_fields_network_policy_exists(self):
qos_policy_id = mock.Mock()
qos_policy = mock.Mock()
qos_policy.id = qos_policy_id
- self._test_extract_resource_fields_for_network(qos_policy_id)
+ self._test_extract_fields_for_network(qos_policy_id)