Djangoの単体テストでモックを使ってセロリのタスクを修正する

Question

モデルがdjangoアプリケーションに保存されたときに実行されるセロリのタスクを修正するためにPythonモックライブラリを使用して、正しく呼び出されているかどうかを確認しようとしています。

基本的には、タスクがmyapp.tasks内で定義されており、そのような私のmodels.pyファイルの先頭にインポートされます：

from .tasks import mytask

...そして使用してモデル内save()上で実行されますmytask.delay(foo, bar)。これまでのところ、Celerydなどを実際に実行しているときにはうまくいきます。

私はタスクを疑似する単体テストを構築したいのですが、正しい引数で呼び出されることを確認するだけです。実際にはこれまでセロリの仕事をしてみてください。

ので、テストファイルで、私は、標準的なテストケースのこの内部のようなものを持っている：

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) 

を...それが呼ばれることは決してありません/常にfalseです。私はいろいろな化身を試しました（代わりにmyapp.models.mytaskをパッチして、代わりにmock_task.delayが呼び出されたかどうかを確認しています）。私は偽のドキュメントからインポート経路が重要であると主張しました。 （私はそれを正しく理解すれば、myapp.models.mytask.delayいうよりmyapp.tasks.mytask.delayだろう）テスト下のモジュール。

私はここで間違っているつもりです

は？セロリのタスクをパッチでいくつかの特定の困難は、私のように使用されているcelery.taskを（パッチを適用できますか？ありますデコレータをmytaskに）代わりに？

Answer 1

あなたが持っている問題は、これがセロリの仕事であるという事実とは関係ありません。 ）

具体的には、あなたが該当する行は次のようになりますので、「MYTASK」をインポートして、そこの上にパッチを適用されて表示したり、他のファイルを見つける必要があります。

with patch('myapp.myview.mytask.delay') as mock_task: 

をするいくつかのより多くの味がありますここでは、この：

http://www.voidspace.org.uk/python/mock/patch.html#where-to-patch

Answer 2

@taskデコレータはTaskオブジェクト（documentationを参照）機能を置き換えます。あなたが自分自身を模擬するならば、のオブジェクトをMagicMockに置き換え、タスクをまったくスケジュールしません。その代わり、Taskオブジェクトのrun()メソッドを模擬そうのような：

@override_settings(CELERY_ALWAYS_EAGER=True) 
@patch('monitor.tasks.monitor_user.run') 
def test_monitor_all(self, monitor_user): 
    """ 
    Test monitor.all task 
    """ 

    user = ApiUserFactory() 
    tasks.monitor_all.delay() 
    monitor_user.assert_called_once_with(user.key)