# Copyright (c) 2025-2026, RTE (https://www.rte-france.com)
# This Source Code Form is subject to the terms of the Mozilla Public License, version 2.0.
# If a copy of the Mozilla Public License, version 2.0 was not distributed with this file,
# you can obtain one at http://mozilla.org/MPL/2.0/.
# SPDX-License-Identifier: MPL-2.0
# This file is part of Co-Study4Grid a Power Grid Study tool Assistant Interface to help solve contigencies for a grid state under study. 

import pytest
import os
from pathlib import Path
from datetime import datetime
from expert_op4grid_recommender import config
from expert_op4grid_recommender.environment_pypowsybl import setup_environment_configs_pypowsybl

def test_non_reconnectable_detection_with_date():
    """
    Verify that non-reconnectable lines are detected correctly even when an analysis_date is provided.
    This was previously a bug where the detection was bypassed if analysis_date was not None.
    """
    # 1. Setup paths relative to Co-Study4Grid root.
    # The upstream `setup_environment_configs_pypowsybl` reads
    # `config.ENV_FOLDER` (parent data dir) + `config.ENV_NAME` (env
    # subdirectory) and joins them. `config.ENV_PATH` is set for any
    # downstream consumer that reads the full path directly. All three
    # MUST be set together to match the contract in
    # `recommender_service.update_config` (which is what other tests
    # implicitly use via POST /api/config) — otherwise a prior test
    # leaves ENV_FOLDER pointing at the env subdir itself and this
    # test ends up with a doubled-suffix path
    # (`.../data/bare_env_small_grid_test/bare_env_small_grid_test`).
    project_root = Path(__file__).parent.parent.parent
    data_dir = project_root / "data"
    env_subdir = data_dir / "bare_env_small_grid_test"

    if not env_subdir.exists():
        pytest.skip(f"Test data not found at {env_subdir}")

    # 2. Configure the environment — snapshot + override all three vars.
    original_env_name = config.ENV_NAME
    original_env_path = config.ENV_PATH
    original_env_folder = config.ENV_FOLDER

    config.ENV_NAME = "bare_env_small_grid_test"
    config.ENV_FOLDER = data_dir
    config.ENV_PATH = env_subdir
    
    # 3. Use a dummy date - before the fix, this would skip topology-based detection
    dummy_date = datetime(2024, 1, 1)
    
    try:
        # 4. Initialize environment
        # We call the real setup_environment_configs_pypowsybl which now has the fix
        env, obs, env_path, chronic_name, layout, dict_actions, lines_non_reco, lines_care = \
            setup_environment_configs_pypowsybl(analysis_date=dummy_date)
        
        # 5. Verify detected lines
        # Expected lines for bare_env_small_grid_test (from the fix verification)
        expected = {'CRENEL71VIELM', 'GEN.PL73VIELM', 'PYMONL61VOUGL', 'CPVANY632', 'PYMONY632'}
        
        detected_set = set(lines_non_reco)
        
        missing = expected - detected_set
        assert not missing, f"Non-reconnectable lines missing from detection: {missing}. Detected: {detected_set}"
        
        print(f"Verified: {len(expected)} lines correctly detected with date={dummy_date}")
        
    finally:
        # Restore configuration
        config.ENV_NAME = original_env_name
        config.ENV_PATH = original_env_path
        config.ENV_FOLDER = original_env_folder

if __name__ == "__main__":
    test_non_reconnectable_detection_with_date()