from pathlib import Path

from clint.config import Config
from clint.index import SymbolIndex
from clint.linter import Position, Range, lint_file
from clint.rules.no_class_based_tests import NoClassBasedTests


def test_no_class_based_tests(index: SymbolIndex) -> None:
    code = """import pytest

# Bad - class-based test with test methods
class TestSomething:
    def test_feature_a(self):
        assert True

    def test_feature_b(self):
        assert True

    def helper_method(self):
        return 42

# Bad - another class-based test
class TestAnotherThing:
    def test_something(self):
        pass

# Good - class without test methods (utility class)
class HelperClass:
    def helper_function(self):
        return 42

    def setup_something(self):
        pass

    def test_something(self):
        pass

# Good - function-based test
def test_valid_function():
    assert True

# Good - regular function
def helper_function():
    return 42
"""
    config = Config(select={NoClassBasedTests.name})
    violations = lint_file(Path("test_something.py"), code, config, index)
    assert len(violations) == 2
    assert all(isinstance(v.rule, NoClassBasedTests) for v in violations)
    assert violations[0].range == Range(Position(3, 0))  # TestSomething class
    assert violations[1].range == Range(Position(14, 0))  # TestAnotherThing class


def test_no_class_based_tests_non_test_file(index: SymbolIndex) -> None:
    code = """import pytest

# This should not be flagged because it's not in a test file
class TestSomething:
    def test_feature_a(self):
        assert True
"""
    config = Config(select={NoClassBasedTests.name})
    violations = lint_file(Path("regular_file.py"), code, config, index)
    assert len(violations) == 0