2016-06-22 1 views
0

私は2つのオブジェクトにパッチを適用する必要がある私の関数の1つとしてUTを書いています。2つのオブジェクトにパッチを当てると、2番目に最初にパッチされた値が返されます

@patch('mypackage.models.db_models.MongoClient', 
     return_value={}) 
@patch('mypackage.models.db_models.GridFS') 
def test_file_in_db(self, mock_mongoclient, mock_gridfs): 
    print "*"*80 
    print mock_gridfs 
    print mock_gridfs.return_value 
    print "*"*80 
    mock_gridfs.return_value.new_file.return_value = {} 

これはエラーを与える:それはMongoClientためMockオブジェクトを返す理由

---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "/venv/lib/python2.7/site-packages/mock/mock.py", line 1305, in patched 
    return func(*args, **keywargs) 
    File "/tests/models/test_db_models.py", line 29, in test_file_in_db 
    mock_gridfs.return_value.new_file.return_value = {} 
AttributeError: 'dict' object has no attribute 'new_file' 
-------------------- >> begin captured stdout << --------------------- 
******************************************************************************** 
<MagicMock name='MongoClient' id='4385486992'> 
{} 
******************************************************************************** 

--------------------- >> end captured stdout << ---------------------- 

私は2番目の引数にアクセスしていますが、mock_gridfsを意味?

答えて

1

間違った順序で指定した場合、定義した逆の順序でパラメータを配置します。

@patch('mypackage.models.db_models.MongoClient', 
     return_value={}) 
@patch('mypackage.models.db_models.GridFS') 
def test_file_in_db(self, mock_gridfs, mock_mongoclient): 
関連する問題