"""
Test that imports work correctly when package is installed
This helps verify CI setup is correct
"""

import pytest


def test_direct_import():
    """Test that gemini_server can be imported directly"""
    try:
        import gemini_server
        assert hasattr(gemini_server, 'GeminiChatRequest')
        assert hasattr(gemini_server, 'CodeAnalysisRequest')
        assert hasattr(gemini_server, 'handle_list_tools')
        assert hasattr(gemini_server, 'handle_call_tool')
    except ImportError as e:
        pytest.fail(f"Failed to import gemini_server: {e}")


def test_from_import():
    """Test that specific items can be imported from gemini_server"""
    try:
        from gemini_server import (
            GeminiChatRequest,
            CodeAnalysisRequest,
            DEFAULT_MODEL,
            DEVELOPER_SYSTEM_PROMPT
        )
        assert GeminiChatRequest is not None
        assert CodeAnalysisRequest is not None
        assert isinstance(DEFAULT_MODEL, str)
        assert isinstance(DEVELOPER_SYSTEM_PROMPT, str)
    except ImportError as e:
        pytest.fail(f"Failed to import from gemini_server: {e}")


def test_google_generativeai_import():
    """Test that google.generativeai can be imported"""
    try:
        import google.generativeai as genai
        assert hasattr(genai, 'GenerativeModel')
        assert hasattr(genai, 'configure')
    except ImportError as e:
        pytest.fail(f"Failed to import google.generativeai: {e}")