def setUp(self):
self.app = create_app('testing')
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
Role.insert_roles()
Gender.insert_genders()
def setUp(self):
self.app = create_app('testing')
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
Role.insert_roles()
Gender.insert_genders()
self.client = self.app.test_client(use_cookies=True)
def setUpClass(cls):
# start Chrome
options = webdriver.ChromeOptions()
options.add_argument('headless')
try:
driver_path = '/path/to/chromedriver'
cls.client = webdriver.Chrome(driver_path, options=options)
except:
pass
# skip these tests if the browser could not be started
if cls.client:
# create the application
cls.app = create_app('testing')
cls.app_context = cls.app.app_context()
cls.app_context.push()
# suppress logging to keep unittest output clean
import logging
logger = logging.getLogger('werkzeug')
logger.setLevel("ERROR")
# create the database and populate with some fake data
db.create_all()
Role.insert_roles()
Gender.insert_genders()
fake.users(10)
fake.posts(10)
# add an administrator user
g = Gender.query.filter_by(name='Male').first()
admin_role = Role.query.filter_by(name='Administrator').first()
admin = User(email='*****@*****.**',
name='john',
password='******',
role=admin_role,
confirmed=True,
gender=g)
db.session.add(admin)
db.session.commit()
# start the Flask server in a thread
cls.server_thread = threading.Thread(target=cls.app.run,
kwargs={'debug': False})
cls.server_thread.start()
# give the server a second to ensure it is up
time.sleep(1)
from app_core import create_app, db
from app_core.models import User, Role, Gender, Permission, Follow, Comment, Post
dot_env_path = os.path.join(os.path.dirname(__file__), '.env')
if os.path.exists(dot_env_path):
load_dotenv(dot_env_path)
COV = None
if os.environ.get('FLASK_COVERAGE'):
import coverage
COV = coverage.coverage(branch=True, include='app_core/*')
COV.start()
app = create_app(os.getenv('FLASK_CONFIG') or 'default')
migrate = Migrate(app, db)
@app.shell_context_processor
def make_shell_context():
return dict(db=db,
Gender=Gender,
Permission=Permission,
User=User,
Role=Role,
Follow=Follow,
Post=Post,
Comment=Comment)
def setUp(self):
self.app = create_app('testing')
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
