def db(app, request):
"""Session-wide test database."""
if os.path.exists(TEST_DB_PATH):
os.unlink(TEST_DB_PATH)
def teardown():
_db.drop_all()
os.unlink(TEST_DB_PATH)
_db.app = app
_db.create_all()
request.addfinalizer(teardown)
return _db
def cleanup_db():
testing_config = TestingConfig()
db_host = testing_config.DB_HOST
db_name = testing_config.DB_NAME
engine = create_engine(f'postgresql://{db_host}')
conn = engine.connect()
try:
conn.execution_options(
isolation_level="AUTOCOMMIT").execute(f"CREATE DATABASE {db_name}")
except Exception:
pass
rv = create_engine(f'postgresql://{db_host}/{db_name}')
db.create_all(rv)
yield rv
db.drop_all(rv)
rv.dispose()
def create_db():
db.create_all()
return "创建完毕"
from myapp.models import db, User, Token
db.create_all()
user = User(email='*****@*****.**',
password='******')
db.session.add(user)
db.session.commit()
token = Token(user_id=user.id)
db.session.add(token)
db.session.commit()
print ('token', token.value)
def setUp(self):
# Setup the database
db.create_all()
# Rollback any lingering transactions
db.session.rollback()
def initialize_database():
db.create_all()
def setUp(self):
app.config['TESTING'] = True
app.config['CSRF_ENABLED'] = False
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////tmp/test2.db'
self.app = app.test_client()
db.create_all()