하나의 모듈에 대해서만 모의 기능을 패치 하시겠습니까?

Question

내 파이썬 기능을 테스트하기 위해 os.listdir 패치와 다른 os 함수를 패치해야합니다. 그러나 패치를 적용하면 import 문이 실패합니다. 이 함수를 실제로 하나의 모듈 안에 패치 할 수 있습니까? 그리고 tests.py는 정상적으로 작동합니까?

import os 
from mock import patch 

# when both isdir and isfile are patched 
# the function crashes 
@patch('os.path.isdir', return_value=False) 
@patch('os.path.isfile', return_value=False) 
def test(*args): 
    import ipdb; ipdb.set_trace() 
    real_function(some_arguments) 
    pass 

test() 

내가 real_function이 패치 os.path을보고 싶어하고, 시험은 정상적인 기능을 볼 수 :

여기 import 나누기 예입니다.

here's the traceback

Answer 1

당신은 컨텍스트 관리자로 patch을 사용할 수 있습니다, 그래서 그것은 단지 with 문 내부의 코드에 적용됩니다

import os 
from mock import patch 

def test(*args): 
    import ipdb; ipdb.set_trace() 
    with patch('os.path.isdir', return_value=False): 
     with patch('os.path.isfile', return_value=False): 
      real_function(some_arguments) 
    pass 

test() 

Answer 2

은 아래 당신이 필요로 작동합니다. 모든 것이 올바른 상태로 반환되었음을 보여주기 위해 테스트 실행 전후에 os.listdir을 출력합니다.

import unittest, os 
from unittest import TestLoader, TextTestRunner 

def tested_func(path): 
    return os.listdir(path * 2) 

class Example(unittest.TestCase): 

    def setUp(self): 
     self._listdir = os.listdir 
     os.listdir = lambda p: ['a', 'b', 'a', 'b'] 

    def test_tested_func(self): 
     self.assertEqual(tested_func('some_path'), ['a', 'b', 'a', 'b']) 

    def tearDown(self): 
     os.listdir = self._listdir 

print os.listdir 
TextTestRunner().run(TestLoader().loadTestsFromTestCase(Example)) 
print os.listdir 

Answer 3

패치는 테스트 메소드 또는 테스트 클래스에서 데코레이터로 사용할 수 있습니다. 아주 깨끗한 코드를 만듭니다. 패치 된 개체가 테스트 메서드로 전달됩니다. 테스트 클래스 수준에서 똑같은 작업을 수행 할 수 있으며 모의 객체는 모든 테스트 메소드에 전달됩니다. 이 경우 setUp 또는 tearDown은 모두 자동으로 처리되므로 사용하지 않아도됩니다.

# module.py 
import os 

def tested_func(path): 
    return os.listdir(path * 2) 


# test.py 
from mock import patch 

@patch('module.os.listdir') 
def test_tested_func(self, mock_listdir): 
    mock_listdir.return_value = ['a', 'b'] 

    self.assertEqual(tested_func('some_path'), ['a', 'b', 'a', 'b']) 
    mock_listdir.assert_called_with('some_path' * 2) 

실제로 테스트 할 모듈에서 패치하려는 기능을 조롱하는 것에주의하십시오.

Answer 4

당신은뿐만 아니라 "수",하지만 당신은 그 http://mock.readthedocs.org/en/latest/patch.html#where-to-patch

추신을 "해야"당신이, 내가 라이브러리 http://mockstar.readthedocs.org/를 사용하는 것이 좋습니다 패치하려는 기기의 side_effects을 설명하기 위해 더 "선언"의 방법을 수행 할 경우