From 95686a1da1b7b94f5c475a215482d2198b5748e6 Mon Sep 17 00:00:00 2001 From: Slawek Kaplonski Date: Tue, 1 Dec 2020 11:24:59 +0100 Subject: [PATCH] Migrate to new engine facade port numa affinity UT module Partially-Implements blueprint: enginefacade-switch Change-Id: Iddcfa7f5a31be328dc6ba918bfaf8eef8bc30688 --- .../tests/unit/extensions/test_port_numa_affinity_policy.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/neutron/tests/unit/extensions/test_port_numa_affinity_policy.py b/neutron/tests/unit/extensions/test_port_numa_affinity_policy.py index 308fc1d7865..185671ff9ef 100644 --- a/neutron/tests/unit/extensions/test_port_numa_affinity_policy.py +++ b/neutron/tests/unit/extensions/test_port_numa_affinity_policy.py @@ -16,6 +16,7 @@ import ddt from neutron_lib.api.definitions import port_numa_affinity_policy as apidef from neutron_lib.api.definitions import portbindings from neutron_lib import constants +from neutron_lib.db import api as db_api from neutron.db import db_base_plugin_v2 from neutron.db import port_numa_affinity_policy_db as pnap_db @@ -30,7 +31,7 @@ class PortNumaAffinityPolicyExtensionExtensionTestPlugin( supported_extension_aliases = [apidef.ALIAS] def create_port(self, context, port): - with context.session.begin(subtransactions=True): + with db_api.CONTEXT_WRITER.using(context): new_port = super( PortNumaAffinityPolicyExtensionExtensionTestPlugin, self).create_port(context, port) @@ -38,7 +39,7 @@ class PortNumaAffinityPolicyExtensionExtensionTestPlugin( return new_port def update_port(self, context, id, port): - with context.session.begin(subtransactions=True): + with db_api.CONTEXT_WRITER.using(context): updated_port = super( PortNumaAffinityPolicyExtensionExtensionTestPlugin, self).update_port(context, id, port)