파이썬/장고 유닛 테스트 기능 재정의

Question

는 내가 테스트를 필요로하는 그 안에 반복이 아닌 미리 정의 된 번호로 시간이 많이 걸리는 방법이있다. 하나의 반복으로 만이 방법을 테스트하는 것이 가장 좋은 방법은 무엇입니까?

Answer 1

아마도 그런 다음 테스트에서 사용자 정의 OBJS 매개 변수를 호출 할 수 있습니다

def my_method(self, objs=None): 
    if objs is None: 
     objs = get_objects() 
    for obj in objs: 
     do_something(obj) 

에 당신의 방법을 설정합니다.

Answer 2

는 업데이트 : 전화 개체가 하나의 개체를 반환받을

람다 아웃 :

는 그래서 여기에 내가 소스 코드를 변경하지 않고 문제를 해결할 방법은, 원래의 질문을 오해. 예 :

from your_module import get_objects 

def test_testmethdod(self): 
    original_get_objects = get_objects 
    one_object = YourObject() 
    get_objects = lambda : [one_object,] 

    # ... 
    # your tests here 
    # ... 

    # Reset the original function, so it doesn't mess up other tests 
    get_objects = original_get_objects