# See the License for the specific language governing permissions and
# limitations under the License.
+import contextlib
import mock
from oslo.config import cfg
self.addCleanup(db.clear_db)
# Use a mock netconf client
self.mock_ncclient = mock.Mock()
- self.patch_obj = mock.patch.dict('sys.modules',
- {'ncclient': self.mock_ncclient})
- self.patch_obj.start()
- with mock.patch.object(cisco_nexus_plugin_v2.NexusPlugin,
- '__init__', new=new_nexus_init):
+ with contextlib.nested(
+ mock.patch.dict('sys.modules', {'ncclient': self.mock_ncclient}),
+ mock.patch.object(cisco_nexus_plugin_v2.NexusPlugin,
+ '__init__', new=new_nexus_init)
+ ):
self._cisco_nexus_plugin = cisco_nexus_plugin_v2.NexusPlugin()
# Set the Cisco config module's first configured device IP address
self.conf = config.setup_conf()
self.conf.register_opts(interface.OPTS)
config.register_root_helper(self.conf)
- self.device_exists_p = mock.patch.object(ip_lib, 'device_exists')
- self.device_exists = self.device_exists_p.start()
self.driver = interface.MidonetInterfaceDriver(self.conf)
self.network_id = uuidutils.generate_uuid()
self.port_id = uuidutils.generate_uuid()