import pytest
import sys
import os
import importlib.util

# Ensure project root is in path
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))

# Load gui.py
spec = importlib.util.spec_from_file_location("gui", "gui.py")
gui = importlib.util.module_from_spec(spec)
sys.modules["gui"] = gui
spec.loader.exec_module(gui)
from gui import App

def test_new_hubs_defined_in_window_info():
    """
    Verifies that the new consolidated Hub windows are defined in the App's window_info.
    This ensures they will be available in the 'Windows' menu.
    """
    # We don't need a full App instance with DPG context for this, 
    # as window_info is initialized in __init__ before DPG starts.
    # But we mock load_config to avoid file access.
    from unittest.mock import patch
    with patch('gui.load_config', return_value={}):
        app = App()
        
    expected_hubs = {
        "Context Hub": "win_context_hub",
        "AI Settings Hub": "win_ai_settings_hub",
        "Discussion Hub": "win_discussion_hub",
        "Operations Hub": "win_operations_hub",
    }
    
    for label, tag in expected_hubs.items():
        assert tag in app.window_info.values(), f"Expected window tag {tag} not found in window_info"
        # Check if the label matches (or is present)
        found = False
        for l, t in app.window_info.items():
            if t == tag:
                found = True
                assert l == label or label in l, f"Label mismatch for {tag}: expected {label}, found {l}"
        assert found, f"Expected window label {label} not found in window_info"

def test_old_windows_removed_from_window_info(app_instance_simple):
    """
    Verifies that the old fragmented windows are removed from window_info.
    """
    old_tags = [
        "win_projects", "win_files", "win_screenshots",
        "win_provider", "win_system_prompts",
        "win_discussion", "win_message", "win_response",
        "win_comms", "win_tool_log", "win_diagnostics"
    ]
    
    for tag in old_tags:
        assert tag not in app_instance_simple.window_info.values(), f"Old window tag {tag} should have been removed from window_info"

@pytest.fixture
def app_instance_simple():
    from unittest.mock import patch
    from gui import App
    with patch('gui.load_config', return_value={}):
        app = App()
    return app

def test_hub_windows_have_correct_flags(app_instance_simple):
    """
    Verifies that the new Hub windows have appropriate flags for a professional workspace.
    (e.g., no_collapse should be True for main hubs).
    """
    import dearpygui.dearpygui as dpg
    dpg.create_context()
    
    # We need to actually call the build methods to check the configuration
    app_instance_simple._build_context_hub()
    app_instance_simple._build_ai_settings_hub()
    app_instance_simple._build_discussion_hub()
    app_instance_simple._build_operations_hub()
    
    hubs = ["win_context_hub", "win_ai_settings_hub", "win_discussion_hub", "win_operations_hub"]
    for hub in hubs:
        assert dpg.does_item_exist(hub)
        # We can't easily check 'no_collapse' after creation without internal DPG calls 
        # but we can check if it's been configured if we mock dpg.window or check it manually
    
    dpg.destroy_context()

def test_indicators_exist(app_instance_simple):
    """
    Verifies that the new thinking and live indicators exist in the UI.
    """
    import dearpygui.dearpygui as dpg
    dpg.create_context()
    
    app_instance_simple._build_discussion_hub()
    app_instance_simple._build_operations_hub()
    
    assert dpg.does_item_exist("thinking_indicator")
    assert dpg.does_item_exist("operations_live_indicator")
    
    dpg.destroy_context()