클래스 함수의 반환 값을 조롱하려면 어떻게해야합니까?

Question

내가 (comicfile.py에서) 다음과 같습니다 파이썬 방법이 있습니다

from zipfile import ZipFile 

... 

class ComicFile(): 
    ... 

    def page_count(self): 
     """Return the number of pages in the file.""" 
     if self.file == None: 
      raise ComicFile.FileNoneError() 

     if not os.path.isfile(self.file): 
      raise ComicFile.FileNotFoundError() 

     with ZipFile(self.file) as zip: 
      members = zip.namelist() 
      pruned = self.prune_dirs(members) 
      length = len(pruned) 
      return length 

나는 이것에 대한 단위 테스트를 (이미 prune_dirs을 테스트 한) 작성하는 중이 야를, 그래서이 무엇입니까 나는 (test_comicfile.py)이 있습니다

F..ss.... 
====================================================================== 
FAIL: test_page_count (test_comicfile.TestPageCount) 
---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "/usr/local/Cellar/python3/3.5.1/Frameworks/Python.framework/Versions/3.5/lib/python3.5/unittest/mock.py", line 1157, in patched 
    return func(*args, **keywargs) 
    File "/Users/chuck/Dropbox/Projects/chiv/chiv.cbstar/test_comicfile.py", line 86, in test_page_count 
    self.assertEqual(count, self.comic_file.page_count()) 
AssertionError: 2 != 0 

---------------------------------------------------------------------- 
Ran 9 tests in 0.010s 

FAILED (failures=1, skipped=2) 

좋아, 그럼:이 테스트를 실행하면

import unittest 
import unittest.mock 

import comicfile 

... 

class TestPageCount(unittest.TestCase): 

    def setUp(self): 
     self.comic_file = comicfile.ComicFile() 

    @unittest.mock.patch('comicfile.ZipFile') 
    def test_page_count(self, mock_zip_file): 
     # Store as tuples to use as dictionary keys. 
     members_dict = {('dir/', 'dir/file1', 'dir/file2'):2, 
         ('file1.jpg', 'file2.jpg', 'file3.jpg'):3 
     } 

     # Make the file point to something to prevent FileNoneError. 
     self.comic_file.file = __file__ 

     for file_tuple, count in members_dict.items(): 
      mock_zip_file.return_value.namelist = list(file_tuple) 
      self.assertEqual(count, self.comic_file.page_count()) 

, 나는 다음을 얻을은 0을 반환합니다. members = zip.namelist() 다음 줄을 page_count에 넣으려고했습니다.

print('\nmembers -> ' + str(members)) 

시험하는 동안,이 얻을 :

members -> <MagicMock name='ZipFile().__enter__().namelist()' id='4483358280'> 

내가 단위 테스트에 아주 새로운 해요 및 unittest.mock 사용에 매우 모호한,하지만 내 이해 mock_zip-file.return_value.namelist = list(file_tuple)가 있도록 만들었습니다해야한다는 것입니다 ZipFile 클래스의 namelist 메서드는 차례로 file_tuple 내용을 각각 반환합니다. 그것 은입니다. 나는 잘 모릅니다.

내가 여기에서하려고하는 것은 분명하지만, 내 단위 테스트가 ZipFile을 처리하는 대신이 하나의 기능 만 테스트하도록 namelist 메서드를 재정의하는 방법을 생각할 수없는 것 같습니다. 게다가.

Answer 1

ZipFile은 context manager으로 인스턴스화됩니다. mock으로 변경하려면 __enter__ 메소드를 참조해야합니다. 당신이 뭘 하려는지

mock_zip_file.return_value.__enter__.return_value.namelist.return_value = list(file_tuple)

은 매우 분명하다,하지만 문맥 관리자는 조롱에 복잡성을 추가합니다. 이것은 당신이 조롱 등록을 안내 할 수

members -> <MagicMock name='ZipFile().__enter__().namelist()' id='4483358280'>

:

---

한 트릭은 모의 레지스터가 모든 통화가로했을 때,이 예에서 그것에서 전화를 가지고 말하는 것입니다 개체를 모두 바꾸기 ()을 return_value