def add_preset(uid):
"""
Save the incoming request body as a preset response
"""
preset = Preset(json_data=bottle.request.body.read())
if preset.times != FOREVER and preset.times <= 0:
raise HTTPResponse(("Preset has {0} times. Must be greater than "
"zero.".format(preset.times).encode()),
status=400)
rule = match_rule_from_dict(preset.rule)
if rule not in PRESETS[uid]:
PRESETS[uid][rule] = []
url_presets = PRESETS[uid][rule]
url_presets.append(preset)
def add_preset(uid):
"""
Save the incoming request body as a preset response
"""
preset = Preset(json_data=bottle.request.body.read())
if preset.times != FOREVER and preset.times <= 0:
raise HTTPResponse(("Preset has {0} times. Must be greater than "
"zero.".format(preset.times).encode()),
status=400)
rule = match_rule_from_dict(preset.rule)
if rule not in PRESETS[uid]:
PRESETS[uid][rule] = []
url_presets = PRESETS[uid][rule]
url_presets.append(preset)
def select_preset(uid, request):
"""
Select a preset to respond with.
Look through the presets for a match. If one is found pop off a preset
response and return it.
:param uid: The uid to look up presets for
:param request: A dictionary representing the mock request which is checked
to see if it matches the rule regex and headers stored in the preset.
Return 404 if no preset found that matches.
"""
preset_dict = PRESETS[uid]
for key, preset_list in preset_dict.items():
preset = preset_list[0]
match_rule = match_rule_from_dict(preset.rule)
if match_rule.matches(request):
knock_off_preset(preset_dict, key)
time.sleep(preset.after)
return preset
raise HTTPResponse(b"No matching preset response", status=404)
def select_preset(uid, request):
"""
Select a preset to respond with.
Look through the presets for a match. If one is found pop off a preset
response and return it.
:param uid: The uid to look up presets for
:param request: A dictionary representing the mock request which is checked
to see if it matches the rule regex and headers stored in the preset.
Return 404 if no preset found that matches.
"""
preset_dict = PRESETS[uid]
for key, preset_list in preset_dict.items():
preset = preset_list[0]
match_rule = match_rule_from_dict(preset.rule)
if match_rule.matches(request):
knock_off_preset(preset_dict, key)
time.sleep(preset.after)
return preset
raise HTTPResponse(b"No matching preset response", status=404)
