145 строки
4.2 KiB
Python
145 строки
4.2 KiB
Python
from django import http, test
|
|
from django.core.cache import caches
|
|
from django.utils import translation
|
|
|
|
import responses
|
|
import pytest
|
|
|
|
from multidb import pinning
|
|
|
|
from olympia import amo, core
|
|
from olympia.access.models import Group, GroupUser
|
|
from olympia.amo.tests import start_es_mocks, stop_es_mocks
|
|
from olympia.translations.hold import clean_translations
|
|
from olympia.users.models import UserProfile
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def unpin_db(request):
|
|
"""Unpin the database from master in the current DB.
|
|
|
|
The `multidb` middleware pins the current thread to master for 15 seconds
|
|
after any POST request, which can lead to unexpected results for tests
|
|
of DB slave functionality."""
|
|
|
|
request.addfinalizer(pinning.unpin_this_thread)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def mock_elasticsearch():
|
|
"""Mock ElasticSearch in tests by default.
|
|
|
|
Tests that do need ES should inherit from ESTestCase, which will stop the
|
|
mock at setup time."""
|
|
start_es_mocks()
|
|
|
|
yield
|
|
|
|
stop_es_mocks()
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def mock_basket(settings):
|
|
"""Mock Basket in tests by default.
|
|
|
|
Tests that do need basket to work should disable `responses`
|
|
and add a passthrough.
|
|
"""
|
|
USER_TOKEN = u'13f64f64-1de7-42f6-8c7f-a19e2fae5021'
|
|
responses.add(
|
|
responses.GET,
|
|
settings.BASKET_URL + '/news/lookup-user/',
|
|
json={'status': 'ok', 'newsletters': [], 'token': USER_TOKEN})
|
|
responses.add(
|
|
responses.POST,
|
|
settings.BASKET_URL + '/news/subscribe/',
|
|
json={'status': 'ok', 'token': USER_TOKEN})
|
|
responses.add(
|
|
responses.POST,
|
|
settings.BASKET_URL + '/news/unsubscribe/{}/'.format(USER_TOKEN),
|
|
json={'status': 'ok', 'token': USER_TOKEN})
|
|
|
|
|
|
def pytest_configure(config):
|
|
from olympia.amo.tests import prefix_indexes
|
|
prefix_indexes(config)
|
|
|
|
|
|
@pytest.fixture(autouse=True, scope='session')
|
|
def instrument_jinja():
|
|
"""Make sure the "templates" list in a response is properly updated, even
|
|
though we're using Jinja2 and not the default django template engine."""
|
|
import jinja2
|
|
old_render = jinja2.Template.render
|
|
|
|
def instrumented_render(self, *args, **kwargs):
|
|
context = dict(*args, **kwargs)
|
|
test.signals.template_rendered.send(
|
|
sender=self, template=self, context=context)
|
|
return old_render(self, *args, **kwargs)
|
|
|
|
jinja2.Template.render = instrumented_render
|
|
|
|
|
|
def default_prefixer(settings):
|
|
"""Make sure each test starts with a default URL prefixer."""
|
|
request = http.HttpRequest()
|
|
request.META['SCRIPT_NAME'] = ''
|
|
prefixer = amo.urlresolvers.Prefixer(request)
|
|
prefixer.app = settings.DEFAULT_APP
|
|
prefixer.locale = settings.LANGUAGE_CODE
|
|
amo.urlresolvers.set_url_prefix(prefixer)
|
|
|
|
|
|
@pytest.yield_fixture(autouse=True)
|
|
def test_pre_setup(request, tmpdir, settings):
|
|
caches['default'].clear()
|
|
caches['filesystem'].clear()
|
|
|
|
translation.trans_real.deactivate()
|
|
# Django fails to clear this cache.
|
|
translation.trans_real._translations = {}
|
|
translation.trans_real.activate(settings.LANGUAGE_CODE)
|
|
|
|
settings.MEDIA_ROOT = str(tmpdir.mkdir('media'))
|
|
settings.TMP_PATH = str(tmpdir.mkdir('tmp'))
|
|
settings.NETAPP_STORAGE = settings.TMP_PATH
|
|
|
|
# Reset the prefixer and urlconf after updating media root
|
|
default_prefixer(settings)
|
|
|
|
from django.core.urlresolvers import clear_url_caches, set_urlconf
|
|
|
|
def _clear_urlconf():
|
|
clear_url_caches()
|
|
set_urlconf(None)
|
|
|
|
_clear_urlconf()
|
|
|
|
request.addfinalizer(_clear_urlconf)
|
|
|
|
yield
|
|
|
|
core.set_user(None)
|
|
clean_translations(None) # Make sure queued translations are removed.
|
|
|
|
# Make sure we revert everything we might have changed to prefixers.
|
|
amo.urlresolvers.clean_url_prefixes()
|
|
|
|
|
|
@pytest.fixture
|
|
def admin_group(db):
|
|
"""Create the Admins group."""
|
|
return Group.objects.create(name='Admins', rules='*:*')
|
|
|
|
|
|
@pytest.fixture
|
|
def mozilla_user(admin_group, settings):
|
|
"""Create a "Mozilla User"."""
|
|
user = UserProfile.objects.create(pk=settings.TASK_USER_ID,
|
|
email='admin@mozilla.com',
|
|
username='admin')
|
|
user.save()
|
|
GroupUser.objects.create(user=user, group=admin_group)
|
|
return user
|