test/test_document_schema.rb in populate-me-0.16.0 vs test/test_document_schema.rb in populate-me-0.17.0

@@ -4,9 +4,29 @@
 
 describe PopulateMe::Document, 'Schema' do
 
   parallelize_me!
 
+  describe "Batch field" do
+
+    class BatchFieldSet < PopulateMe::Document
+      field :name
+      field :image
+      batch_on_field :image
+    end
+
+    class NotBatchFieldSet < PopulateMe::Document
+      field :name
+      field :image
+    end
+
+    it 'Sets batch field correctly' do
+      assert_equal :image, BatchFieldSet.batch_field
+      assert_nil NotBatchFieldSet.batch_field
+    end
+
+  end
+
   describe "Relationships" do
 
     class Relative < PopulateMe::Document
       relationship :siblings
       relationship :home_remedies