browserid_mock.py

# From Mozillians
# https://github.com/mozilla/mozillians/blob/master/apps/common/\
#   browserid_mock.py
from django.conf import settings
from django.utils.functional import wraps

from mock import patch


class mock_browserid(object):
    """Mock BrowserID."""

    def __init__(self, email=None):
        self.settings_patches = (
            patch.object(
                settings, 'AUTHENTICATION_BACKENDS',
                ('django_browserid.auth.BrowserIDBackend',)
            ),
            patch.object(
                settings, 'BROWSERID_AUDIENCES',
                ['http://testserver'],
            )
        )

        self.patcher = patch('django_browserid.base.requests.post')
        if email is not None:
            self.return_value = {'status': 'okay', 'email': email}
        else:
            self.return_value = {'status': 'failure'}

    def __enter__(self):
        for p in self.settings_patches:
            p.start()
        self.patcher.start().return_value = self.return_value

    def __exit__(self, exc_type, exc_value, traceback):
        for p in self.settings_patches:
            p.stop()
        self.patcher.stop()

    def __call__(self, func):

        @wraps(func)
        def inner(*args, **kwargs):
            with self:
                return func(*args, **kwargs)
        return inner