Small fixes in test_qos_agent UT

Change-Id: Ib09824552edd287ec1df3bdd6700c9ce8a02df29

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 e369bf4483f4ebaa0d2d1be855d6cb0f365134bc..2f8ceb19040734e8326327fe27c20e526b367caa 100755 (executable)
--- 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)