Flask-SQLALchemy 단위 테스트 문제

Question

The Flask Mega-Tutorial을 모방하여 웹 응용 프로그램을 쓰고 있습니다.

내가 단위 테스트 케이스를 코드에 추가하려고했기 때문에. 튜토리얼의 테스트 케이스에 많은 중복 코드가 있다는 것을 알게되었습니다. 여기

는 코드 세그먼트이다 : 그래서

u = User(nickname='john', email='john@example.com') 
db.session.add(u) 
db.session.commit() 

, 나는이 과정을 이동 :

문제는 내가 새로운 케이스를 테스트 할 때마다 나는이 과정을 반복해야한다는 것입니다

def test_avatar(self): 
    u = User(nickname='john', email='john@example.com') 
    avatar = u.avatar(128) 
    expected = 'http://www.gravatar.com/avatar/d4c74594d841139328695756648b6bd6' 
    ... 

def test_make_unique_nickname(self): 
    u = User(nickname='john', email='john@example.com') 
    db.session.add(u) 
    db.session.commit() 
    ... 

다음과 같이 만들었습니다.

import unittest 

from config import basedir 
from app import app, db 
from app.models import User 

u = User(nickname='john', email='john@example.com') # I put this out because some cases may want to use this stuff. 

def _init_database(): 
    db.session.add(u) 
    db.session.commit(u) 

class TestCase(unittest.TestCase): 
    def setUp(self): 
     app.config['TESTING'] = True 
     app.config['WTF_CSRF_ENABLED'] = False 
     app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:') 
     self.app = app.test_client() 
     db.create_all() 
     _init_database() # init database every time 

    def tearDown(self): 
     db.session.remove() 
     db.drop_all() 

    def test_case_1(self): 
     self.assertTrue(User.query.count() == 1) # case 1 

    def test_case_2(self): 
     self.assertTrue(User.query.count() == 1) # case 1 

if __name__ == '__main__': 
    unittest.main() 

두 가지 테스트 케이스가 같습니다. 그러나 한 케이스 만 통과 할 수 있습니다. 다른 하나는 실패합니다.

하지만 u = User(nickname='john', email='john@example.com')_init_database()로 이동하는 경우 : 모든 것은 지금은 괜찮

def _init_database(): 
    u = User(nickname='john', email='john@example.com') 
    db.session.add(u) 
    db.session.commit(u) 

.

나는 정말로 이유를 모른다. 당신이 나를 도울 수?

Answer 1

User 인스턴스는 ORM에 의해 제어되는 개체이므로 전역 변수로 설정하는 것은 좋지 않습니다. 이 개체는 데이터를 포함 할뿐만 아니라 데이터베이스 정보도 포함합니다. 글로벌하게 만들면 첫 번째 테스트에서는 데이터베이스에서 사용하고, 두 번째 테스트에서는 데이터베이스에서 사용합니다.

def _init_database(): 
    u = User(nickname='john', email='john@example.com') 
    db.session.add(u) 
    db.session.commit() 
    return u 

그런 다음 당신이 당신의 테스트 케이스와 테스트에서 액세스를이 사용자를 첨부 할 수 있습니다

더 좋은 방법은 각 시험에 대한 새로운 User 인스턴스를 생성하고 setUp() 방법을 반환하는 것입니다.