26 lines
554 B
Python
26 lines
554 B
Python
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
|
|
import os
|
|
import unittest
|
|
|
|
from config import basedir
|
|
from app import app
|
|
from app import db
|
|
|
|
class TestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
app.config['TESTING'] = True
|
|
app.config['WTF_CSRF_ENABLED'] = False
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + os.path.join(basedir, 'test.db')
|
|
self.app = app.test_client()
|
|
db.create_all()
|
|
|
|
def tearDown(self):
|
|
db.session.remove()
|
|
db.drop_all()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|