return_valueではなくmockを返すメソッドのpythonモック？

Question

は、以下の（本当に簡単）ユニットテストファイル考えてみましょう：

import unittest 
from mock import Mock 

class TestMuPE(unittest.TestCase): 

    def test_01(self): 
    mock_buffer = Mock() 
    mock_buffer.frame_count.return_value(42) 
    self.assertEquals(mock_buffer.frame_count(), 42) 

    # => AssertionError: <Mock name='mock.frame_count()' id='4413645520'> != 42 

私は42に評価するmock_buffer.frame_count()を期待するが、代わりにそれはモックオブジェクトを返します。私は何が欠けていますか？

P .:私はPython 2.7.11をmock-2.0.0バックポートをロードして実行しています。

Answer 1

あなたは使用して戻り値を設定することができます。

mock_buffer.frame_count.return_value = 42 

の代わりに：。

mock_buffer.frame_count.return_value(42)