kappa123/kappa123/tests/base.py

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()