#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
tests/test_callout_edges.py
Unit-Tests für extract_callout_relations Funktion.
Testet einfache und verschachtelte Callout-Formate.
"""

import unittest
from app.core.graph.graph_extractors import extract_callout_relations


class TestCalloutEdges(unittest.TestCase):
    """Testet die Extraktion von Edge-Callouts aus Markdown."""

    def test_simple_callout_single_target(self):
        """Test: Einfaches Callout mit einem Target."""
        text = """> [!edge] related_to
> [[Target]]"""
        
        pairs, remaining = extract_callout_relations(text)
        
        self.assertEqual(len(pairs), 1)
        self.assertEqual(pairs[0], ("related_to", "Target"))
        self.assertNotIn("[!edge]", remaining)

    def test_simple_callout_multiple_targets(self):
        """Test: Einfaches Callout mit mehreren Targets."""
        text = """> [!edge] related_to
> [[Target1]]
> [[Target2]]
> [[Target3]]"""
        
        pairs, remaining = extract_callout_relations(text)
        
        self.assertEqual(len(pairs), 3)
        self.assertIn(("related_to", "Target1"), pairs)
        self.assertIn(("related_to", "Target2"), pairs)
        self.assertIn(("related_to", "Target3"), pairs)

    def test_nested_callout_single_edge(self):
        """Test: Verschachteltes Callout mit einem Edge-Typ."""
        text = """> [!abstract]- 🕸️ Semantic Mapping
>> [!edge] related_to
>> [[Brigitte]]
>> [[Klaus]]
>> [[Salami Fertigpizza]]"""
        
        pairs, remaining = extract_callout_relations(text)
        
        self.assertEqual(len(pairs), 3)
        self.assertIn(("related_to", "Brigitte"), pairs)
        self.assertIn(("related_to", "Klaus"), pairs)
        self.assertIn(("related_to", "Salami Fertigpizza"), pairs)

    def test_nested_callout_multiple_edges(self):
        """Test: Verschachteltes Callout mit mehreren Edge-Typen."""
        text = """> [!abstract]- 🕸️ Semantic Mapping
>> [!edge] related_to
>> [[Brigitte]]
>> [[Klaus]]
>> [[Salami Fertigpizza]]
>
>>  [!edge] derived_from
>> [[Link 2]]"""
        
        pairs, remaining = extract_callout_relations(text)
        
        # Prüfe related_to Edges
        related_pairs = [p for p in pairs if p[0] == "related_to"]
        self.assertEqual(len(related_pairs), 3)
        self.assertIn(("related_to", "Brigitte"), pairs)
        self.assertIn(("related_to", "Klaus"), pairs)
        self.assertIn(("related_to", "Salami Fertigpizza"), pairs)
        
        # Prüfe derived_from Edge
        derived_pairs = [p for p in pairs if p[0] == "derived_from"]
        self.assertEqual(len(derived_pairs), 1)
        self.assertIn(("derived_from", "Link 2"), pairs)

    def test_mixed_callouts(self):
        """Test: Gemischte einfache und verschachtelte Callouts."""
        text = """> [!edge] depends_on
> [[Dependency1]]

> [!abstract] Test
>> [!edge] related_to
>> [[Target1]]
>> [!edge] derived_from
>> [[Target2]]"""
        
        pairs, remaining = extract_callout_relations(text)
        
        # Einfaches Callout
        self.assertIn(("depends_on", "Dependency1"), pairs)
        
        # Verschachtelte Callouts
        self.assertIn(("related_to", "Target1"), pairs)
        self.assertIn(("derived_from", "Target2"), pairs)

    def test_callout_with_explicit_format(self):
        """Test: Callout mit explizitem 'kind: targets' Format."""
        text = """> [!edge] related_to: [[Target1]] [[Target2]]"""
        
        pairs, remaining = extract_callout_relations(text)
        
        self.assertEqual(len(pairs), 2)
        self.assertIn(("related_to", "Target1"), pairs)
        self.assertIn(("related_to", "Target2"), pairs)

    def test_empty_callout(self):
        """Test: Leeres Callout ohne Targets."""
        text = """> [!edge] related_to"""
        
        pairs, remaining = extract_callout_relations(text)
        
        self.assertEqual(len(pairs), 0)

    def test_no_callouts(self):
        """Test: Text ohne Callouts bleibt unverändert."""
        text = """Normaler Text ohne Callouts.
[[Ein normaler Link]]"""
        
        pairs, remaining = extract_callout_relations(text)
        
        self.assertEqual(len(pairs), 0)
        self.assertEqual(remaining, text)

    def test_callout_with_section_links(self):
        """Test: Callout mit Section-Links (Deep-Links)."""
        text = """> [!edge] related_to
> [[Target#Section1]]
> [[Target#Section2]]"""
        
        pairs, remaining = extract_callout_relations(text)
        
        self.assertEqual(len(pairs), 2)
        self.assertIn(("related_to", "Target#Section1"), pairs)
        self.assertIn(("related_to", "Target#Section2"), pairs)

    def test_original_format_example(self):
        """Test: Das ursprünglich gefragte Format aus der Roadmap."""
        text = """> [!abstract]- 🕸️ Semantic Mapping
>> [!edge] related_to
>> [[Brigitte]]
>> [[Klaus]]
>> [[Salami Fertigpizza]]
>
>>  [!edge] derived_from
>> [[Link 2]]"""
        
        pairs, remaining = extract_callout_relations(text)
        
        # Prüfe, dass alle Edges erkannt wurden
        self.assertEqual(len(pairs), 4)
        
        # Prüfe related_to Edges
        related_pairs = [p for p in pairs if p[0] == "related_to"]
        self.assertEqual(len(related_pairs), 3)
        self.assertIn(("related_to", "Brigitte"), pairs)
        self.assertIn(("related_to", "Klaus"), pairs)
        self.assertIn(("related_to", "Salami Fertigpizza"), pairs)
        
        # Prüfe derived_from Edge
        derived_pairs = [p for p in pairs if p[0] == "derived_from"]
        self.assertEqual(len(derived_pairs), 1)
        self.assertIn(("derived_from", "Link 2"), pairs)


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