spec/spec_helper.rb in schematic-0.1.9 vs spec/spec_helper.rb in schematic-0.2.1

@@ -37,11 +37,11 @@
 
 def sanitize_xml(xml)
   xml.split("\n").reject(&:blank?).map(&:strip).join("\n")
 end
 
-def generate_xsd_for_model(model)
+def generate_xsd_for_model(model, header_element = nil)
   output = <<-XML
 <?xml version="1.0" encoding="UTF-8"?>
 <xs:schema xmlns:xs="http://www.w3.org/2001/XMLSchema">
   <xs:complexType name="Integer">
   <xs:simpleContent>
@@ -90,10 +90,12 @@
   <xs:extension base="xs:boolean">
   <xs:attribute name="type" type="xs:string" use="optional"/>
   </xs:extension>
   </xs:simpleContent>
   </xs:complexType>
-  <xs:element name="#{model.xsd_generator.names.element_collection}" type="#{model.xsd_generator.names.collection_type}"/>
+  <xs:element name="#{model.xsd_generator.names.element_collection}" type="#{model.xsd_generator.names.collection_type}">
+  #{header_element}
+  </xs:element>
   <xs:complexType name="#{model.xsd_generator.names.collection_type}">
     <xs:sequence>
       <xs:element name="#{model.xsd_generator.names.element}" type="#{model.xsd_generator.names.type}" minOccurs="0" maxOccurs="unbounded"/>
     </xs:sequence>
     <xs:attribute name="type" type="xs:string" fixed="array"/>