# trim out 'HTTP_' prefix part and replace "_" wiht "-".
def originHeaders(request):
- regex = re.compile('^HTTP_')
- return dict((regex.sub('', header).replace("_", "-"), value)
- for (header, value) in request.META.items()
- if header.startswith('HTTP_'))
+ headers = {}
+ for key, value in request.META.items():
+ if key.startswith('HTTP_') and key != 'HTTP_HOST':
+ headers[key[5:].replace('_', '-')] = value
+ elif key in ('CONTENT_TYPE', 'CONTENT_LENGTH'):
+ headers[key.replace('_', '-')] = value
+ return headers
def findMultivimDriver(vim=None):
req = mock.Mock()
req.META = {
"HTTP_X_AUTH_TOKEN": "token_1",
- "NOT_STARTSWITH_HTTP": "value_1"
+ "NOT_STARTSWITH_HTTP": "value_1",
+ "CONTENT_TYPE": "application/json"
}
expect_headers = {
- "X-AUTH-TOKEN": "token_1"
+ "X-AUTH-TOKEN": "token_1",
+ "CONTENT-TYPE": "application/json"
}
ret_headers = syscomm.originHeaders(req)
self.assertDictEqual(expect_headers, ret_headers)