import shutil

import admin_analytics.config as config
from admin_analytics.config import UD_UNITID
from admin_analytics.ipeds.staff import load_staff


def test_load_staff_filters_to_ud(db_conn, fixtures_dir, tmp_path):
    ipeds_dir = tmp_path / "ipeds" / "staff" / "2023"
    ipeds_dir.mkdir(parents=True)
    shutil.copy(fixtures_dir / "s2023.csv", ipeds_dir / "s2023.csv")

    original = config.IPEDS_DATA_DIR
    config.IPEDS_DATA_DIR = tmp_path / "ipeds"
    try:
        count = load_staff(db_conn, range(2023, 2024), unitid_filter=UD_UNITID)
    finally:
        config.IPEDS_DATA_DIR = original

    assert count == 1
    row = db_conn.execute(
        "SELECT total_staff, faculty_total, management_total FROM raw_ipeds_staff WHERE unitid = ?",
        [UD_UNITID],
    ).fetchone()
    # faculty_total=OCCUPCAT 200 (instruction/research), management_total=OCCUPCAT 300
    assert row == (5081, 1271, 666)