import Testing
@testable import PureXML

/// Unique Particle Attribution (cos-nonambig, XSD 1.0 3.8.6): a content model
/// must be deterministic, so each element in an instance is attributable to a
/// single particle without lookahead. The check runs on a literal, QName-resolved
/// view of the raw schema tree (refs are not substitution-group expanded, the
/// overlap test is QName-only), and is per particle: a particle competing with a
/// repetition of itself is never a violation. These cases pin both the rejections
/// and the deterministic shapes that must keep compiling (XSTS valid mgZ005,
/// ctZ008, particlesHa142, groupH001).
@Suite("Content model determinism (UPA)")
struct ContentModelDeterminismTests {
    private func compile(_ body: String) throws {
        _ = try PureXML.Schema.Document("""
        <xs:schema xmlns:xs="http://www.w3.org/2001/XMLSchema">
        \(body)
        </xs:schema>
        """)
    }

    private func rejects(_ body: String) -> Bool {
        do { try compile(body)
            return false
        } catch { return true }
    }

    private func complexType(_ model: String) -> String {
        #"<xs:complexType name="t">\#(model)</xs:complexType>"#
    }

    // MARK: - Ambiguous models are rejected

    @Test("a choice of two same-name elements is ambiguous")
    func test_choiceOfSameName() {
        #expect(rejects(complexType(#"<xs:choice><xs:element name="a"/><xs:element name="a"/></xs:choice>"#)))
    }

    @Test("an optional element followed by a same-name element is ambiguous")
    func test_optionalThenSameName() {
        #expect(rejects(complexType(#"<xs:sequence><xs:element name="a" minOccurs="0"/><xs:element name="a"/></xs:sequence>"#)))
    }

    @Test("a repeating element followed by a same-name element is ambiguous")
    func test_repeatThenSameName() {
        #expect(rejects(complexType(#"<xs:sequence><xs:element name="a" maxOccurs="unbounded"/><xs:element name="a"/></xs:sequence>"#)))
    }

    @Test("two same-name members of an all group are ambiguous")
    func test_allWithSameName() {
        #expect(rejects(complexType(#"<xs:all><xs:element name="a"/><xs:element name="a"/></xs:all>"#)))
    }

    @Test("an element competing with a wildcard that admits it is ambiguous")
    func test_elementAndWildcard() {
        #expect(rejects(complexType(#"<xs:choice><xs:element name="a"/><xs:any processContents="skip"/></xs:choice>"#)))
    }

    // MARK: - Deterministic models keep compiling

    @Test("distinct elements in sequence and choice are deterministic")
    func test_distinctNames() throws {
        try compile(complexType(#"<xs:sequence><xs:element name="a"/><xs:element name="b" minOccurs="0"/></xs:sequence>"#))
        try compile(complexType(#"<xs:choice><xs:element name="a"/><xs:element name="b"/></xs:choice>"#))
        try compile(complexType(#"<xs:all><xs:element name="a"/><xs:element name="b"/><xs:element name="c"/></xs:all>"#))
    }

    @Test("a fixed-count element followed by a distinct same-name element is deterministic (mgZ005)")
    func test_fixedCountThenSameName() throws {
        try compile(complexType(#"""
        <xs:sequence>
        <xs:element name="a" minOccurs="0"/>
        <xs:element name="b" minOccurs="2" maxOccurs="2"/>
        <xs:element name="b"/>
        </xs:sequence>
        """#))
    }

    @Test("a fixed-count group wrapping a repeating element is deterministic (ctZ008)")
    func test_fixedCountGroupWithRepeat() throws {
        try compile(complexType(#"""
        <xs:sequence minOccurs="2" maxOccurs="2">
        <xs:element name="a" minOccurs="1" maxOccurs="2"/>
        <xs:element name="b" minOccurs="0"/>
        </xs:sequence>
        """#))
    }

    @Test("an element followed by its own optional repeat is deterministic (particlesHa142)")
    func test_elementThenOptionalSameName() throws {
        try compile(complexType(#"""
        <xs:choice>
        <xs:sequence minOccurs="0">
        <xs:element name="a" type="xs:string"/>
        <xs:element name="a" type="xs:string" minOccurs="0"/>
        </xs:sequence>
        </xs:choice>
        """#))
    }

    @Test("a bounded-repeat choice of group references is deterministic (groupH001)")
    func test_repeatedChoiceOfGroups() throws {
        try compile(#"""
        <xs:complexType name="A">
        <xs:choice minOccurs="0" maxOccurs="4">
        <xs:group ref="x"/>
        <xs:group ref="y"/>
        </xs:choice>
        </xs:complexType>
        <xs:group name="x"><xs:sequence><xs:element name="x1"/><xs:element name="x2"/></xs:sequence></xs:group>
        <xs:group name="y"><xs:choice><xs:element name="y1"/><xs:element name="y2"/></xs:choice></xs:group>
        """#)
    }
}