# retoor <retoor@molodetz.nl>

import unittest.mock

from devplacepy.services.correction import correct_text


def test_correct_text_extracts_and_restores_code_blocks():
    captured = {}

    def fake_gateway(api_key, system, text, timeout, max_growth_factor):
        captured["system"] = system
        captured["text"] = text
        return ("correction result", {"calls": 1})

    with unittest.mock.patch(
        "devplacepy.services.correction.gateway_complete", fake_gateway
    ):
        result, _ = correct_text(
            "test-key", "Fix spelling", "Some text\n```python\nx = 1\n```\nMore text"
        )

    assert result == "correction result"


def test_correct_text_preserves_fenced_code_blocks_from_ai_corruption():
    input_text = "Some text\n```python\nx = 1\n```\nMore text"

    def fake_gateway(api_key, system, text, timeout, max_growth_factor):
        return ("Some text\n{%CODE_BLOCK_0%}\nMore corrected text", {"calls": 1})

    with unittest.mock.patch(
        "devplacepy.services.correction.gateway_complete", fake_gateway
    ):
        result, _ = correct_text("test-key", "Fix spelling", input_text)

    assert "```python" in result
    assert "x = 1" in result
    assert "corrected text" in result


def test_correct_text_preserves_multiple_fenced_code_blocks():
    input_text = "A\n```python\nx = 1\n```\nB\n```js\ny = 2\n```\nC"

    def fake_gateway(api_key, system, text, timeout, max_growth_factor):
        return ("A\n{%CODE_BLOCK_0%}\nX\n{%CODE_BLOCK_1%}\nZ", {"calls": 1})

    with unittest.mock.patch(
        "devplacepy.services.correction.gateway_complete", fake_gateway
    ):
        result, _ = correct_text("test-key", "Fix spelling", input_text)

    assert "```python" in result
    assert "x = 1" in result
    assert "```js" in result
    assert "y = 2" in result


def test_correct_text_preserves_inline_code():
    input_text = "Use the `os.path.join` function for paths."

    def fake_gateway(api_key, system, text, timeout, max_growth_factor):
        return ("Use {%CODE_BLOCK_0%} always.", {"calls": 1})

    with unittest.mock.patch(
        "devplacepy.services.correction.gateway_complete", fake_gateway
    ):
        result, _ = correct_text("test-key", "Fix spelling", input_text)

    assert "`os.path.join`" in result


def test_correct_text_passes_plain_text_untouched():
    captured = {}

    def fake_gateway(api_key, system, text, timeout, max_growth_factor):
        captured["text"] = text
        return ("corrected plain text", {"calls": 1})

    with unittest.mock.patch(
        "devplacepy.services.correction.gateway_complete", fake_gateway
    ):
        result, _ = correct_text(
            "test-key", "Fix spelling", "Just some plain text."
        )

    assert captured["text"] == "Just some plain text."
    assert result == "corrected plain text"


def test_correct_text_sanitized_text_has_no_code_blocks():
    captured = {}

    def fake_gateway(api_key, system, text, timeout, max_growth_factor):
        captured["text"] = text
        return ("corrected", {"calls": 1})

    with unittest.mock.patch(
        "devplacepy.services.correction.gateway_complete", fake_gateway
    ):
        correct_text(
            "test-key",
            "Fix spelling",
            "Before\n```python\ncode\n```\nAfter\n`inline`",
        )

    assert "```" not in captured["text"]
    assert "`inline`" not in captured["text"]