@testable import SwiftLintBuiltInRules
import TestHelpers
import XCTest

final class TypeBodyLengthRuleTests: SwiftLintTestCase {
    func testWarning() {
        let example = Example("""
            actor A {
                let x = 0
                let y = 1
                let z = 2
            }
            """)

        XCTAssertEqual(
            self.violations(example, configuration: ["warning": 2, "error": 4]),
            [
                StyleViolation(
                    ruleDescription: TypeBodyLengthRule.description,
                    severity: .warning,
                    location: Location(file: nil, line: 1, character: 1),
                    reason: """
                        Actor body should span 2 lines or less excluding comments and \
                        whitespace: currently spans 3 lines
                        """
                ),
            ]
        )
    }

    func testError() {
        let example = Example("""
            class C {
                let x = 0
                let y = 1
                let z = 2
            }
            """)

        XCTAssertEqual(
            self.violations(example, configuration: ["warning": 1, "error": 2]),
            [
                StyleViolation(
                    ruleDescription: TypeBodyLengthRule.description,
                    severity: .error,
                    location: Location(file: nil, line: 1, character: 1),
                    reason: """
                        Class body should span 2 lines or less excluding comments and \
                        whitespace: currently spans 3 lines
                        """
                ),
            ]
        )
    }

    func testViolationMessages() {
        let types = TypeBodyLengthRule.description.triggeringExamples.flatMap {
            self.violations($0, configuration: ["warning": 2])
        }.compactMap {
            $0.reason.split(separator: " ", maxSplits: 1).first
        }

        XCTAssertEqual(
            types,
            ["Actor", "Class", "Enum", "Extension", "Protocol", "Struct"]
        )
    }

    private func violations(_ example: Example, configuration: Any? = nil) -> [StyleViolation] {
        let config = makeConfig(configuration, TypeBodyLengthRule.identifier)!
        return TestHelpers.violations(example, config: config)
    }
}