9d8adc726a
* Removes duplication of code in validate * Breaks large validation functions into atomic, reusable pieces in super classes. * Started applying the rule where validation does not leave any side effects on class members other than the valid boolean. * Validate functions are not called after the first fail, this gives less feedback to user, but is probably reasonable general assumption as some validation relies on others passing. * Adding some tests for NewUser validation. * Refactored how the FakeManager handled resource objects or ids being passed to it. * Fixed some issues with logs being spat out during tests. Change-Id: Iea0afce06e92d8f1a1bda0cc03a32c00909828d1
101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
# Copyright (C) 2015 Catalyst IT Ltd
|
|
#
|
|
# 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 time import time
|
|
from logging import getLogger
|
|
from django.utils import timezone
|
|
|
|
|
|
class KeystoneHeaderUnwrapper(object):
|
|
"""
|
|
Middleware to build an easy to use dict of important data from
|
|
what the keystone wsgi middleware gives us.
|
|
"""
|
|
def process_request(self, request):
|
|
try:
|
|
token_data = {
|
|
'project_domain_id': request.META['HTTP_X_PROJECT_DOMAIN_ID'],
|
|
'project_name': request.META['HTTP_X_PROJECT_NAME'],
|
|
'project_id': request.META['HTTP_X_PROJECT_ID'],
|
|
'roles': request.META['HTTP_X_ROLES'].split(','),
|
|
'user_domain_id': request.META['HTTP_X_USER_DOMAIN_ID'],
|
|
'username': request.META['HTTP_X_USER_NAME'],
|
|
'user_id': request.META['HTTP_X_USER_ID'],
|
|
'authenticated': request.META['HTTP_X_IDENTITY_STATUS']
|
|
}
|
|
except KeyError:
|
|
token_data = {}
|
|
request.keystone_user = token_data
|
|
|
|
|
|
class TestingHeaderUnwrapper(object):
|
|
"""
|
|
Replacement for the KeystoneHeaderUnwrapper for testing purposes.
|
|
"""
|
|
def process_request(self, request):
|
|
try:
|
|
token_data = {
|
|
# TODO(adriant): follow up patch to update all the test
|
|
# headers to provide domain values.
|
|
# Default here is just a temporary measure.
|
|
'project_domain_id':
|
|
request.META['headers'].get(
|
|
'project_domain_id', 'default'),
|
|
'project_name': request.META['headers']['project_name'],
|
|
'project_id': request.META['headers']['project_id'],
|
|
'roles': request.META['headers']['roles'].split(','),
|
|
'user_domain_id':
|
|
request.META['headers'].get('user_domain_id', 'default'),
|
|
'username': request.META['headers']['username'],
|
|
'user_id': request.META['headers']['user_id'],
|
|
'authenticated': request.META['headers']['authenticated']
|
|
}
|
|
except KeyError:
|
|
token_data = {}
|
|
request.keystone_user = token_data
|
|
|
|
|
|
class RequestLoggingMiddleware(object):
|
|
"""
|
|
Middleware to log the requests and responses.
|
|
Will time the duration of a request and log that.
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.logger = getLogger('stacktask')
|
|
|
|
def process_request(self, request):
|
|
self.logger.info(
|
|
'(%s) - <%s> %s [%s]',
|
|
timezone.now(),
|
|
request.method,
|
|
request.META['REMOTE_ADDR'],
|
|
request.get_full_path()
|
|
)
|
|
request.timer = time()
|
|
|
|
def process_response(self, request, response):
|
|
if hasattr(request, 'timer'):
|
|
time_delta = time() - request.timer
|
|
else:
|
|
time_delta = -1
|
|
self.logger.info(
|
|
'(%s) - <%s> [%s] - (%.1fs)',
|
|
timezone.now(),
|
|
response.status_code,
|
|
request.get_full_path(),
|
|
time_delta
|
|
)
|
|
return response
|