JSON client wrapper for Flask testing

jsontestclient.py

#!/usr/bin/env python

import json

from flask.testing import FlaskClient

class JSONTestClient(FlaskClient):
    """Use by setting:

        app.test_client_class = JSONTestClient

    The app.test_client() will always use content-type 'applicaiton/json' and
    will automatically encode python objects passed in to `data` parameter
    """

    def open(self, *args, **kwargs):
        if kwargs.get('data', None):
            kwargs['data'] = json.dumps(kwargs['data'])
        kwargs['content_type'] = 'application/json'

        return super(FlaskClient, self).open(*args, **kwargs)