Django ユニットテストでモックを使用して Celery タスクにパッチを当てる 質問する

私は、Python モック ライブラリを使用して、Django アプリでモデルが保存されるときに実行される Celery タスクにパッチを適用し、正しく呼び出されているかどうかを確認しようとしています。

基本的に、タスクは 内で定義されmyapp.tasks、次のように models.py ファイルの先頭にインポートされます。

from .tasks import mytask

...そして、save()を使用してモデル内で実行しますmytask.delay(foo, bar)。 ここまでは順調です。実際に Celeryd などを実行しているときにも問題なく動作します。

タスクをモックする単体テストを構築して、正しい引数で呼び出され、実際には Celery タスクを実行しようとしないことを確認したいと思います。

したがって、テスト ファイルでは、標準の TestCase 内に次のような内容が含まれています。

from mock import patch # at the top of the file

# ...then later
def test_celery_task(self):
    with patch('myapp.models.mytask.delay') as mock_task:
        # ...create an instance of the model and save it etc
        self.assertTrue(mock_task.called)