ScoDoc/tests/conftest.py

52 lines
1.9 KiB
Python
Raw Normal View History

import pytest
from flask import g
from flask_login import login_user, logout_user, current_user
from config import TestConfig
from app import db, create_app, initialize_scodoc_database
2021-08-09 17:01:47 +02:00
from app import models
from app.auth.models import User, Role, UserRole, Permission
from app.scodoc import sco_bulletins_standard
from app.scodoc import notesdb as ndb
@pytest.fixture()
def test_client():
# Setup
apptest = create_app(TestConfig)
# Run tests:
with apptest.test_client() as client:
with apptest.app_context():
with apptest.test_request_context():
# erase and reset database:
initialize_scodoc_database(erase=True)
# Loge l'utilisateur super-admin
admin_role = Role.query.filter_by(name="SuperAdmin").first()
assert admin_role
admin_user = (
User.query.join(UserRole)
.filter(
(UserRole.user_id == User.id)
& (UserRole.role_id == admin_role.id)
)
.first()
)
assert admin_user
login_user(admin_user)
# Vérifie que l'utilisateur "bach" existe
u = User.query.filter_by(user_name="bach").first()
if u is None:
u = User(user_name="bach")
if not "Admin" in {r.name for r in u.roles}:
admin_role = Role.query.filter_by(name="Admin").first()
u.add_role(admin_role, "TEST00")
db.session.add(u)
2021-07-31 22:57:54 +02:00
ndb.set_sco_dept("TEST00") # set db connection
truncate_database() # erase tables
yield client
# ndb.close_dept_connection()
# Teardown:
db.session.remove()
db.drop_all()