from openhands.linter import DefaultLinter, LintResult
from openhands.linter.languages.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 == []