test_seed.py 479 B

12345678910
  1. def test_seed_creates_admin_and_roles(db_session):
  2. from app.services.seed import seed_if_empty
  3. from app.models.system import SysUser, SysRole, SysMenu, SysRoleMenu
  4. seed_if_empty(db_session)
  5. seed_if_empty(db_session) # idempotent
  6. assert db_session.query(SysRole).count() == 3
  7. assert db_session.query(SysUser).filter_by(username="admin").count() == 1
  8. assert db_session.query(SysMenu).count() >= 8
  9. assert db_session.query(SysRoleMenu).count() >= 1