oslo.policy/oslo_policy/fixture.py

70 lines
2.1 KiB
Python

# 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 fixtures
__all__ = ['HttpCheckFixture', 'HttpsCheckFixture']
class HttpCheckFixture(fixtures.Fixture):
"""Helps short circuit the external http call"""
def __init__(self, return_value=True):
"""Initialize the fixture.
:param return_value: True implies the policy check passed and False
implies that the policy check failed
:type return_value: boolean
"""
super(HttpCheckFixture, self).__init__()
self.return_value = return_value
def setUp(self):
super(HttpCheckFixture, self).setUp()
def mocked_call(target, cred, enforcer, rule):
return self.return_value
self.useFixture(
fixtures.MonkeyPatch(
'oslo_policy._external.HttpCheck.__call__',
mocked_call,
)
)
class HttpsCheckFixture(fixtures.Fixture):
"""Helps short circuit the external http call"""
def __init__(self, return_value=True):
"""Initialize the fixture.
:param return_value: True implies the policy check passed and False
implies that the policy check failed
:type return_value: boolean
"""
super(HttpsCheckFixture, self).__init__()
self.return_value = return_value
def setUp(self):
super(HttpsCheckFixture, self).setUp()
def mocked_call(target, cred, enforcer, rule):
return self.return_value
self.useFixture(
fixtures.MonkeyPatch(
'oslo_policy._external.HttpsCheck.__call__',
mocked_call,
)
)