"""
Tests for session analytics

Issue: #753
"""

import tempfile
import unittest
from pathlib import Path
from unittest.mock import patch

from agent.session_analytics import (
    SessionTracker,
    SessionStats,
    format_stats,
    get_daily_stats,
    format_daily_report,
)


class TestSessionStats(unittest.TestCase):
    
    def test_defaults(self):
        stats = SessionStats(session_id="test", start_time="2026-01-01")
        self.assertEqual(stats.input_tokens, 0)
        self.assertEqual(stats.output_tokens, 0)
        self.assertEqual(stats.tool_calls, 0)
    
    def test_to_dict(self):
        stats = SessionStats(session_id="test", start_time="2026-01-01")
        d = stats.to_dict()
        self.assertEqual(d["session_id"], "test")
        self.assertIn("input_tokens", d)


class TestSessionTracker(unittest.TestCase):
    
    def test_record_tokens(self):
        tracker = SessionTracker("test", provider="openai")
        tracker.record_tokens(100, 50)
        stats = tracker.get_current_stats()
        self.assertEqual(stats.input_tokens, 100)
        self.assertEqual(stats.output_tokens, 50)
        self.assertGreater(stats.estimated_cost_usd, 0)
    
    def test_record_tool_call(self):
        tracker = SessionTracker("test")
        tracker.record_tool_call(success=True)
        tracker.record_tool_call(success=False)
        stats = tracker.get_current_stats()
        self.assertEqual(stats.tool_calls, 2)
        self.assertEqual(stats.tool_errors, 1)
    
    def test_free_provider(self):
        tracker = SessionTracker("test", provider="ollama")
        tracker.record_tokens(1000, 500)
        stats = tracker.get_current_stats()
        self.assertEqual(stats.estimated_cost_usd, 0.0)
    
    def test_finish(self):
        tracker = SessionTracker("test")
        stats = tracker.finish()
        self.assertIsNotNone(stats.end_time)
        self.assertGreater(stats.wall_time_seconds, 0)


class TestFormatStats(unittest.TestCase):
    
    def test_format(self):
        stats = SessionStats(
            session_id="test123",
            start_time="2026-01-01",
            input_tokens=1000,
            output_tokens=500,
            total_tokens=1500,
            tool_calls=5,
            tool_errors=1,
            wall_time_seconds=30.5,
            api_calls=3
        )
        formatted = format_stats(stats)
        self.assertIn("1,000", formatted)
        self.assertIn("500", formatted)


class TestDailyStats(unittest.TestCase):
    
    def test_empty(self):
        with patch("agent.session_analytics.ANALYTICS_DIR", Path(tempfile.mkdtemp())):
            stats = get_daily_stats("2020-01-01")
            self.assertEqual(stats["sessions"], 0)
    
    def test_format_report(self):
        stats = {
            "date": "2026-04-14",
            "sessions": 10,
            "total_tokens": 50000,
            "total_cost_usd": 0.50,
            "total_wall_time_seconds": 300,
            "total_tool_calls": 100,
            "total_tool_errors": 5,
            "avg_tokens_per_session": 5000,
            "avg_cost_per_session": 0.05,
        }
        report = format_daily_report(stats)
        self.assertIn("10", report)
        self.assertIn("50,000", report)


if __name__ == "__main__":
    unittest.main()