A responses TestCase Mixin

responses-mixin.py

"""
A unittest.TestCase mixin that allows using the
responses (https://pypi.python.org/pypi/responses/) package in tests.

Usage
-----

Install responses with `pip install responses`.

Add `ResponsesMixin` to your `TestCase` parent classes instead of using
the `@responses.activate` decorator; for details see issue here:
https://github.com/getsentry/responses/issues/31
"""

import responses


class ResponsesMixin(object):
    def setUp(self):
        assert responses, 'responses package required to use ReponsesMixin'
        responses.start()
        self.add_default_response()
        super(ResponsesMixin, self).setUp()

    def tearDown(self):
        super(ResponsesMixin, self).tearDown()
        responses.stop()
        responses.reset()

    def add_default_response(self):
        """ Override to add default response/s for all test in this TestCase """
        pass