Django-Nose 테스트 기능의 데이터베이스 변경 사항을 정리하는 방법은 무엇입니까?

Question

우리는 다양한 이유로 코 기능 테스트를 사용하여 테스트 스위트를 작성합니다.

Django 응용 프로그램 용 테스트 스위트를 실행할 때 이러한 테스트에서 데이터가 누출되지 않도록해야합니다 (예 : django.test.TestCase). 이는 연결로 이어지고 오류를 진단하기가 어렵 기 때문입니다.

이 문제를 해결하는 가장 확실한 방법은 나중에 정리할 테스트를 둘러 쌀 수있는 데코레이터입니다. 그러나 다른 솔루션으로 우리가 원하는 것을 얻을 수 있다면 결혼하지 않을 것입니다.

우리는 PostgreSQL에서 실행되므로 Postgres 관련 솔루션이 좋을 것입니다.

Answer 1

나는이 오늘보고 약간의 시간을 보냈어요, 다음과 같은 장식과 함께 올라와있다 : 장고 테스트 클라이언트없이 트랜잭션을 종료하지 않도록

from functools import wraps 

from django.db import transaction 
from mock import patch 

def rollback_db_changes(func): 
    """Decorate a function so that it will be rolled back once completed.""" 
    @wraps(func) 
    @transaction.commit_manually 
    def new_f(*args, **kwargs): 
     def fake_commit(using=None): 
      # Don't properly commit the transaction, so we can roll it back 
      transaction.set_clean(using) 
     patcher = patch('django.db.transaction.commit', fake_commit) 
     patcher.start() 
     try: 
      return func(*args, **kwargs) 
     finally: 
      patcher.stop() 
      transaction.rollback() 
    return new_f 

우리는 패치를 수행 우리는 그것을 되돌릴 수 있습니다. 중복 된 사용자 이름을 가진 사용자를 만들 수 없습니다 실행

from django.contrib.auth.models import User 

@rollback_db_changes 
def test_allowed_access(): 
    user = User.objects.create(username='test_user') 
    eq_(1, User.objects.count()) 


@rollback_db_changes 
def test_allowed_access_2(): 
    user = User.objects.create(username='test_user') 
    eq_(1, User.objects.count()) 

이전에 두 번째 테스트를이 다음과 같은 테스트를 통과 할 수 있습니다.