def _pre_setup(self): super(TestCase, self)._pre_setup() # XXX See if we can remove this when we switch to Django 1.8. # Some crud from the migration system gets stuck here and we have to # kick it loose manually. from mkt.webapps.models import AddonUser UserProfile.addonuser_set.related.model = AddonUser # Clean the slate. cache.clear() post_request_task._discard_tasks() trans_real.deactivate() trans_real._translations = {} # Django fails to clear this cache. trans_real.activate(settings.LANGUAGE_CODE) self.mock_browser_id() global JINJA_INSTRUMENTED if not JINJA_INSTRUMENTED: 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 JINJA_INSTRUMENTED = True
def _pre_setup(self): super(TestCase, self)._pre_setup() # If we have a settings_test.py let's roll it into our settings. try: import settings_test # Use setattr to update Django's proxies: for k in dir(settings_test): setattr(settings, k, getattr(settings_test, k)) except ImportError: pass # Clean the slate. cache.clear() post_request_task._discard_tasks() trans_real.deactivate() trans_real._translations = {} # Django fails to clear this cache. trans_real.activate(settings.LANGUAGE_CODE) self.mock_browser_id() global JINJA_INSTRUMENTED if not JINJA_INSTRUMENTED: 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 JINJA_INSTRUMENTED = True
def _pre_setup(self): super(TestCase, self)._pre_setup() # Clean the slate. cache.clear() post_request_task._discard_tasks() trans_real.deactivate() trans_real._translations = {} # Django fails to clear this cache. trans_real.activate(settings.LANGUAGE_CODE) self.mock_browser_id() global JINJA_INSTRUMENTED if not JINJA_INSTRUMENTED: 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 JINJA_INSTRUMENTED = True
def test_request_failed(self, _mock): with self.settings(CELERY_ALWAYS_EAGER=False): test_task.delay() self._verify_task_filled() # Simulate a request exception. _discard_tasks() self._verify_task_empty() # Assert the original `apply_async` was not called. assert not _mock.called, ( 'Unexpected PostRequestTask.original_apply_async call')
def tearDown(self): task_mock.reset_mock() _discard_tasks()
def _pre_setup(self): super(TestCase, self)._pre_setup() self.mock_browser_id() post_request_task._discard_tasks()
def tearDown(self): task_mock.reset_mock() _discard_tasks() super(TestTask, self).tearDown()
def setUp(self): super(TestTask, self).setUp() task_mock.reset_mock() _discard_tasks()