94 lines
3.0 KiB
Python
94 lines
3.0 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.
|
|
|
|
|
|
from oslo_config import fixture as cfg_fixture
|
|
from oslotest import createfile
|
|
|
|
from keystonemiddleware import audit
|
|
from keystonemiddleware.tests.unit import utils
|
|
|
|
|
|
audit_map_content = """
|
|
[custom_actions]
|
|
reboot = start/reboot
|
|
os-migrations/get = read
|
|
|
|
[path_keywords]
|
|
action = None
|
|
os-hosts = host
|
|
os-migrations = None
|
|
reboot = None
|
|
servers = server
|
|
|
|
[service_endpoints]
|
|
compute = service/compute
|
|
"""
|
|
|
|
|
|
class FakeApp(object):
|
|
def __call__(self, env, start_response):
|
|
body = 'Some response'
|
|
start_response('200 OK', [
|
|
('Content-Type', 'text/plain'),
|
|
('Content-Length', str(sum(map(len, body))))
|
|
])
|
|
return [body]
|
|
|
|
|
|
class FakeFailingApp(object):
|
|
def __call__(self, env, start_response):
|
|
raise Exception('It happens!')
|
|
|
|
|
|
class BaseAuditMiddlewareTest(utils.BaseTestCase):
|
|
PROJECT_NAME = 'keystonemiddleware'
|
|
|
|
def setUp(self):
|
|
super(BaseAuditMiddlewareTest, self).setUp()
|
|
|
|
self.audit_map_file_fixture = self.useFixture(
|
|
createfile.CreateFileWithContent('audit', audit_map_content))
|
|
|
|
self.cfg = self.useFixture(cfg_fixture.Config())
|
|
self.cfg.conf([], project=self.PROJECT_NAME)
|
|
|
|
self.middleware = audit.AuditMiddleware(
|
|
FakeApp(), audit_map_file=self.audit_map,
|
|
service_name='pycadf')
|
|
|
|
@property
|
|
def audit_map(self):
|
|
return self.audit_map_file_fixture.path
|
|
|
|
@staticmethod
|
|
def get_environ_header(req_type):
|
|
env_headers = {'HTTP_X_SERVICE_CATALOG':
|
|
'''[{"endpoints_links": [],
|
|
"endpoints": [{"adminURL":
|
|
"http://admin_host:8774",
|
|
"region": "RegionOne",
|
|
"publicURL":
|
|
"http://public_host:8774",
|
|
"internalURL":
|
|
"http://internal_host:8774",
|
|
"id": "resource_id"}],
|
|
"type": "compute",
|
|
"name": "nova"}]''',
|
|
'HTTP_X_USER_ID': 'user_id',
|
|
'HTTP_X_USER_NAME': 'user_name',
|
|
'HTTP_X_AUTH_TOKEN': 'token',
|
|
'HTTP_X_PROJECT_ID': 'tenant_id',
|
|
'HTTP_X_IDENTITY_STATUS': 'Confirmed'}
|
|
env_headers['REQUEST_METHOD'] = req_type
|
|
return env_headers
|