797865c1d5
Improved the hook to be more flexible, and support v3 request/response masking. Improved testing, adding full json samples to be parsed. Change-Id: I5e7e2bc9627f423abcaaaa1e3335a46ab5377ef8
59 lines
2.2 KiB
Python
59 lines
2.2 KiB
Python
# Copyright (c) 2016 Hewlett-Packard Enterprise Development Company, L.P.
|
|
#
|
|
# 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.
|
|
"""Custom hooks for betamax and keystoneauth.
|
|
|
|
Module providing a set of hooks specially designed for
|
|
interacting with clouds and keystone authentication.
|
|
|
|
:author: Yolanda Robla
|
|
"""
|
|
|
|
import json
|
|
|
|
|
|
def mask_fixture_values(nested, prev_key):
|
|
for key, value in nested.items():
|
|
if isinstance(value, dict):
|
|
mask_fixture_values(value, key)
|
|
else:
|
|
if key in ('tenantName', 'username'):
|
|
nested[key] = 'dummy'
|
|
elif prev_key in ('user', 'project', 'tenant') and key == 'name':
|
|
nested[key] = 'dummy'
|
|
elif prev_key == 'domain' and key == 'id':
|
|
nested[key] = 'dummy'
|
|
elif key == 'password':
|
|
nested[key] = '********'
|
|
elif prev_key == 'token' and key in ('expires', 'expires_at'):
|
|
nested[key] = '9999-12-31T23:59:59Z'
|
|
|
|
|
|
def pre_record_hook(interaction, cassette):
|
|
"""Hook to mask saved data.
|
|
|
|
This hook will be triggered before saving the interaction, and
|
|
will perform two tasks:
|
|
- mask user, project and password in the saved data
|
|
- set token expiration time to an inifinite time.
|
|
"""
|
|
request_body = interaction.data['request']['body']
|
|
parsed_content = json.loads(request_body['string'])
|
|
mask_fixture_values(parsed_content, None)
|
|
request_body['string'] = json.dumps(parsed_content)
|
|
|
|
response_body = interaction.data['response']['body']
|
|
parsed_content = json.loads(response_body['string'])
|
|
mask_fixture_values(parsed_content, None)
|
|
response_body['string'] = json.dumps(parsed_content)
|