"""Tests for Streamlit console display functionality."""

from unittest.mock import patch


def test_scrollable_console_basic_text():
    """Test scrollable_console renders basic text correctly."""
    # Import after mocking to avoid Streamlit initialization
    with patch('streamlit_sftp_interface_v2.components') as mock_components:
        with patch('streamlit_sftp_interface_v2.st') as mock_st:
            mock_st.session_state.get.return_value = True  # auto_scroll enabled

            from streamlit_sftp_interface_v2 import scrollable_console

            test_text = "Line 1\nLine 2\nLine 3"
            scrollable_console(test_text, height=300, container_id="test_console")

            # Verify components.html was called
            assert mock_components.html.called
            html_content = mock_components.html.call_args[0][0]

            # Check that text appears in HTML (escaped)
            assert "Line 1" in html_content
            assert "Line 2" in html_content
            assert "Line 3" in html_content

            # Check styling
            assert "height: 300px" in html_content
            assert "overflow-y: auto" in html_content
            assert "background-color: #1e1e1e" in html_content
            assert "color: #d4d4d4" in html_content


def test_scrollable_console_html_escaping():
    """Test that HTML special characters are properly escaped."""
    with patch('streamlit_sftp_interface_v2.components') as mock_components:
        with patch('streamlit_sftp_interface_v2.st') as mock_st:
            mock_st.session_state.get.return_value = True

            from streamlit_sftp_interface_v2 import scrollable_console

            # Text with HTML special characters
            dangerous_text = "<script>alert('XSS')</script> & \"quotes\""
            scrollable_console(dangerous_text, container_id="escape_test")

            html_content = mock_components.html.call_args[0][0]

            # Verify escaping in the content div
            assert "&lt;script&gt;" in html_content
            assert "&amp;" in html_content
            assert "&quot;" in html_content
            # Verify the user's dangerous text is escaped in content
            assert "alert(&#x27;XSS&#x27;)" in html_content
            # The auto-scroll script tag is OK (it's ours)
            assert "consoleDiv.scrollTop" in html_content


def test_scrollable_console_auto_scroll_enabled():
    """Test auto-scroll JavaScript is included when enabled."""
    with patch('streamlit_sftp_interface_v2.components') as mock_components:
        with patch('streamlit_sftp_interface_v2.st') as mock_st:
            mock_st.session_state.get.return_value = True  # auto_scroll enabled

            from streamlit_sftp_interface_v2 import scrollable_console

            scrollable_console("Test text", container_id="auto_scroll_test")

            html_content = mock_components.html.call_args[0][0]

            # Verify auto-scroll script is present
            assert "<script>" in html_content
            assert "scrollTop" in html_content
            assert "scrollHeight" in html_content
            assert "auto_scroll_test" in html_content


def test_scrollable_console_auto_scroll_disabled():
    """Test auto-scroll JavaScript is NOT included when disabled."""
    with patch('streamlit_sftp_interface_v2.components') as mock_components:
        with patch('streamlit_sftp_interface_v2.st') as mock_st:
            mock_st.session_state.get.return_value = False  # auto_scroll disabled

            from streamlit_sftp_interface_v2 import scrollable_console

            scrollable_console("Test text", container_id="no_auto_scroll")

            html_content = mock_components.html.call_args[0][0]

            # Verify auto-scroll script is NOT present
            assert "<script>" not in html_content
            assert "scrollTop" not in html_content


def test_scrollable_console_custom_height():
    """Test custom height parameter is applied."""
    with patch('streamlit_sftp_interface_v2.components') as mock_components:
        with patch('streamlit_sftp_interface_v2.st') as mock_st:
            mock_st.session_state.get.return_value = True

            from streamlit_sftp_interface_v2 import scrollable_console

            custom_height = 500
            scrollable_console("Test", height=custom_height, container_id="height_test")

            html_content = mock_components.html.call_args[0][0]

            # Check height in div style
            assert f"height: {custom_height}px" in html_content

            # Check height parameter to components.html
            height_param = mock_components.html.call_args[1]['height']
            assert height_param == custom_height + 30


def test_scrollable_console_unique_container_ids():
    """Test unique container IDs prevent ID collisions."""
    with patch('streamlit_sftp_interface_v2.components') as mock_components:
        with patch('streamlit_sftp_interface_v2.st') as mock_st:
            mock_st.session_state.get.return_value = True

            from streamlit_sftp_interface_v2 import scrollable_console

            # Call with different container IDs
            scrollable_console("Text 1", container_id="console_1")
            html_1 = mock_components.html.call_args[0][0]

            scrollable_console("Text 2", container_id="console_2")
            html_2 = mock_components.html.call_args[0][0]

            # Verify unique IDs
            assert 'id="console_1"' in html_1
            assert 'id="console_2"' in html_2


def test_log_to_console():
    """Test log_to_console appends formatted messages."""
    with patch('streamlit_sftp_interface_v2.st') as mock_st:
        with patch('streamlit_sftp_interface_v2.datetime') as mock_datetime:
            mock_st.session_state.console_output = []
            mock_datetime.now.return_value.strftime.return_value = "12:34:56"

            from streamlit_sftp_interface_v2 import log_to_console

            log_to_console("Test message", level="INFO")

            assert len(mock_st.session_state.console_output) == 1
            log_entry = mock_st.session_state.console_output[0]
            assert "[12:34:56]" in log_entry
            assert "INFO:" in log_entry
            assert "Test message" in log_entry


def test_get_console_text():
    """Test get_console_text joins all log entries."""
    with patch('streamlit_sftp_interface_v2.st') as mock_st:
        mock_st.session_state.console_output = [
            "[10:00:00] INFO: First",
            "[10:00:01] INFO: Second",
            "[10:00:02] INFO: Third"
        ]

        from streamlit_sftp_interface_v2 import get_console_text

        result = get_console_text()

        assert "First" in result
        assert "Second" in result
        assert "Third" in result
        assert result.count("\n") == 2  # Two newlines joining three lines


def test_clear_console():
    """Test clear_console empties the output list."""
    with patch('streamlit_sftp_interface_v2.st') as mock_st:
        mock_st.session_state.console_output = ["Line 1", "Line 2", "Line 3"]

        from streamlit_sftp_interface_v2 import clear_console

        clear_console()

        assert mock_st.session_state.console_output == []


def test_scrollable_console_copy_friendly():
    """Test console has copy-friendly styling."""
    with patch('streamlit_sftp_interface_v2.components') as mock_components:
        with patch('streamlit_sftp_interface_v2.st') as mock_st:
            mock_st.session_state.get.return_value = True

            from streamlit_sftp_interface_v2 import scrollable_console

            scrollable_console("Copyable text", container_id="copy_test")

            html_content = mock_components.html.call_args[0][0]

            # Verify user-select styles for copy-friendly text
            assert "user-select: text" in html_content
            assert "-webkit-user-select: text" in html_content
            assert "-moz-user-select: text" in html_content
            assert "-ms-user-select: text" in html_content