from dev_aci.linter import DefaultLinter, LintResult
from dev_aci.linter.impl.python import (
    PythonLinter,
    flake_lint,
    python_compile_lint,
)


def test_wrongly_indented_py_file(wrongly_indented_py_file):
    # Test Python linter
    linter = PythonLinter()
    assert ".py" in linter.supported_extensions
    result = linter.lint(wrongly_indented_py_file)
    print(result)
    assert isinstance(result, list) and len(result) == 1
    assert result[0] == LintResult(
        file=wrongly_indented_py_file,
        line=2,
        column=5,
        message="E999 IndentationError: unexpected indent",
    )
    print(result[0].visualize())
    assert result[0].visualize() == (
        "1|\n"
        "\033[91m2|    def foo():\033[0m\n"
        "      ^ ERROR HERE: E999 IndentationError: unexpected indent\n"
        '3|            print("Hello, World!")\n'
        "4|"
    )

    # General linter should have same result as Python linter
    # bc it uses PythonLinter under the hood
    general_linter = DefaultLinter()
    assert ".py" in general_linter.supported_extensions
    result = general_linter.lint(wrongly_indented_py_file)
    assert result == linter.lint(wrongly_indented_py_file)

    # Test flake8_lint
    assert result == flake_lint(wrongly_indented_py_file)

    # Test python_compile_lint
    compile_result = python_compile_lint(wrongly_indented_py_file)
    assert isinstance(compile_result, list) and len(compile_result) == 1
    assert compile_result[0] == LintResult(
        file=wrongly_indented_py_file, line=2, column=4, message="unexpected indent"
    )


def test_simple_correct_py_file(simple_correct_py_file):
    linter = PythonLinter()
    assert ".py" in linter.supported_extensions
    result = linter.lint(simple_correct_py_file)
    assert result == []

    general_linter = DefaultLinter()
    assert ".py" in general_linter.supported_extensions
    result = general_linter.lint(simple_correct_py_file)
    assert result == linter.lint(simple_correct_py_file)

    # Test python_compile_lint
    compile_result = python_compile_lint(simple_correct_py_file)
    assert compile_result == []

    # Test flake_lint
    flake_result = flake_lint(simple_correct_py_file)
    assert flake_result == []


def test_simple_correct_py_func_def(simple_correct_py_func_def):
    linter = PythonLinter()
    result = linter.lint(simple_correct_py_func_def)
    assert result == []

    general_linter = DefaultLinter()
    assert ".py" in general_linter.supported_extensions
    result = general_linter.lint(simple_correct_py_func_def)
    assert result == linter.lint(simple_correct_py_func_def)

    # Test flake_lint
    assert result == flake_lint(simple_correct_py_func_def)

    # Test python_compile_lint
    compile_result = python_compile_lint(simple_correct_py_func_def)
    assert compile_result == []