diff --git a/keystoneclient/tests/unit/client_fixtures.py b/keystoneclient/tests/unit/client_fixtures.py index dfb2b2170..46266ce61 100644 --- a/keystoneclient/tests/unit/client_fixtures.py +++ b/keystoneclient/tests/unit/client_fixtures.py @@ -12,6 +12,7 @@ # License for the specific language governing permissions and limitations # under the License. +import contextlib import os import warnings @@ -607,3 +608,14 @@ class Deprecations(fixtures.Fixture): warnings.filterwarnings('error', category=DeprecationWarning, module='^keystoneclient\\.') self.addCleanup(warnings.resetwarnings) + + def expect_deprecations(self): + """Call this if the test expects to call deprecated function.""" + warnings.resetwarnings() + + @contextlib.contextmanager + def expect_deprecations_here(self): + warnings.resetwarnings() + yield + warnings.filterwarnings('error', category=DeprecationWarning, + module='^keystoneclient\\.') diff --git a/keystoneclient/tests/unit/utils.py b/keystoneclient/tests/unit/utils.py index d865e68dc..7c6de95bc 100644 --- a/keystoneclient/tests/unit/utils.py +++ b/keystoneclient/tests/unit/utils.py @@ -42,7 +42,7 @@ class TestCase(testtools.TestCase): def setUp(self): super(TestCase, self).setUp() - self.useFixture(client_fixtures.Deprecations()) + self.deprecations = self.useFixture(client_fixtures.Deprecations()) self.logger = self.useFixture(fixtures.FakeLogger(level=logging.DEBUG)) self.requests_mock = self.useFixture(fixture.Fixture())