Timeline_Generator/tests/unit/test_renderer.py

"""
時間軸渲染模組單元測試

對應 TDD.md:
- UT-REN-01: 時間刻度演算法
- UT-REN-02: 節點避碰演算法

Version: 1.0.0
DocID: TDD-UT-REN-001
"""

import pytest
from datetime import datetime, timedelta
from backend.schemas import Event, TimelineConfig, RenderResult, EventType
from backend.renderer import (
    TimeScaleCalculator, CollisionResolver, ThemeManager,
    TimelineRenderer, TimeUnit
)


class TestTimeScaleCalculator:
    """時間刻度演算法測試"""

    def test_calculate_time_range(self):
        """測試時間範圍計算"""
        events = [
            Event(id="1", title="E1", start=datetime(2024, 1, 1)),
            Event(id="2", title="E2", start=datetime(2024, 1, 10))
        ]

        start, end = TimeScaleCalculator.calculate_time_range(events)

        assert start < datetime(2024, 1, 1)
        assert end > datetime(2024, 1, 10)

    def test_determine_time_unit_days(self):
        """測試天級別刻度判斷"""
        start = datetime(2024, 1, 1)
        end = datetime(2024, 1, 7)

        unit = TimeScaleCalculator.determine_time_unit(start, end)

        assert unit == TimeUnit.DAY

    def test_determine_time_unit_weeks(self):
        """測試週級別刻度判斷"""
        start = datetime(2024, 1, 1)
        end = datetime(2024, 3, 1)  # 約 2 個月

        unit = TimeScaleCalculator.determine_time_unit(start, end)

        assert unit == TimeUnit.WEEK

    def test_determine_time_unit_months(self):
        """測試月級別刻度判斷"""
        start = datetime(2024, 1, 1)
        end = datetime(2024, 6, 1)  # 6 個月

        unit = TimeScaleCalculator.determine_time_unit(start, end)

        assert unit == TimeUnit.MONTH

    def test_generate_tick_values_days(self):
        """測試天級別刻度生成"""
        start = datetime(2024, 1, 1)
        end = datetime(2024, 1, 5)

        ticks = TimeScaleCalculator.generate_tick_values(start, end, TimeUnit.DAY)

        assert len(ticks) >= 5
        assert all(isinstance(t, datetime) for t in ticks)

    def test_generate_tick_values_months(self):
        """測試月級別刻度生成"""
        start = datetime(2024, 1, 1)
        end = datetime(2024, 6, 1)

        ticks = TimeScaleCalculator.generate_tick_values(start, end, TimeUnit.MONTH)

        assert len(ticks) >= 6
        # 驗證是每月第一天
        assert all(t.day == 1 for t in ticks)


class TestCollisionResolver:
    """節點避碰演算法測試"""

    def test_no_overlapping_events(self):
        """測試無重疊事件"""
        events = [
            Event(id="1", title="E1", start=datetime(2024, 1, 1), end=datetime(2024, 1, 2)),
            Event(id="2", title="E2", start=datetime(2024, 1, 3), end=datetime(2024, 1, 4))
        ]

        resolver = CollisionResolver()
        layers = resolver.resolve_collisions(events)

        # 無重疊，都在第 0 層
        assert layers["1"] == 0
        assert layers["2"] == 0

    def test_overlapping_events(self):
        """測試重疊事件分層"""
        events = [
            Event(id="1", title="E1", start=datetime(2024, 1, 1), end=datetime(2024, 1, 5)),
            Event(id="2", title="E2", start=datetime(2024, 1, 3), end=datetime(2024, 1, 7))
        ]

        resolver = CollisionResolver()
        layers = resolver.resolve_collisions(events)

        # 重疊，應該在不同層
        assert layers["1"] != layers["2"]

    def test_group_based_layout(self):
        """測試基於群組的排版"""
        events = [
            Event(id="1", title="E1", start=datetime(2024, 1, 1), group="A"),
            Event(id="2", title="E2", start=datetime(2024, 1, 1), group="B")
        ]

        resolver = CollisionResolver()
        layers = resolver.group_based_layout(events)

        # 不同群組，應該在不同層
        assert layers["1"] != layers["2"]

    def test_empty_events(self):
        """測試空事件列表"""
        resolver = CollisionResolver()
        layers = resolver.resolve_collisions([])

        assert layers == {}


