fix: flatten array on lazy header coerce each

marcosgz · Aug 8, 2024 · 4092de5 · 4092de5
1 parent 0a9bf46
commit 4092de5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/lib/esse/lazy_document_header.rb b/lib/esse/lazy_document_header.rb
@@ -4,7 +4,7 @@ module Esse
   class LazyDocumentHeader
     def self.coerce_each(values)
       arr = []
-      Esse::ArrayUtils.wrap(values).map do |value|
+      Esse::ArrayUtils.wrap(values).flatten.map do |value|
         instance = coerce(value)
         arr << instance if instance&.valid?
       end

diff --git a/spec/esse/lazzy_document_header_spec.rb b/spec/esse/lazzy_document_header_spec.rb
@@ -169,6 +169,10 @@
     it 'removes invalid instances' do
       expect(described_class.coerce_each([nil, {_id: 1}, {}]).size).to eq(1)
     end
+
+    it 'flattens the array' do
+      expect(described_class.coerce_each([[{_id: 1}], {_id: 2}]).size).to eq(2)
+    end
   end
 
   describe '#to_doc' do