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 from app import initialize_scodoc_database, clear_scodoc_cache 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, TestConfig.DEPT_TEST) db.session.add(u) db.session.commit() ndb.set_sco_dept(TestConfig.DEPT_TEST) # set db connection yield client # ndb.close_dept_connection() # Teardown: db.session.commit() db.session.remove() clear_scodoc_cache() # db.drop_all() # => laisse la base en état (l'efface au début) # utile pour les tests en cours de développement