diff --git a/neutron/tests/unit/agent/l2/extensions/test_qos_agent.py b/neutron/tests/unit/agent/l2/extensions/test_qos_agent.py index e369bf4483f..2f8ceb19040 100755 --- a/neutron/tests/unit/agent/l2/extensions/test_qos_agent.py +++ b/neutron/tests/unit/agent/l2/extensions/test_qos_agent.py @@ -13,13 +13,12 @@ # License for the specific language governing permissions and limitations # under the License. -import copy - import mock from oslo_utils import uuidutils from neutron.agent.l2.extensions import qos_agent from neutron.api.rpc.callbacks import resources +from neutron import context from neutron.tests import base # This is a minimalistic mock of rules to be passed/checked around @@ -32,7 +31,7 @@ class QosAgentExtensionTestCase(base.BaseTestCase): def setUp(self): super(QosAgentExtensionTestCase, self).setUp() self.qos_agent = qos_agent.QosAgentExtension() - self.context = mock.Mock() + self.context = context.get_admin_context() # Force our fake underlying QoS driver #TODO(QoS): change config value when we tie this to a configuration @@ -77,7 +76,7 @@ class QosAgentExtensionTestCase(base.BaseTestCase): def test_handle_known_port(self): port_obj1 = self._create_test_port_dict() - port_obj2 = copy.copy(port_obj1) + 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)