import mock
from oslo_utils import uuidutils
-from neutron.agent.l2.extensions import qos_agent
+from neutron.agent.l2.extensions import qos
from neutron.api.rpc.callbacks import resources
from neutron import context
from neutron.tests import base
def setUp(self):
super(QosAgentExtensionTestCase, self).setUp()
- self.qos_agent = qos_agent.QosAgentExtension()
+ self.qos_ext = qos.QosAgentExtension()
self.context = context.get_admin_context()
# Don't rely on used driver
mock.patch(
'neutron.manager.NeutronManager.load_class_for_provider',
- return_value=lambda: mock.Mock(spec=qos_agent.QosAgentDriver)
+ return_value=lambda: mock.Mock(spec=qos.QosAgentDriver)
).start()
- self.qos_agent.initialize()
+ self.qos_ext.initialize()
self._create_fake_resource_rpc()
def _create_fake_resource_rpc(self):
self.get_info_mock = mock.Mock(return_value=TEST_GET_INFO_RULES)
- self.qos_agent.resource_rpc.get_info = self.get_info_mock
+ self.qos_ext.resource_rpc.get_info = self.get_info_mock
def _create_test_port_dict(self):
return {'port_id': uuidutils.generate_uuid(),
def test_handle_port_with_no_policy(self):
port = self._create_test_port_dict()
del port['qos_policy_id']
- self.qos_agent._process_rules_updates = mock.Mock()
- self.qos_agent.handle_port(self.context, port)
- self.assertFalse(self.qos_agent._process_rules_updates.called)
+ self.qos_ext._process_rules_updates = mock.Mock()
+ self.qos_ext.handle_port(self.context, port)
+ self.assertFalse(self.qos_ext._process_rules_updates.called)
def test_handle_unknown_port(self):
port = self._create_test_port_dict()
qos_policy_id = port['qos_policy_id']
port_id = port['port_id']
- self.qos_agent.handle_port(self.context, port)
+ self.qos_ext.handle_port(self.context, port)
# we make sure the underlaying qos driver is called with the
# right parameters
- self.qos_agent.qos_driver.create.assert_called_once_with(
+ self.qos_ext.qos_driver.create.assert_called_once_with(
port, TEST_GET_INFO_RULES)
self.assertEqual(port,
- self.qos_agent.qos_policy_ports[qos_policy_id][port_id])
- self.assertTrue(port_id in self.qos_agent.known_ports)
+ self.qos_ext.qos_policy_ports[qos_policy_id][port_id])
+ self.assertTrue(port_id in self.qos_ext.known_ports)
def test_handle_known_port(self):
port_obj1 = self._create_test_port_dict()
port_obj2 = dict(port_obj1)
- self.qos_agent.handle_port(self.context, port_obj1)
- self.qos_agent.qos_driver.reset_mock()
- self.qos_agent.handle_port(self.context, port_obj2)
- self.assertFalse(self.qos_agent.qos_driver.create.called)
+ self.qos_ext.handle_port(self.context, port_obj1)
+ self.qos_ext.qos_driver.reset_mock()
+ self.qos_ext.handle_port(self.context, port_obj2)
+ self.assertFalse(self.qos_ext.qos_driver.create.called)
def test_handle_known_port_change_policy_id(self):
port = self._create_test_port_dict()
- self.qos_agent.handle_port(self.context, port)
- self.qos_agent.resource_rpc.get_info.reset_mock()
+ self.qos_ext.handle_port(self.context, port)
+ self.qos_ext.resource_rpc.get_info.reset_mock()
port['qos_policy_id'] = uuidutils.generate_uuid()
- self.qos_agent.handle_port(self.context, port)
+ self.qos_ext.handle_port(self.context, port)
self.get_info_mock.assert_called_once_with(
self.context, resources.QOS_POLICY,
port['qos_policy_id'])