52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
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
|
|
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)
|
|
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()
|