Use new mock_me fixture to avoid DB call with me method

ansible · Apr 22, 2022 · 9a1a601 · 9a1a601
1 parent 0e0f5b1
commit 9a1a601
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/awx/main/tests/unit/tasks/test_runner_callback.py b/awx/main/tests/unit/tasks/test_runner_callback.py
@@ -3,7 +3,7 @@
 from django.utils.translation import ugettext_lazy as _
 
 
-def test_delay_update():
+def test_delay_update(mock_me):
     rc = RunnerCallback()
     rc.delay_update(foo='bar')
     assert rc.extra_update_fields == {'foo': 'bar'}
@@ -13,15 +13,15 @@ def test_delay_update():
     assert rc.get_extra_update_fields() == {'foo': 'foobar', 'bar': 'foo', 'emitted_events': 0}
 
 
-def test_delay_update_skip_if_set():
+def test_delay_update_skip_if_set(mock_me):
     rc = RunnerCallback()
     rc.delay_update(foo='bar', skip_if_already_set=True)
     assert rc.extra_update_fields == {'foo': 'bar'}
     rc.delay_update(foo='foobar', skip_if_already_set=True)
     assert rc.extra_update_fields == {'foo': 'bar'}
 
 
-def test_delay_update_failure_fields():
+def test_delay_update_failure_fields(mock_me):
     rc = RunnerCallback()
     rc.delay_update(job_explanation='1')
     rc.delay_update(job_explanation=_('2'))