lutaml-model 0.6.7 → 0.7.1

data/spec/lutaml/model/polymorphic_spec.rb

@@ -0,0 +1,348 @@
+require "spec_helper"
+
+module PolymorphicSpec
+  module Base
+    class Reference < Lutaml::Model::Serializable
+      attribute :_class, :string, polymorphic_class: true
+      attribute :name, :string
+
+      xml do
+        map_attribute "reference-type", to: :_class, polymorphic_map: {
+          "document-ref" => "PolymorphicSpec::Base::DocumentReference",
+          "anchor-ref" => "PolymorphicSpec::Base::AnchorReference",
+        }
+        map_element "name", to: :name
+      end
+
+      key_value do
+        map "_class", to: :_class, polymorphic_map: {
+          "Document" => "PolymorphicSpec::Base::DocumentReference",
+          "Anchor" => "PolymorphicSpec::Base::AnchorReference",
+        }
+        map "name", to: :name
+      end
+    end
+
+    class DocumentReference < Reference
+      attribute :document_id, :string
+
+      xml do
+        map_element "document_id", to: :document_id
+      end
+
+      key_value do
+        map "document_id", to: :document_id
+      end
+    end
+
+    class AnchorReference < Reference
+      attribute :anchor_id, :string
+
+      xml do
+        map_element "anchor_id", to: :anchor_id
+      end
+
+      key_value do
+        map "anchor_id", to: :anchor_id
+      end
+    end
+
+    class ReferenceSet < Lutaml::Model::Serializable
+      attribute :references, Reference, collection: true, polymorphic: [
+        DocumentReference,
+        AnchorReference,
+      ]
+    end
+  end
+
+  module Child
+    # Case: If we have no access to the base class and we need to
+    # define polymorphism in the sub-classes.
+    class Reference < Lutaml::Model::Serializable
+      attribute :name, :string
+
+      xml do
+        map_element "name", to: :name
+      end
+
+      key_value do
+        map "name", to: :name
+      end
+    end
+
+    class DocumentReference < Reference
+      attribute :_class, :string
+      attribute :document_id, :string
+
+      xml do
+        map_element "document_id", to: :document_id
+        map_attribute "_class", to: :_class
+      end
+
+      key_value do
+        map "document_id", to: :document_id
+        map "_class", to: :_class
+      end
+    end
+
+    class AnchorReference < Reference
+      attribute :_class, :string
+      attribute :anchor_id, :string
+
+      xml do
+        map_element "anchor_id", to: :anchor_id
+        map_attribute "_class", to: :_class
+      end
+
+      key_value do
+        map "anchor_id", to: :anchor_id
+        map "_class", to: :_class
+      end
+    end
+
+    class ReferenceSet < Lutaml::Model::Serializable
+      attribute :references, Reference, collection: true, polymorphic: [
+        DocumentReference,
+        AnchorReference,
+      ]
+
+      xml do
+        root "ReferenceSet"
+
+        map_element "references", to: :references, polymorphic: {
+          # This refers to the attribute in the polymorphic model, you need
+          # to specify the attribute name (which is specified in the sub-classed model).
+          attribute: "_class",
+          class_map: {
+            "document-ref" => "PolymorphicSpec::Child::DocumentReference",
+            "anchor-ref" => "PolymorphicSpec::Child::AnchorReference",
+          },
+        }
+      end
+
+      key_value do
+        map "references", to: :references, polymorphic: {
+          attribute: "_class",
+          class_map: {
+            "Document" => "PolymorphicSpec::Child::DocumentReference",
+            "Anchor" => "PolymorphicSpec::Child::AnchorReference",
+          },
+        }
+      end
+    end
+  end
+end
+
+RSpec.describe "Polymorphic" do
+  context "when defined in base class" do
+    context "when key_value formats" do
+      let(:reference_set) do
+        PolymorphicSpec::Base::ReferenceSet.new(
+          references: [
+            PolymorphicSpec::Base::DocumentReference.new(
+              _class: "Document",
+              document_id: "book:tbtd",
+              name: "The Tibetan Book of the Dead",
+            ),
+            PolymorphicSpec::Base::AnchorReference.new(
+              _class: "Anchor",
+              anchor_id: "book:tbtd:anchor-1",
+              name: "Chapter 1",
+            ),
+          ],
+        )
+      end
+
+      let(:yaml) do
+        <<~YAML
+          ---
+          references:
+          - _class: Document
+            name: The Tibetan Book of the Dead
+            document_id: book:tbtd
+          - _class: Anchor
+            name: Chapter 1
+            anchor_id: book:tbtd:anchor-1
+        YAML
+      end
+
+      let(:parsed_yaml) do
+        PolymorphicSpec::Base::ReferenceSet.from_yaml(yaml)
+      end
+
+      it "deserializes correctly" do
+        expect(parsed_yaml).to eq(reference_set)
+      end
+
+      it "serializes correctly" do
+        expect(parsed_yaml.to_yaml).to eq(yaml)
+      end
+    end
+
+    context "when XML format" do
+      let(:reference_set) do
+        PolymorphicSpec::Base::ReferenceSet.new(
+          references: [
+            PolymorphicSpec::Base::DocumentReference.new(
+              _class: "document-ref",
+              document_id: "book:tbtd",
+              name: "The Tibetan Book of the Dead",
+            ),
+            PolymorphicSpec::Base::AnchorReference.new(
+              _class: "anchor-ref",
+              anchor_id: "book:tbtd:anchor-1",
+              name: "Chapter 1",
+            ),
+          ],
+        )
+      end
+
+      let(:xml) do
+        <<~XML
+          <ReferenceSet>
+            <references reference-type="document-ref">
+              <name>The Tibetan Book of the Dead</name>
+              <document_id>book:tbtd</document_id>
+            </references>
+            <references reference-type="anchor-ref">
+              <name>Chapter 1</name>
+              <anchor_id>book:tbtd:anchor-1</anchor_id>
+            </references>
+          </ReferenceSet>
+        XML
+      end
+
+      let(:parsed_xml) do
+        PolymorphicSpec::Base::ReferenceSet.from_xml(xml)
+      end
+
+      it "deserializes correctly" do
+        expect(parsed_xml).to eq(reference_set)
+      end
+
+      it "serializes correctly" do
+        expect(parsed_xml.to_xml.strip).to be_equivalent_to(xml.strip)
+      end
+    end
+  end
+
+  context "when defined in child class" do
+    context "when key_value formats" do
+      let(:reference_set) do
+        PolymorphicSpec::Base::ReferenceSet.new(
+          references: [
+            PolymorphicSpec::Base::DocumentReference.new(
+              _class: "Document",
+              document_id: "book:tbtd",
+              name: "The Tibetan Book of the Dead",
+            ),
+            PolymorphicSpec::Base::AnchorReference.new(
+              _class: "Anchor",
+              anchor_id: "book:tbtd:anchor-1",
+              name: "Chapter 1",
+            ),
+          ],
+        )
+      end
+
+      let(:yaml) do
+        <<~YAML
+          ---
+          references:
+          - _class: Document
+            name: The Tibetan Book of the Dead
+            document_id: book:tbtd
+          - _class: Anchor
+            name: Chapter 1
+            anchor_id: book:tbtd:anchor-1
+        YAML
+      end
+
+      let(:parsed_yaml) do
+        PolymorphicSpec::Base::ReferenceSet.from_yaml(yaml)
+      end
+
+      it "deserializes correctly" do
+        expect(parsed_yaml).to eq(reference_set)
+      end
+
+      it "serializes correctly" do
+        expect(parsed_yaml.to_yaml).to eq(yaml)
+      end
+    end
+
+    context "when XML format" do
+      let(:reference_set) do
+        PolymorphicSpec::Child::ReferenceSet.new(
+          references: [
+            PolymorphicSpec::Child::DocumentReference.new(
+              _class: "document-ref",
+              document_id: "book:tbtd",
+              name: "The Tibetan Book of the Dead",
+            ),
+            PolymorphicSpec::Child::AnchorReference.new(
+              _class: "anchor-ref",
+              anchor_id: "book:tbtd:anchor-1",
+              name: "Chapter 1",
+            ),
+          ],
+        )
+      end
+
+      let(:invalid_ref) do
+        PolymorphicSpec::Child::ReferenceSet.new(
+          references: [
+            PolymorphicSpec::Child::DocumentReference.new(
+              _class: "document-ref",
+              document_id: "book:tbtd",
+              name: "The Tibetan Book of the Dead",
+            ),
+            PolymorphicSpec::Base::AnchorReference.new(
+              _class: "anchor-ref",
+              anchor_id: "book:tbtd:anchor-1",
+              name: "Chapter 1",
+            ),
+          ],
+        )
+      end
+
+      let(:xml) do
+        <<~XML
+          <ReferenceSet>
+            <references _class="document-ref">
+              <name>The Tibetan Book of the Dead</name>
+              <document_id>book:tbtd</document_id>
+            </references>
+            <references _class="anchor-ref">
+              <name>Chapter 1</name>
+              <anchor_id>book:tbtd:anchor-1</anchor_id>
+            </references>
+          </ReferenceSet>
+        XML
+      end
+
+      let(:parsed_xml) do
+        PolymorphicSpec::Child::ReferenceSet.from_xml(xml)
+      end
+
+      it "deserializes correctly" do
+        expect(parsed_xml).to eq(reference_set)
+      end
+
+      it "serializes correctly" do
+        expect(parsed_xml.to_xml.strip).to be_equivalent_to(xml.strip)
+      end
+
+      it "does not raises error for valid polymorphic class" do
+        expect { reference_set.validate! }.not_to raise_error
+      end
+
+      it "raises error if polymorphic class is not in list" do
+        expect { invalid_ref.validate! }.to raise_error(Lutaml::Model::ValidationError) do |error|
+          expect(error).to include(Lutaml::Model::PolymorphicError)
+          expect(error.error_messages).to include("PolymorphicSpec::Base::AnchorReference not in [PolymorphicSpec::Child::DocumentReference, PolymorphicSpec::Child::AnchorReference]")
+        end
+      end
+    end
+  end
+end

