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

import pytest

from src.app_helpers.quant_defaults import default_quant_method


@pytest.mark.parametrize('size, expected', [
    (7, 'Tuning'), (100, 'Tuning'), (100.01, 'RTN'), (235, 'RTN'), (None, 'Tuning'),
])
def test_parameter_threshold(size, expected):
    with patch('src.app_helpers.quant_defaults.get_model_size_b', return_value=size):
        assert default_quant_method('org/model')['value'].startswith(expected)


def test_empty_model_resets_without_hub_request():
    with patch('src.app_helpers.quant_defaults.get_model_size_b') as lookup:
        assert default_quant_method('')['value'].startswith('Tuning')
        lookup.assert_not_called()


def test_metadata_error_keeps_form_usable():
    with patch('src.app_helpers.quant_defaults.get_model_size_b', side_effect=ValueError):
        assert default_quant_method('org/model')['value'].startswith('Tuning')


def test_default_handler_does_not_shift_existing_gradio_events():
    source = (Path(__file__).resolve().parents[1] / 'app.py').read_text()
    tree = ast.parse(source)
    default_handlers = [
        node for node in ast.walk(tree)
        if isinstance(node, ast.Call)
        and isinstance(node.func, ast.Attribute)
        and any(
            keyword.arg == 'fn'
            and isinstance(keyword.value, ast.Name)
            and keyword.value.id == 'default_quant_method'
            for keyword in node.keywords
        )
    ]
    assert len(default_handlers) == 1
    handler = default_handlers[0]
    assert handler.func.attr == 'then'
    assert isinstance(handler.func.value, ast.Name)
    assert handler.func.value.id == '_quant_license_evt'
    timer_ticks = [
        node for node in ast.walk(tree)
        if isinstance(node, ast.Call)
        and isinstance(node.func, ast.Attribute)
        and node.func.attr == 'tick'
        and isinstance(node.func.value, ast.Name)
        and node.func.value.id in {'_queue_timer', '_pipeline_timer'}
    ]
    assert len(timer_ticks) == 2
    assert handler.lineno > max(tick.lineno for tick in timer_ticks)