provider_client.py

# -*- coding: utf-8 -*-
import datetime
import uuid

from flask import Flask, request, jsonify, abort

fakedb = {}

app = Flask(__name__)


@app.route('/_pact/provider_states', methods=['POST'])
def provider_states():
    mapping = {'UserA does not exist': setup_no_user_a,
               'UserA exists and is not an administrator': setup_user_a_nonadmin}
    mapping[request.json['state']]()
    return jsonify({'result': request.json['state']})


def setup_no_user_a():
    if 'UserA' in fakedb:
        del fakedb['UserA']


def setup_user_a_nonadmin():
    fakedb['UserA'] = {'name': "UserA", 'admin': False, 'id': uuid.uuid4(),
                       'created_on': datetime.datetime.now().isoformat()}


@app.route('/users/<name>')
def get_user_by_name(name):
    user_data = fakedb.get(name)
    if not user_data:
        abort(404)
    response = jsonify(**user_data)
    app.logger.debug('get user for %s returns data:\n%s', name, response.data)
    return response


if __name__ == '__main__':
    app.run(host='localhost', port=3001, debug=True)