ffad78a29f
The recent commit I45d73891f3c0829a2378a032dfc02e5ac0ee34b7 deduplicated the base test class methods from the tempest.lib reintegration. However the surviving base class was put in the wrong namespace. The base class is just that a base to build the individual test subclass off of. It's very confusing for that to not live in the base tests namespace. While the code in lib is supposed to be self contained there isn't a reason to do that here for the unit tests, especially because in this case the import heiarchy becomes more confusing with the base living in a submodule. This commit moves the base class to the base tempest.tests namespace where it really belongs. Change-Id: I079eeb1135eed3254e5e9dbebac8a52b979303c7
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
# Copyright 2013 IBM Corp.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import mock
|
|
from oslotest import base
|
|
|
|
|
|
class TestCase(base.BaseTestCase):
|
|
|
|
def patch(self, target, **kwargs):
|
|
"""Returns a started `mock.patch` object for the supplied target.
|
|
|
|
The caller may then call the returned patcher to create a mock object.
|
|
|
|
The caller does not need to call stop() on the returned
|
|
patcher object, as this method automatically adds a cleanup
|
|
to the test class to stop the patcher.
|
|
|
|
:param target: String module.class or module.object expression to patch
|
|
:param **kwargs: Passed as-is to `mock.patch`. See mock documentation
|
|
for details.
|
|
"""
|
|
p = mock.patch(target, **kwargs)
|
|
m = p.start()
|
|
self.addCleanup(p.stop)
|
|
return m
|
|
|
|
def patchobject(self, target, attribute, new=mock.DEFAULT):
|
|
"""Convenient wrapper around `mock.patch.object`
|
|
|
|
Returns a started mock that will be automatically stopped after the
|
|
test ran.
|
|
"""
|
|
|
|
p = mock.patch.object(target, attribute, new)
|
|
m = p.start()
|
|
self.addCleanup(p.stop)
|
|
return m
|