data/spec/lutaml/model/render_empty_spec.rb

@@ -0,0 +1,194 @@
+module RenderEmptySpec
+  class DefaultModel < Lutaml::Model::Serializable
+    attribute :items, :string, collection: true
+
+    xml do
+      root "default-model"
+      map_element "items", to: :items
+    end
+  end
+
+  class EmptyInitModel < Lutaml::Model::Serializable
+    attribute :items, :string, collection: true, initialize_empty: true
+
+    xml do
+      root "empty-init-model"
+      map_element "items", to: :items
+    end
+  end
+
+  class OmitEmptyModel < Lutaml::Model::Serializable
+    attribute :items, :string, collection: true
+
+    xml do
+      root "omit-empty-model"
+      map_element "items", to: :items, render_empty: :omit
+    end
+
+    key_value do
+      map "items", to: :items, render_empty: :omit
+    end
+  end
+
+  class ExplicitEmptyModel < Lutaml::Model::Serializable
+    attribute :items, :string, collection: true
+
+    key_value do
+      map "items", to: :items, render_empty: :as_empty
+    end
+  end
+
+  class ExplicitBlankModel < Lutaml::Model::Serializable
+    attribute :items, :string, collection: true
+
+    xml do
+      map_element "items", to: :items, render_empty: :as_blank
+    end
+  end
+
+  class ExplicitNilModel < Lutaml::Model::Serializable
+    attribute :items, :string, collection: true
+
+    xml do
+      root "explicit-nil-model"
+      map_element "items", to: :items, render_empty: :as_nil
+    end
+
+    yaml do
+      map "items", to: :items, render_empty: :as_nil
+    end
+  end
+end
+
+RSpec.describe "RenderEmptySpec" do
+  describe "Collection States" do
+    context "with default behavior (initialize_empty: false)" do
+      it "defaults to nil" do
+        model = RenderEmptySpec::DefaultModel.new
+        expect(model.items).to be_nil
+        expect(model.to_xml).to eq("<default-model/>")
+      end
+    end
+
+    context "with initialize_empty: true" do
+      it "defaults to empty array" do
+        model = RenderEmptySpec::EmptyInitModel.new
+        expect(model.items).to eq([])
+        expect(model.to_xml).to eq("<empty-init-model/>")
+      end
+    end
+  end
+
+  describe "render_empty option" do
+    context "when :omit" do
+      let(:model) do
+        RenderEmptySpec::OmitEmptyModel.new(items: [])
+      end
+
+      it "omits empty collections in XML" do
+        expect(model.to_xml).not_to include("<items")
+      end
+
+      it "omits empty collections in YAML" do
+        expect(model.to_yaml.strip).to eq("--- {}")
+      end
+    end
+
+    context "when :as_empty" do
+      let(:model) do
+        RenderEmptySpec::ExplicitEmptyModel.new(items: [])
+      end
+
+      let(:parsed) do
+        RenderEmptySpec::ExplicitEmptyModel.from_yaml(yaml)
+      end
+
+      let(:yaml) do
+        <<~YAML
+          ---
+          items: []
+        YAML
+      end
+
+      it "renders explicit empty collection" do
+        expect(model.to_yaml).to eq(yaml)
+      end
+
+      it "sets empty values while deserialize" do
+        expect(parsed.items).to eq([])
+      end
+    end
+
+    context "when :as_blank" do
+      let(:model) do
+        RenderEmptySpec::ExplicitBlankModel.new(items: [])
+      end
+
+      let(:parsed) do
+        RenderEmptySpec::ExplicitBlankModel.from_xml(xml)
+      end
+
+      let(:xml) do
+        <<~XML
+          <explicit-blank-model>
+            <items/>
+          </explicit-blank-model>
+        XML
+      end
+
+      it "creates blank element from empty collections" do
+        expect(model.to_xml).to include("<items/>")
+      end
+
+      it "sets blank values while deserialize" do
+        expect(parsed.items).to eq([])
+      end
+    end
+
+    context "when :as_nil" do
+      let(:model) do
+        RenderEmptySpec::ExplicitNilModel.new(items: [])
+      end
+
+      describe "XML" do
+        let(:xml) do
+          <<~XML
+            <explicit-nil-model>
+              <items xsi:nil="true"/>
+            </explicit-nil-model>
+          XML
+        end
+
+        let(:parsed) do
+          RenderEmptySpec::ExplicitNilModel.from_xml(xml)
+        end
+
+        it "renders explicit nil" do
+          expect(model.to_xml).to include('<items xsi:nil="true"/>')
+        end
+
+        it "sets nil values while deserializing" do
+          expect(parsed.items).to eq([])
+        end
+      end
+
+      describe "YAML" do
+        let(:yaml) do
+          <<~YAML
+            ---
+            items: []
+          YAML
+        end
+
+        it "renders explicit nil in YAML" do
+          expect(model.to_yaml).to include("items:")
+        end
+
+        it "sets nil values while deserializing YAML" do
+          model = RenderEmptySpec::ExplicitNilModel.from_yaml(yaml)
+          expect(model.items).to eq([])
+        end
+      end
+    end
+  end
+end