def test_does_not_affect_other_threads(self, es_with_signals, monkeypatch):
"""
Test that signal receivers are not disabled for other threads.
This is important when using gevent (e.g. via Gunicorn).
Note that as this test uses another thread, the usual test transaction and clean-up
logic does not kick in. Hence, cleaning up has to be manually handled.
"""
def _task():
try:
# Ensure a transaction is used
transaction.set_autocommit(False)
try:
SimpleModel.objects.create()
finally:
transaction.rollback()
finally:
close_old_connections()
callback_mock = Mock()
monkeypatch.setattr(
'datahub.search.test.search_support.relatedmodel.signals._dummy_callback',
callback_mock,
)
with disable_search_signal_receivers(SimpleModel), \
ThreadPoolExecutor() as executor:
future = executor.submit(_task)
future.result()
callback_mock.assert_called_once()
def test_does_not_affect_other_threads(self, setup_es, monkeypatch):
"""
Test that signal receivers are not disabled for other threads.
This is important when using gevent (e.g. via Gunicorn).
"""
def _task():
try:
SimpleModel().save()
finally:
close_old_connections()
callback_mock = Mock()
monkeypatch.setattr(
'datahub.search.test.search_support.relatedmodel.signals._dummy_callback',
callback_mock,
)
with disable_search_signal_receivers(SimpleModel), \
ThreadPoolExecutor() as executor:
future = executor.submit(_task)
future.result()
callback_mock.assert_called_once()
def test_reconnects_if_was_connected(self, es_with_signals):
"""Test that signal receivers are reconnected on context manager exit."""
with disable_search_signal_receivers(SimpleModel):
pass
assert all(receiver.is_connected for search_app in get_search_apps()
for receiver in search_app.get_signal_receivers()
if receiver.sender is SimpleModel)
def test_does_not_affect_other_models(self, es_with_signals, monkeypatch):
"""Test that signal receivers are not disabled for other models."""
callback_mock = Mock()
monkeypatch.setattr(
'datahub.search.test.search_support.relatedmodel.signals._dummy_callback',
callback_mock,
)
with disable_search_signal_receivers(SimpleModel):
RelatedModel().save()
callback_mock.assert_called_once()
def test_does_not_reconnect_if_was_disconnected(self):
"""
Test that signal receivers are not reconnected when not originally connected.
Note: Signal receivers are not connected as the es_with_signals fixture is not used.
"""
with disable_search_signal_receivers(SimpleModel):
pass
assert not any(receiver.is_connected
for search_app in get_search_apps()
for receiver in search_app.get_signal_receivers()
if receiver.sender is SimpleModel)
def test_reconnects_if_exception_raised(self, es_with_signals):
"""
Test that signal receivers are reconnected if an exception is raised while the
context manager is active.
"""
try:
with disable_search_signal_receivers(SimpleModel):
raise ValueError
except ValueError:
pass
assert all(receiver.is_connected for search_app in get_search_apps()
for receiver in search_app.get_signal_receivers()
if receiver.sender is SimpleModel)
def test_signal_receivers_disabled_for_model(self, setup_es, monkeypatch):
"""
Test that signal receivers are disabled for the specified model.
Note: Model refers to the signal sender, not the search app model. Hence while signal
receivers are disabled for SimpleModel, it is the relatedmodel app signal receivers that
are checked.
"""
callback_mock = Mock()
monkeypatch.setattr(
'datahub.search.test.search_support.relatedmodel.signals._dummy_callback',
callback_mock,
)
with disable_search_signal_receivers(SimpleModel):
SimpleModel().save()
callback_mock.assert_not_called()