class TestThemeManager:
    """主題管理器測試"""

    def test_get_modern_theme(self):
        """測試現代主題"""
        from backend.schemas import ThemeStyle
        theme = ThemeManager.get_theme(ThemeStyle.MODERN)

        assert 'background' in theme
        assert 'text' in theme
        assert 'primary' in theme

    def test_get_all_themes(self):
        """測試所有主題可用性"""
        from backend.schemas import ThemeStyle

        for style in ThemeStyle:
            theme = ThemeManager.get_theme(style)
            assert theme is not None
            assert 'background' in theme


class TestTimelineRenderer:
    """時間軸渲染器測試"""

    def test_render_basic_timeline(self):
        """測試基本時間軸渲染"""
        events = [
            Event(id="1", title="Event 1", start=datetime(2024, 1, 1)),
            Event(id="2", title="Event 2", start=datetime(2024, 1, 5))
        ]
        config = TimelineConfig()

        renderer = TimelineRenderer()
        result = renderer.render(events, config)

        assert result.success is True
        assert 'data' in result.data
        assert result.layout is not None

    def test_render_empty_timeline(self):
        """測試空白時間軸渲染"""
        renderer = TimelineRenderer()
        result = renderer.render([], TimelineConfig())

        assert result.success is True
        assert 'data' in result.data

    def test_render_with_horizontal_direction(self):
        """測試水平方向渲染"""
        events = [Event(id="1", title="E1", start=datetime(2024, 1, 1))]
        config = TimelineConfig(direction='horizontal')

        renderer = TimelineRenderer()
        result = renderer.render(events, config)

        assert result.success is True

    def test_render_with_vertical_direction(self):
        """測試垂直方向渲染"""
        events = [Event(id="1", title="E1", start=datetime(2024, 1, 1))]
        config = TimelineConfig(direction='vertical')

        renderer = TimelineRenderer()
        result = renderer.render(events, config)

        assert result.success is True

    def test_render_with_different_themes(self):
        """測試不同主題渲染"""
        from backend.schemas import ThemeStyle
        events = [Event(id="1", title="E1", start=datetime(2024, 1, 1))]

        renderer = TimelineRenderer()

        for theme in [ThemeStyle.MODERN, ThemeStyle.CLASSIC]:
            config = TimelineConfig(theme=theme)
            result = renderer.render(events, config)
            assert result.success is True

    def test_render_with_grid(self):
        """測試顯示網格"""
        events = [Event(id="1", title="E1", start=datetime(2024, 1, 1))]
        config = TimelineConfig(show_grid=True)

        renderer = TimelineRenderer()
        result = renderer.render(events, config)

        assert result.success is True

    def test_render_single_event(self):
        """測試單一事件渲染"""
        events = [Event(id="1", title="Single", start=datetime(2024, 1, 1))]
        config = TimelineConfig()

        renderer = TimelineRenderer()
        result = renderer.render(events, config)

        assert result.success is True
        assert len(result.data['data']) == 1

    def test_hover_text_generation(self):
        """測試提示訊息生成"""
        event = Event(
            id="1",
            title="Test Event",
            start=datetime(2024, 1, 1),
            end=datetime(2024, 1, 2),
            description="Test description"
        )

        renderer = TimelineRenderer()
        hover_text = renderer._generate_hover_text(event)

        assert "Test Event" in hover_text
        assert "Test description" in hover_text


if __name__ == "__main__":
    pytest.main([__file__, "-v"])