def replay_smtp(uid): """ Replay a previously recorded preset, and save the request in history. Update the mock server identified by ``uid``. :returns: An HTTP response * Status Code 200 containing json data found in preset. * Status Code 404 if there are no matching presets. """ # Make a note that this mock server is still in use. pretender.keep_alive('smtp', uid) bottle.response.content_type = 'application/json' selected = replay(uid, bottle.request.body.read().decode('ascii')) return selected.as_json()
def replay_http(uid, url): """ Replay a previously recorded preset, and save the request in history """ pretender.exists_or_404('http', uid) request_info = RequestSerialiser(url, bottle.request) body = request_info.serialize() LOGGER.debug("KEEPING UID {0} ALIVE".format(uid)) pretender.keep_alive('http', uid) boss_response = replay(uid, body) preset = Preset(boss_response.as_json().encode('ascii')) # ^ Any suggestions about what we can do here? # Preset expects a string like object that can be decoded. # in py3k that means a 'bytes' object. in py2.X that means a string. # So the above works, but it looks ugly - ideally we'd handle both in # Preset constructor. return preset.as_http_response(bottle.response)