25 lines
532 B
Python
25 lines
532 B
Python
|
# project/server/tests/base.py
|
||
|
|
|
||
|
|
|
||
|
from flask_testing import TestCase
|
||
|
|
|
||
|
|
from project.server import app, db
|
||
|
|
from project.server.models import User
|
||
|
|
|
||
|
|
|
||
|
|
class BaseTestCase(TestCase):
|
||
|
|
|
||
|
|
def create_app(self):
|
||
|
|
app.config.from_object('project.server.config.TestingConfig')
|
||
|
|
return app
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
db.create_all()
|
||
|
|
user = User(email="ad@min.com", password="admin_user")
|
||
|
|
db.session.add(user)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
db.session.remove()
|
||
|
|
db.drop_all()
|