先上代码
import os
import sys
import click
from app import create_app, db
from app.models import User, Role, Permission, Follow, Post, Comment
from flask_migrate import Migrate
COV = None
if os.environ.get("FLASK_COVERAGE"):
import coverage
COV = coverage.coverage(branch=True, include="app/*")
COV.start()
app = create_app("default")
migrate = Migrate(app, db)
@app.cli.command()
@click.option("--coverage/--no-coverage", default=False, help="Run tests under code coverage")
def test(coverage):
"""Run the unit tests."""
if coverage and not os.environ.get("FLASK_COVERAGE"):
os.environ["FLASK_COVERAGE"] = "1"
os.execvp(sys.executable, [sys.executable]+sys.argv)
import unittest
tests = unittest.TestLoader().discover("tests")
unittest.TextTestR