py3.5 os.scandir()을 사용하여 함수에 대한 테스트를 작성하는 방법은 무엇입니까?

Question

새로 추가 된 python 3.5 빌드를 사용하여 함수에 테스트를 작성하는 방법 os.scandir()? DirEntry 객체를 모의 도우미가 있습니까?

os.scandir() 빈 폴더와 2 개의 파일이 거의없는 폴더를 모의하는 방법에 대한 제안 사항이 있습니까?

Answer 1

제안한대로 내 솔루션에는 @mock.patch() 데코레이터가 필요합니다. 난 당신이 [mock.patch()] (https://docs.python.org/3/library/unittest.mock.html#unittest.mock을 찾고 생각

from django.test import TestCase, mock 
from my_app.scan_files import my_method_that_return_count_of_files 

mock_one_file = mock.MagicMock(return_value = ['one_file', ]) 

class MyMethodThatReturnCountOfFilesfilesTestCase(TestCase): 
    @mock.patch('os.scandir', mock_one_file) 
    def test_get_one(self): 
     """ Test it receives 1 (one) file """ 
     files_count = my_method_that_return_count_of_files('/anything/') 
     self.assertEqual(files_count, 1)