Python에서 디렉토리 구조 조롱

Question

파일 입력, 열기 및 처리를 수행하고 일부 데이터를 출력하는 데 사용하는 코드가 있습니다. 나는 기능을 작동 시켰고 지금은 단위 테스트를하고있다. 아래는 코드 예제이다.

def foo(dir): 
    path_to_search = join(dir, "/baz/foo") 

    if isdir(path_to_search): 
    #path exists so do stuff... 

     for fname in listdir(path_to_search): 
      do_stuff() 

    else: 
     print "path doesn't exist" 

내가 과거가 충분히 쉽게 존재하지 않는 테스트를 만들 수 있었다,하지만 난 디렉토리 구조의 "/ 바즈/foo는"부분이 존재한다는 주장의 위 당신이 볼 수 있듯이했습니다 (에 프로덕션은 디렉토리 구조가이 파일을 가져야 만하는 경우도 있고 처리하지 않아도됩니다.

TempDir을 사용하여 임시 디렉토리 구조를 만들고 결합하려고 시도했지만 코드는 항상 그 길은 존재하지 않는다는 말을 시작합니다.

os.listdir의 출력을 조롱하여 필요한/baz/foo 규칙을 따르는 임시 디렉토리 구조를 만들 필요가 없습니까?

Answer 1

가짜 디렉토리 구조를 만들 필요가 없습니다. isdir() 및 listdir() 기능을 모의하기 만하면됩니다. 사용

unittest.mock library (또는 파이썬 버전 < 3.3 똑같은 일이 외부 mock library) :

try: 
    # Python >= 3.3 
    from unittest import mock 
except ImportError: 
    # Python < 3.3 
    import mock 

with mock.patch('yourmodule.isdir') as mocked_isdir, \ 
     mock.patch('yourmodule.listdir') as mocked_listdir: 
    mocked_isdir.return_value = True 
    mocked_listdir.return_value = ['filename1', 'filename2'] 

    yourmodule.foo('/spam/eggs') 

    mocked_isdir.assert_called_with('/spam/eggs/baz/foo') 
    mocked_listdir.assert_called_with('/spam/eggs/baz/foo') 

당신은 Labix 거만한를 사용할 수