from pathlib import Path

import pytest
from clint.config import Config
from clint.index import SymbolIndex
from clint.linter import Position, Range, lint_file
from clint.rules.unknown_mlflow_function import UnknownMlflowFunction


def test_unknown_mlflow_function(index: SymbolIndex) -> None:
    code = '''
def bad():
    """
    .. code-block:: python

        import mlflow

        mlflow.foo()

    """


def good():

    """
    .. code-block:: python

        import mlflow

        mlflow.log_param("k", "v")

    """
'''
    config = Config(select={UnknownMlflowFunction.name}, example_rules=[UnknownMlflowFunction.name])
    violations = lint_file(Path("test.py"), code, config, index)
    assert len(violations) == 1
    assert all(isinstance(v.rule, UnknownMlflowFunction) for v in violations)
    assert violations[0].range == Range(Position(7, 8))


@pytest.mark.parametrize("suffix", [".md", ".mdx"])
def test_unknown_mlflow_function_markdown(index: SymbolIndex, suffix: str) -> None:
    code = """
# Bad

```python
import mlflow

mlflow.foo()
```

# Good

```python
import mlflow

mlflow.log_param("k", "v")
```

"""
    config = Config(
        select={UnknownMlflowFunction.name},
        example_rules=[UnknownMlflowFunction.name],
    )
    violations = lint_file(Path("test").with_suffix(suffix), code, config, index)
    assert len(violations) == 1
    assert all(isinstance(v.rule, UnknownMlflowFunction) for v in violations)
    assert violations[0].range == Range(Position(6, 0))