PureXML/Tests/AttributeDocumentOrderTests.swift
Mihaela Mihaljevic d86e4effbe
fix(xpath): cache the attribute index for document-order keys (#305)
An attribute node's document-order key bands by the attribute's index among its
owner's attributes, found with owner.attributes.firstIndex { $0.name == name }.
That linear scan ran for every attribute node, so sorting an element's K
attributes into document order was O(K^2): an 8000-attribute element took ~0.18s
and grew quadratically.

Cache a per-owner attribute-name-to-index map in SiblingIndexCache, built once
on the second lookup (a single lookup still scans), exactly as the sibling-index
cache already does for child positions. The index is the same first-occurrence
position; only its cost changes.

The 8000-attribute sort drops to ~0.01s (~16x) and is now linear. The full
suite, the W3C XPath and Apache Xalan corpora, the C14N suite (which orders
attributes byte-exactly), and the WASM build stay green; AttributeDocumentOrderTests
pins declaration order through the cached sort.
2026-06-25 22:01:13 +02:00

40 lines
1.7 KiB
Swift

import Testing
@testable import PureXML
/// An attribute node's document-order key bands by its index among its owner's
/// attributes. That index now comes from a cached per-owner name-to-index map
/// (built once on the second lookup) rather than a linear scan per attribute,
/// so sorting many attribute nodes from one element is linear, not quadratic.
/// These pin that the order is unchanged: declaration order, for attributes
/// handed to the sort out of order, including a many-attribute element.
@Suite("attribute document order")
struct AttributeDocumentOrderTests {
private func values(_ path: String, _ xml: String) throws -> [String] {
try PureXML.XPath.Query(path).evaluate(over: PureXML.parse(xml)).compactMap {
if case let .attribute(attribute) = $0 { return attribute.value }
return nil
}
}
@Test("a union of attributes comes back in declaration order")
func test_unionOrder() throws {
let xml = #"<e a="1" b="2" c="3" d="4"/>"#
// The union lists them out of declaration order; the result is the
// deduplicated node-set in document (declaration) order.
#expect(try values("//e/@c | //e/@a | //e/@d | //e/@b", xml) == ["1", "2", "3", "4"])
}
@Test("attributes of a many-attribute element sort in declaration order")
func test_manyAttributes() throws {
// The quadratic-prone case: one element with many attributes, all sorted.
var attrs = ""
for index in 0 ..< 200 {
attrs += " a\(index)=\"\(index)\""
}
let xml = "<e\(attrs)/>"
let ordered = try values("//e/@* | //e/@*", xml)
#expect(ordered.count == 200)
#expect(ordered == (0 ..< 200).map(String.init))
}
}