2016-03-29 7 views
1

私は次のコードを持っているモック:私のコードで同じクラスからの二つの方法

@istest 
@patch.object(Chapter, 'get_author') 
@patch.object(Chapter, 'get_page_count') 
def test_book(self, mock_get_author, mock_get_page_count): 
    book = Book() # Chapter is a field in book 

    mock_get_author.return_value = 'Author name' 
    mock_get_page_count.return_value = 43 

    book.get_information() # calls get_author and then get_page_count 

、get_page_count、get_author後に呼び出され、「Autor名」の代わりに、43方法の期待値を返します私はこれを修正できますか?私は、次のことを試してみた:

@patch('application.Chapter') 
def test_book(self, chapter): 
    mock_chapters = [chapter.Mock(), chapter.Mock()] 
    mock_chapters[0].get_author.return_value = 'Author name' 
    mock_chapters[1].get_page_count.return_value = 43 
    chapter.side_effect = mock_chapters 

    book.get_information() 

しかし、私はエラーを取得:任意の提案を事前に

TypeError: must be type, not MagicMock 

感謝を!

答えて

1

あなたのデコレータの使用方法が正しい順序でないため、その理由があります。下から始めて、あなたのtest_bookメソッドであなたの引数をどのように設定しているかに基づいて、 'get_author'のためのパッチ、そして 'get_page_count'をパッチします。

@istest 
@patch.object(Chapter, 'get_page_count') 
@patch.object(Chapter, 'get_author') 
def test_book(self, mock_get_author, mock_get_page_count): 
    book = Book() # Chapter is a field in book 

    mock_get_author.return_value = 'Author name' 
    mock_get_page_count.return_value = 43 

    book.get_information() # calls get_author and then get_page_count 

は、複数のデコレータを使用する場合、正確に何が起こっているのかを説明するのthis優れた答えを参照するよりも、この他を説明するための良い方法はありません。

+1

ありがとうございます! – Oreo

関連する問題