StyledRangeContainerTests.swift

import XCTest
@testable import CodeEditSourceEditor

final class StyledRangeContainerTests: XCTestCase {
    typealias Run = StyledRangeStoreRun

    @MainActor
    func test_init() {
        let providers = [0, 1]
        let store = StyledRangeContainer(documentLength: 100, providers: providers)

        // Have to do string conversion due to missing Comparable conformance pre-macOS 14
        XCTAssertEqual(store._storage.keys.sorted(), providers)
        XCTAssert(store._storage.values.allSatisfy({ $0.length == 100 }), "One or more providers have incorrect length")
    }

    @MainActor
    func test_setHighlights() {
        let providers = [0, 1]
        let store = StyledRangeContainer(documentLength: 100, providers: providers)

        store.applyHighlightResult(
            provider: providers[0],
            highlights: [HighlightRange(range: NSRange(location: 40, length: 10), capture: .comment)],
            rangeToHighlight: NSRange(location: 0, length: 100)
        )

        XCTAssertNotNil(store._storage[providers[0]])
        XCTAssertEqual(store._storage[providers[0]]!.count, 3)
        XCTAssertNil(store._storage[providers[0]]!.runs(in: 0..<100)[0].capture)
        XCTAssertEqual(store._storage[providers[0]]!.runs(in: 0..<100)[1].capture, .comment)
        XCTAssertNil(store._storage[providers[0]]!.runs(in: 0..<100)[2].capture)

        XCTAssertEqual(
            store.runsIn(range: NSRange(location: 0, length: 100)),
            [
                Run(length: 40, capture: nil, modifiers: []),
                Run(length: 10, capture: .comment, modifiers: []),
                Run(length: 50, capture: nil, modifiers: [])
            ]
        )
    }

    @MainActor
    func test_overlappingRuns() {
        let providers = [0, 1]
        let store = StyledRangeContainer(documentLength: 100, providers: providers)

        store.applyHighlightResult(
            provider: providers[0],
            highlights: [HighlightRange(range: NSRange(location: 40, length: 10), capture: .comment)],
            rangeToHighlight: NSRange(location: 0, length: 100)
        )

        store.applyHighlightResult(
            provider: providers[1],
            highlights: [
                HighlightRange(range: NSRange(location: 45, length: 5), capture: nil, modifiers: [.declaration])
            ],
            rangeToHighlight: NSRange(location: 0, length: 100)
        )

        XCTAssertEqual(
            store.runsIn(range: NSRange(location: 0, length: 100)),
            [
                Run(length: 40, capture: nil, modifiers: []),
                Run(length: 5, capture: .comment, modifiers: []),
                Run(length: 5, capture: .comment, modifiers: [.declaration]),
                Run(length: 50, capture: nil, modifiers: [])
            ]
        )
    }

    @MainActor
    func test_overlappingRunsWithMoreProviders() {
        let providers = [0, 1, 2]
        let store = StyledRangeContainer(documentLength: 200, providers: providers)

        store.applyHighlightResult(
            provider: providers[0],
            highlights: [
                HighlightRange(range: NSRange(location: 30, length: 20), capture: .comment),
                HighlightRange(range: NSRange(location: 80, length: 30), capture: .string)
            ],
            rangeToHighlight: NSRange(location: 0, length: 200)
        )

        store.applyHighlightResult(
            provider: providers[1],
            highlights: [
                HighlightRange(range: NSRange(location: 35, length: 10), capture: nil, modifiers: [.declaration]),
                HighlightRange(range: NSRange(location: 90, length: 15), capture: .comment, modifiers: [.static])
            ],
            rangeToHighlight: NSRange(location: 0, length: 200)
        )

        store.applyHighlightResult(
            provider: providers[2],
            highlights: [
                HighlightRange(range: NSRange(location: 40, length: 5), capture: .function, modifiers: [.abstract]),
                HighlightRange(range: NSRange(location: 100, length: 10), capture: .number, modifiers: [.modification])
            ],
            rangeToHighlight: NSRange(location: 0, length: 200)
        )

        let runs = store.runsIn(range: NSRange(location: 0, length: 200))

        XCTAssertEqual(runs.reduce(0, { $0 + $1.length}), 200)

        XCTAssertEqual(runs[0], Run(length: 30, capture: nil, modifiers: []))
        XCTAssertEqual(runs[1], Run(length: 5, capture: .comment, modifiers: []))
        XCTAssertEqual(runs[2], Run(length: 5, capture: .comment, modifiers: [.declaration]))
        XCTAssertEqual(runs[3], Run(length: 5, capture: .comment, modifiers: [.abstract, .declaration]))
        XCTAssertEqual(runs[4], Run(length: 5, capture: .comment, modifiers: []))
        XCTAssertEqual(runs[5], Run(length: 30, capture: nil, modifiers: []))
        XCTAssertEqual(runs[6], Run(length: 10, capture: .string, modifiers: []))
        XCTAssertEqual(runs[7], Run(length: 10, capture: .string, modifiers: [.static]))
        XCTAssertEqual(runs[8], Run(length: 5, capture: .string, modifiers: [.static, .modification]))
        XCTAssertEqual(runs[9], Run(length: 5, capture: .string, modifiers: [.modification]))
        XCTAssertEqual(runs[10], Run(length: 90, capture: nil, modifiers: []))
    }
}