diff --git a/neutron/tests/unit/db/test_provisioning_blocks.py b/neutron/tests/unit/db/test_provisioning_blocks.py index 02f48a750bf..9920de6b27a 100644 --- a/neutron/tests/unit/db/test_provisioning_blocks.py +++ b/neutron/tests/unit/db/test_provisioning_blocks.py @@ -13,15 +13,16 @@ # under the License. import mock +import netaddr from neutron_lib.callbacks import registry from neutron_lib.callbacks import resources from neutron_lib import context as n_ctx import testtools -from neutron.db import api as db_api from neutron.db import models_v2 from neutron.db import provisioning_blocks as pb from neutron.objects import network as net_obj +from neutron.objects import ports as port_obj from neutron.tests.unit import testlib_api CORE_PLUGIN = 'neutron.db.db_base_plugin_v2.NeutronDbPluginV2' @@ -47,12 +48,11 @@ class TestStatusBarriers(testlib_api.SqlTestCase): def _make_port(self): net = self._make_net() - with db_api.context_manager.writer.using(self.ctx): - port = models_v2.Port(network_id=net.id, mac_address='1', - tenant_id='1', admin_state_up=True, - status='DOWN', device_id='2', - device_owner='3') - self.ctx.session.add(port) + mac_address = netaddr.EUI('1') + port = port_obj.Port(self.ctx, network_id=net.id, device_owner='3', + project_id='1', admin_state_up=True, status='DOWN', device_id='2', + mac_address=mac_address) + port.create() return port def test_no_callback_on_missing_object(self):