kappa123/kappa123/tests/base.py

25 lines
536 B
Python

# kappa123/server/tests/base.py
from flask_testing import TestCase
from kappa123.server import app, db
from kappa123.server.models import User
class BaseTestCase(TestCase):
def create_app(self):
app.config.from_object('kappa123.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()