mongoid 7.4.0 → 8.0.3

data/spec/mongoid/contextual/memory_spec.rb

@@ -379,36 +379,77 @@ describe Mongoid::Contextual::Memory do
 
   describe "#distinct" do
 
-    let(:hobrecht) do
-      Address.new(street: "hobrecht")
-    end
+    context "when legacy_pluck_distinct is true" do
+      config_override :legacy_pluck_distinct, true
 
-    let(:friedel) do
-      Address.new(street: "friedel")
-    end
+      let(:hobrecht) do
+        Address.new(street: "hobrecht")
+      end
+
+      let(:friedel) do
+        Address.new(street: "friedel")
+      end
 
-    context "when limiting the result set" do
+      context "when limiting the result set" do
 
-      let(:criteria) do
-        Address.where(street: "hobrecht").tap do |crit|
-          crit.documents = [ hobrecht, hobrecht, friedel ]
+        let(:criteria) do
+          Address.where(street: "hobrecht").tap do |crit|
+            crit.documents = [ hobrecht, hobrecht, friedel ]
+          end
+        end
+
+        let(:context) do
+          described_class.new(criteria)
+        end
+
+        it "returns the distinct field values" do
+          expect(context.distinct(:street)).to eq([ "hobrecht" ])
         end
       end
 
-      let(:context) do
-        described_class.new(criteria)
+      context "when not limiting the result set" do
+
+        let(:criteria) do
+          Address.all.tap do |crit|
+            crit.documents = [ hobrecht, friedel, friedel ]
+          end
+        end
+
+        let(:context) do
+          described_class.new(criteria)
+        end
+
+        it "returns the distinct field values" do
+          expect(context.distinct(:street)).to eq([ "hobrecht", "friedel" ])
+        end
       end
 
-      it "returns the distinct field values" do
-        expect(context.distinct(:street)).to eq([ "hobrecht" ])
+      context 'when there is a collation on the criteria' do
+
+        let(:criteria) do
+          Address.where(street: "hobrecht").tap do |crit|
+            crit.documents = [ hobrecht, hobrecht, friedel ]
+          end.collation(locale: 'en_US', strength: 2)
+        end
+
+        it "raises an exception" do
+          expect {
+            context.distinct(:street)
+          }.to raise_exception(Mongoid::Errors::InMemoryCollationNotSupported)
+        end
       end
     end
 
-    context "when not limiting the result set" do
+    context "when legacy_pluck_distinct is false" do
+      config_override :legacy_pluck_distinct, false
+
+      let(:depeche) { Band.create!(name: "Depeche Mode", years: 30, sales: "1E2") }
+      let(:new_order) { Band.create!(name: "New Order", years: 25, sales: "2E3") }
+      let(:maniacs) { Band.create!(name: "10,000 Maniacs", years: 20, sales: "1E2") }
 
       let(:criteria) do
-        Address.all.tap do |crit|
-          crit.documents = [ hobrecht, friedel, friedel ]
+        Band.all.tap do |crit|
+          crit.documents = [ depeche, new_order, maniacs ]
         end
       end
 
@@ -416,23 +457,180 @@ describe Mongoid::Contextual::Memory do
         described_class.new(criteria)
       end
 
-      it "returns the distinct field values" do
-        expect(context.distinct(:street)).to eq([ "hobrecht", "friedel" ])
+      context "when limiting the result set" do
+
+        let(:criteria) do
+          Band.where(name: "Depeche Mode").tap do |crit|
+            crit.documents = [ depeche ]
+          end
+        end
+
+        it "returns the distinct matching fields" do
+          expect(context.distinct(:name)).to eq([ "Depeche Mode" ])
+        end
       end
-    end
 
-    context 'when there is a collation on the criteria' do
+      context "when not limiting the result set" do
 
-      let(:criteria) do
-        Address.where(street: "hobrecht").tap do |crit|
-          crit.documents = [ hobrecht, hobrecht, friedel ]
-        end.collation(locale: 'en_US', strength: 2)
+        it "returns the distinct field values" do
+          expect(context.distinct(:name).sort).to eq([ "10,000 Maniacs", "Depeche Mode", "New Order" ].sort)
+        end
       end
 
-      it "raises an exception" do
-        expect {
-          context.distinct(:street)
-        }.to raise_exception(Mongoid::Errors::InMemoryCollationNotSupported)
+      context "when providing an aliased field" do
+
+        it "returns the distinct field values" do
+          expect(context.distinct(:years).sort).to eq([ 20, 25, 30 ])
+        end
+      end
+
+      context "when providing a demongoizable field" do
+
+        it "returns the non-demongoized distinct field values" do
+          expect(context.distinct(:sales).sort).to eq([ BigDecimal("1E2"), BigDecimal("2E3") ])
+        end
+      end
+
+      context "when getting a localized field" do
+        before do
+          I18n.locale = :en
+          d = Dictionary.create!(description: 'english-text')
+          I18n.locale = :de
+          d.description = 'deutsch-text'
+          d.save!
+        end
+
+        after do
+          I18n.locale = :en
+        end
+
+        let(:criteria) do
+          Dictionary.all.tap do |crit|
+            crit.documents = [ Dictionary.first ]
+          end
+        end
+
+        context "when getting the field without _translations" do
+          it "gets the demongoized localized field" do
+            expect(context.distinct(:description)).to eq([ 'deutsch-text' ])
+          end
+        end
+
+        context "when getting the field with _translations" do
+
+          it "gets the full hash" do
+            expect(context.distinct(:description_translations)).to eq([ { "de" => "deutsch-text", "en" => "english-text" } ])
+          end
+        end
+
+        context 'when plucking a specific locale' do
+
+          let(:distinct) do
+            context.distinct(:'description.de')
+          end
+
+          it 'returns the specific translation' do
+            expect(distinct).to eq([ "deutsch-text" ])
+          end
+        end
+
+        context 'when plucking a specific locale from _translations field' do
+
+          let(:distinct) do
+            context.distinct(:'description_translations.de')
+          end
+
+          it 'returns the specific translations' do
+            expect(distinct).to eq(['deutsch-text'])
+          end
+        end
+
+        context 'when fallbacks are enabled with a locale list' do
+          with_i18n_fallbacks
+
+          around(:all) do |example|
+            prev_fallbacks = I18n.fallbacks.dup
+            I18n.fallbacks[:he] = [ :en ]
+            example.run
+            I18n.fallbacks = prev_fallbacks
+          end
+
+          after do
+            I18n.locale = :en
+          end
+
+          let(:distinct) do
+            context.distinct(:description).first
+          end
+
+          it "correctly uses the fallback" do
+            I18n.locale = :en
+            Dictionary.create!(description: 'english-text')
+            I18n.locale = :he
+            distinct.should == "english-text"
+          end
+        end
+
+        context "when the localized field is embedded" do
+          let(:person) do
+            p = Passport.new
+            I18n.locale = :en
+            p.name = "Neil"
+            I18n.locale = :he
+            p.name = "Nissim"
+
+            Person.create!(passport: p, employer_id: 12345)
+          end
+
+          after do
+            I18n.locale = :en
+          end
+
+          let(:criteria) do
+            Person.where(employer_id: 12345).tap do |crit|
+              crit.documents = [ person ]
+            end
+          end
+
+          let(:distinct) do
+            context.distinct("pass.name").first
+          end
+
+          let(:distinct_translations) do
+            context.distinct("pass.name_translations").first
+          end
+
+          let(:distinct_translations_field) do
+            context.distinct("pass.name_translations.en").first
+          end
+
+          it "returns the translation for the current locale" do
+            expect(distinct).to eq("Nissim")
+          end
+
+          it "returns the full _translation hash" do
+            expect(distinct_translations).to eq({ "en" => "Neil", "he" => "Nissim" })
+          end
+
+          it "returns the translation for the requested locale" do
+            expect(distinct_translations_field).to eq("Neil")
+          end
+        end
+      end
+
+      context "when getting an embedded field" do
+
+        let(:label) { Label.new(sales: "1E2") }
+        let!(:band) { Band.create!(label: label) }
+        let(:criteria) do
+          Band.where(_id: band.id).tap do |crit|
+            crit.documents = [ band ]
+          end
+        end
+
+        it "returns the distinct matching fields" do
+          expect(context.distinct("label.sales")).to eq([ BigDecimal("1E2") ])
+        end
       end
     end
   end
@@ -624,6 +822,14 @@ describe Mongoid::Contextual::Memory do
         expect(context.send(method)).to eq(hobrecht)
       end
 
+      it "returns a list when passing a limit" do
+        expect(context.send(method, 2)).to eq([ hobrecht, friedel ])
+      end
+
+      it "returns a list when passing 1" do
+        expect(context.send(method, 1)).to eq([ hobrecht ])
+      end
+
       context 'when there is a collation on the criteria' do
 
         let(:criteria) do
@@ -641,6 +847,108 @@ describe Mongoid::Contextual::Memory do
     end
   end
 
+  describe "#take" do
+
+    let(:hobrecht) do
+      Address.new(street: "hobrecht")
+    end
+
+    let(:friedel) do
+      Address.new(street: "friedel")
+    end
+
+    let(:criteria) do
+      Address.where(:street.in => [ "hobrecht", "friedel" ]).tap do |crit|
+        crit.documents = [ hobrecht, friedel ]
+      end
+    end
+
+    let(:context) do
+      described_class.new(criteria)
+    end
+
+    it "returns the first matching document" do
+      expect(context.take).to eq(hobrecht)
+    end
+
+    it "returns an array when passing a limit" do
+      expect(context.take(2)).to eq([ hobrecht, friedel ])
+    end
+
+    it "returns an array when passing a limit as 1" do
+      expect(context.take(1)).to eq([ hobrecht ])
+    end
+
+    context 'when there is a collation on the criteria' do
+
+      let(:criteria) do
+        Address.where(:street.in => [ "hobrecht", "friedel" ]).tap do |crit|
+          crit.documents = [ hobrecht, friedel ]
+        end.collation(locale: 'en_US', strength: 2)
+      end
+
+      it "raises an exception" do
+        expect {
+          context.take
+        }.to raise_exception(Mongoid::Errors::InMemoryCollationNotSupported)
+      end
+    end
+  end
+
+  describe "#take!" do
+
+    let(:hobrecht) do
+      Address.new(street: "hobrecht")
+    end
+
+    let(:friedel) do
+      Address.new(street: "friedel")
+    end
+
+    let(:criteria) do
+      Address.where(:street.in => [ "hobrecht", "friedel" ]).tap do |crit|
+        crit.documents = [ hobrecht, friedel ]
+      end
+    end
+
+    let(:context) do
+      described_class.new(criteria)
+    end
+
+    it "returns the first matching document" do
+      expect(context.take!).to eq(hobrecht)
+    end
+
+    context "when the criteria is empty" do
+      let(:criteria) do
+        Address.where(street: "bogus").tap do |crit|
+          crit.documents = []
+        end
+      end
+
+      it "raise an error" do
+        expect do
+          context.take!
+        end.to raise_error(Mongoid::Errors::DocumentNotFound, /Could not find a document of class Address./)
+      end
+    end
+
+    context 'when there is a collation on the criteria' do
+
+      let(:criteria) do
+        Address.where(:street.in => [ "hobrecht", "friedel" ]).tap do |crit|
+          crit.documents = [ hobrecht, friedel ]
+        end.collation(locale: 'en_US', strength: 2)
+      end
+
+      it "raises an exception" do
+        expect {
+          context.take
+        }.to raise_exception(Mongoid::Errors::InMemoryCollationNotSupported)
+      end
+    end
+  end
+
   describe "#initialize" do
 
     context "when the criteria has no options" do
@@ -766,6 +1074,14 @@ describe Mongoid::Contextual::Memory do
       expect(context.last).to eq(friedel)
     end
 
+    it "returns a list when a limit is passed" do
+      expect(context.last(2)).to eq([ hobrecht, friedel ])
+    end
+
+    it "returns a list when the limit is 1" do
+      expect(context.last(1)).to eq([ friedel ])
+    end
+
     context 'when there is a collation on the criteria' do
 
       let(:criteria) do
@@ -919,72 +1235,1023 @@ describe Mongoid::Contextual::Memory do
 
   describe "#pluck" do
 
-    let(:hobrecht) do
-      Address.new(street: "hobrecht")
-    end
-
-    let(:friedel) do
-      Address.new(street: "friedel")
-    end
-
-    let(:criteria) do
-      Address.all.tap do |crit|
-        crit.documents = [ hobrecht, friedel ]
-      end
-    end
-
     let(:context) do
       described_class.new(criteria)
     end
 
-    context "when plucking" do
+    context "when legacy_pluck_distinct is true" do
+      config_override :legacy_pluck_distinct, true
 
-      let!(:plucked) do
-        context.pluck(:street)
+      let(:hobrecht) do
+        Address.new(street: "hobrecht")
       end
 
-      it "returns the values" do
-        expect(plucked).to eq([ "hobrecht", "friedel" ])
+      let(:friedel) do
+        Address.new(street: "friedel")
       end
-    end
 
-    context "when plucking a field that doesnt exist" do
+      let(:criteria) do
+        Address.all.tap do |crit|
+          crit.documents = [ hobrecht, friedel ]
+        end
+      end
 
-      context "when pluck one field" do
+      context "when plucking" do
 
-        let(:plucked) do
-          context.pluck(:foo)
+        let!(:plucked) do
+          context.pluck(:street)
         end
 
-        it "returns a empty array" do
-          expect(plucked).to eq([])
+        it "returns the values" do
+          expect(plucked).to eq([ "hobrecht", "friedel" ])
         end
       end
 
-      context "when pluck multiple fields" do
+      context "when plucking a mix of empty and non-empty values" do
 
-        let(:plucked) do
-          context.pluck(:foo, :bar)
+        let(:empty_doc) do
+          Address.new(street: nil)
         end
 
-        it "returns a empty array" do
-          expect(plucked).to eq([[], []])
-        end
+        let(:criteria) do
+          Address.all.tap do |crit|
+            crit.documents = [ hobrecht, friedel, empty_doc ]
+          end
+        end
+
+        let!(:plucked) do
+          context.pluck(:street)
+        end
+
+        it "returns the values" do
+          expect(plucked).to eq([ "hobrecht", "friedel", nil ])
+        end
+      end
+
+      context "when plucking a field that doesnt exist" do
+
+        context "when pluck one field" do
+
+          let(:plucked) do
+            context.pluck(:foo)
+          end
+
+          it "returns a empty array" do
+            expect(plucked).to eq([nil, nil])
+          end
+        end
+
+        context "when pluck multiple fields" do
+
+          let(:plucked) do
+            context.pluck(:foo, :bar)
+          end
+
+          it "returns a empty array" do
+            expect(plucked).to eq([[nil, nil], [nil, nil]])
+          end
+        end
+      end
+
+      context 'when there is a collation on the criteria' do
+
+        let(:criteria) do
+          Address.all.tap do |crit|
+            crit.documents = [ hobrecht, friedel ]
+          end.collation(locale: 'en_US', strength: 2)
+        end
+
+        it "raises an exception" do
+          expect {
+            context.pluck(:foo, :bar)
+          }.to raise_exception(Mongoid::Errors::InMemoryCollationNotSupported)
+        end
       end
     end
 
-    context 'when there is a collation on the criteria' do
+    context "when legacy_pluck_distinct is false" do
+      config_override :legacy_pluck_distinct, false
+
+      let!(:depeche) do
+        Band.create!(name: "Depeche Mode", likes: 3)
+      end
+
+      let!(:tool) do
+        Band.create!(name: "Tool", likes: 3)
+      end
+
+      let!(:photek) do
+        Band.create!(name: "Photek", likes: 1)
+      end
+
+      let!(:maniacs) do
+        Band.create!(name: "10,000 Maniacs", likes: 1, sales: "1E2")
+      end
 
       let(:criteria) do
-        Address.all.tap do |crit|
-          crit.documents = [ hobrecht, friedel ]
-        end.collation(locale: 'en_US', strength: 2)
+        Band.all.tap do |crit|
+          crit.documents = [ depeche, tool, photek, maniacs ]
+        end
       end
 
-      it "raises an exception" do
-        expect {
-          context.pluck(:foo, :bar)
-        }.to raise_exception(Mongoid::Errors::InMemoryCollationNotSupported)
+      context "when the field is aliased" do
+
+        let!(:expensive) do
+          Product.create!(price: 100000)
+        end
+
+        let!(:cheap) do
+          Product.create!(price: 1)
+        end
+
+        let(:criteria) do
+          Product.all.tap do |crit|
+            crit.documents = [ expensive, cheap ]
+          end
+        end
+
+        context "when using alias_attribute" do
+
+          let(:plucked) do
+            context.pluck(:price)
+          end
+
+          it "uses the aliases" do
+            expect(plucked).to eq([ 100000, 1 ])
+          end
+        end
+      end
+
+      context "when the criteria matches" do
+
+        context "when there are no duplicate values" do
+
+          let!(:plucked) do
+            context.pluck(:name)
+          end
+
+          it "returns the values" do
+            expect(plucked).to contain_exactly("10,000 Maniacs", "Depeche Mode", "Tool", "Photek")
+          end
+
+          context "when subsequently executing the criteria without a pluck" do
+
+            it "does not limit the fields" do
+              expect(context.first.likes).to eq(3)
+            end
+          end
+
+          context 'when the field is a subdocument' do
+
+            context 'when a top-level field and a subdocument field are plucked' do
+              let(:criteria) do
+                Band.where(name: 'FKA Twigs').tap do |crit|
+                  crit.documents = [
+                    Band.create!(name: 'FKA Twigs'),
+                    Band.create!(name: 'FKA Twigs', records: [ Record.new(name: 'LP1') ])
+                  ]
+                end
+              end
+
+              let(:embedded_pluck) do
+                context.pluck(:name, 'records.name')
+              end
+
+              let(:expected) do
+                [
+                  ["FKA Twigs", []],
+                  ['FKA Twigs', ["LP1"]]
+                ]
+              end
+
+              it 'returns the list of top-level field and subdocument values' do
+                expect(embedded_pluck).to eq(expected)
+              end
+            end
+
+            context 'when only a subdocument field is plucked' do
+
+              let(:criteria) do
+                Band.where(name: 'FKA Twigs').tap do |crit|
+                  crit.documents = [
+                    Band.create!(name: 'FKA Twigs'),
+                    Band.create!(name: 'FKA Twigs', records: [ Record.new(name: 'LP1') ])
+                  ]
+                end
+              end
+
+              let(:embedded_pluck) do
+                context.pluck('records.name')
+              end
+
+              let(:expected) do
+                [
+                  [],
+                  ["LP1"]
+                ]
+              end
+
+              it 'returns the list of subdocument values' do
+                expect(embedded_pluck).to eq(expected)
+              end
+            end
+          end
+        end
+
+        context "when plucking multi-fields" do
+
+          let(:plucked) do
+            context.pluck(:name, :likes)
+          end
+
+          it "returns the values" do
+            expect(plucked).to contain_exactly(["10,000 Maniacs", 1], ["Depeche Mode", 3], ["Tool", 3], ["Photek", 1])
+          end
+        end
+
+        context "when there are duplicate values" do
+
+          let(:plucked) do
+            context.pluck(:likes)
+          end
+
+          it "returns the duplicates" do
+            expect(plucked).to contain_exactly(1, 3, 3, 1)
+          end
+        end
+      end
+
+      context "when plucking an aliased field" do
+
+        let(:plucked) do
+          context.pluck(:id)
+        end
+
+        it "returns the field values" do
+          expect(plucked).to eq([ depeche.id, tool.id, photek.id, maniacs.id ])
+        end
+      end
+
+      context "when plucking existent and non-existent fields" do
+
+        let(:plucked) do
+          context.pluck(:id, :fooz)
+        end
+
+        it "returns nil for the field that doesnt exist" do
+          expect(plucked).to eq([[depeche.id, nil], [tool.id, nil], [photek.id, nil], [maniacs.id, nil] ])
+        end
+      end
+
+      context "when plucking a field that doesnt exist" do
+
+        context "when pluck one field" do
+
+          let(:plucked) do
+            context.pluck(:foo)
+          end
+
+          it "returns an array with nil values" do
+            expect(plucked).to eq([nil, nil, nil, nil])
+          end
+        end
+
+        context "when pluck multiple fields" do
+
+          let(:plucked) do
+            context.pluck(:foo, :bar)
+          end
+
+          it "returns an array of arrays with nil values" do
+            expect(plucked).to eq([[nil, nil], [nil, nil], [nil, nil], [nil, nil]])
+          end
+        end
+      end
+
+      context 'when plucking a localized field' do
+
+        before do
+          I18n.locale = :en
+          d = Dictionary.create!(description: 'english-text')
+          I18n.locale = :de
+          d.description = 'deutsch-text'
+          d.save!
+        end
+
+        after do
+          I18n.locale = :en
+        end
+
+        let(:criteria) do
+          Dictionary.all.tap do |crit|
+            crit.documents = [ Dictionary.first ]
+          end
+        end
+
+        context 'when plucking the entire field' do
+
+          let(:plucked) do
+            context.pluck(:description)
+          end
+
+          let(:plucked_translations) do
+            context.pluck(:description_translations)
+          end
+
+          let(:plucked_translations_both) do
+            context.pluck(:description_translations, :description)
+          end
+
+          it 'returns the demongoized translations' do
+            expect(plucked.first).to eq('deutsch-text')
+          end
+
+          it 'returns the full translations hash to _translations' do
+            expect(plucked_translations.first).to eq({"de"=>"deutsch-text", "en"=>"english-text"})
+          end
+
+          it 'returns both' do
+            expect(plucked_translations_both.first).to eq([{"de"=>"deutsch-text", "en"=>"english-text"}, "deutsch-text"])
+          end
+        end
+
+        context 'when plucking a specific locale' do
+
+          let(:plucked) do
+            context.pluck(:'description.de')
+          end
+
+          it 'returns the specific translations' do
+            expect(plucked.first).to eq('deutsch-text')
+          end
+        end
+
+        context 'when plucking a specific locale from _translations field' do
+
+          let(:plucked) do
+            context.pluck(:'description_translations.de')
+          end
+
+          it 'returns the specific translations' do
+            expect(plucked.first).to eq('deutsch-text')
+          end
+        end
+
+        context 'when fallbacks are enabled with a locale list' do
+          with_i18n_fallbacks
+
+          around(:all) do |example|
+            prev_fallbacks = I18n.fallbacks.dup
+            I18n.fallbacks[:he] = [ :en ]
+            example.run
+            I18n.fallbacks = prev_fallbacks
+          end
+
+          after do
+            I18n.locale = :en
+          end
+
+          let(:plucked) do
+            context.pluck(:description).first
+          end
+
+          it "correctly uses the fallback" do
+            I18n.locale = :en
+            Dictionary.create!(description: 'english-text')
+            I18n.locale = :he
+            plucked.should == "english-text"
+          end
+        end
+
+        context "when the localized field is embedded" do
+          before do
+            p = Passport.new
+            I18n.locale = :en
+            p.name = "Neil"
+            I18n.locale = :he
+            p.name = "Nissim"
+
+            Person.create!(passport: p, employer_id: 12345)
+          end
+
+          after do
+            I18n.locale = :en
+          end
+
+          let(:plucked) do
+            Person.where(employer_id: 12345).pluck("pass.name").first
+          end
+
+          let(:plucked_translations) do
+            Person.where(employer_id: 12345).pluck("pass.name_translations").first
+          end
+
+          let(:plucked_translations_field) do
+            Person.where(employer_id: 12345).pluck("pass.name_translations.en").first
+          end
+
+          it "returns the translation for the current locale" do
+            expect(plucked).to eq("Nissim")
+          end
+
+          it "returns the full _translation hash" do
+            expect(plucked_translations).to eq({ "en" => "Neil", "he" => "Nissim" })
+          end
+
+          it "returns the translation for the requested locale" do
+            expect(plucked_translations_field).to eq("Neil")
+          end
+        end
+      end
+
+      context 'when plucking a field to be demongoized' do
+
+        let(:plucked) do
+          Band.where(name: maniacs.name).pluck(:sales)
+        end
+
+        with_config_values :map_big_decimal_to_decimal128, true, false do
+
+          it "demongoizes the field" do
+            expect(plucked.first).to be_a(BigDecimal)
+            expect(plucked.first).to eq(BigDecimal("1E2"))
+          end
+        end
+      end
+
+      context "when plucking an embedded field" do
+        let(:label) { Label.new(sales: "1E2") }
+        let!(:band) { Band.create!(label: label) }
+
+        let(:plucked) do
+          Band.where(_id: band.id).tap do |crit|
+            crit.documents = [ band ]
+          end.pluck("label.sales")
+        end
+
+        it "demongoizes the field" do
+          expect(plucked).to eq([ BigDecimal("1E2") ])
+        end
+      end
+
+      context "when plucking an embeds_many field" do
+        let(:label) { Label.new(sales: "1E2") }
+        let!(:band) { Band.create!(labels: [label]) }
+
+        let(:plucked) { Band.where(_id: band.id).pluck("labels.sales") }
+
+        it "demongoizes the field" do
+          expect(plucked.first).to eq([ BigDecimal("1E2") ])
+        end
+      end
+
+      context "when plucking a nonexistent embedded field" do
+        let(:label) { Label.new(sales: "1E2") }
+        let!(:band) { Band.create!(label: label) }
+
+        let(:plucked) do
+          Band.where(_id: band.id).tap do |crit|
+            crit.documents = [ band ]
+          end.pluck("label.qwerty")
+        end
+
+        it "returns nil" do
+          expect(plucked.first).to eq(nil)
+        end
+      end
+
+      context "when tallying deeply nested arrays/embedded associations" do
+
+        let(:criteria) do
+          Person.all.tap do |crit|
+            crit.documents = [
+              Person.create!(addresses: [ Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 2 } } ]))) ]),
+              Person.create!(addresses: [ Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 2 } } ]))) ]),
+              Person.create!(addresses: [ Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 3 } } ]))) ]),
+            ]
+          end
+        end
+
+        let(:plucked) do
+          context.pluck("addresses.code.deepest.array.y.z")
+        end
+
+        it "returns the correct hash" do
+          expect(plucked).to eq([
+            [ [ 1, 2 ] ], [ [ 1, 2 ] ], [ [ 1, 3 ] ]
+          ])
+        end
+      end
+    end
+  end
+
+  describe "#pick" do
+
+    let(:depeche) do
+      Band.create!(name: "Depeche Mode", likes: 3)
+    end
+
+    let(:tool) do
+      Band.create!(name: "Tool", likes: 3)
+    end
+
+    let(:criteria) do
+      Band.all.tap do |crit|
+        crit.documents = [ depeche, tool ]
+      end
+    end
+
+    let(:context) do
+      described_class.new(criteria)
+    end
+
+    context "when picking a field" do
+
+      let(:picked) do
+        context.pick(:name)
+      end
+
+      it "returns one element" do
+        expect(picked).to eq("Depeche Mode")
+      end
+    end
+
+    context "when picking multiple fields" do
+
+      let(:picked) do
+        context.pick(:name, :likes)
+      end
+
+      it "returns an array" do
+        expect(picked).to eq([ "Depeche Mode", 3 ])
+      end
+    end
+
+    context "when no documents to pick" do
+
+      let(:criteria) do
+        Band.all.tap do |crit|
+          crit.documents = []
+        end
+      end
+
+      let(:picked) do
+        context.pick(:name)
+      end
+
+      it "returns nil" do
+        expect(picked).to be_nil
+      end
+    end
+  end
+
+  describe "#tally" do
+    let(:fans1) { [ Fanatic.new(age:1), Fanatic.new(age:2) ] }
+    let(:fans2) { [ Fanatic.new(age:1), Fanatic.new(age:2) ] }
+    let(:fans3) { [ Fanatic.new(age:1), Fanatic.new(age:3) ] }
+
+    let(:genres1) { [ { x: 1, y: { z: 1 } }, { x: 2, y: { z: 2 } }, { y: 3 } ]}
+    let(:genres2) { [ { x: 1, y: { z: 1 } }, { x: 2, y: { z: 2 } }, { y: 4 } ]}
+    let(:genres3) { [ { x: 1, y: { z: 1 } }, { x: 3, y: { z: 3 } }, { y: 5 } ]}
+
+    let(:label1) {  Label.new(name: "Atlantic") }
+    let(:label2) {  Label.new(name: "Atlantic") }
+    let(:label3) {  Label.new(name: "Columbia") }
+
+
+    let(:band1) { Band.new(origin: "tally", name: "Depeche Mode", years: 30, sales: "1E2", label: label1, genres: genres1) }
+    let(:band2) { Band.new(origin: "tally", name: "New Order", years: 30, sales: "2E3", label: label2, genres: genres2) }
+    let(:band3) { Band.new(origin: "tally", name: "10,000 Maniacs", years: 30, sales: "1E2", label: label3, genres: genres3) }
+    let(:band4) { Band.new(origin: "tally2", fanatics: fans1, genres: [1, 2]) }
+    let(:band5) { Band.new(origin: "tally2", fanatics: fans2, genres: [1, 2]) }
+    let(:band6) { Band.new(origin: "tally2", fanatics: fans3, genres: [1, 3]) }
+
+    let(:criteria) do
+      Band.where(origin: "tally").all.tap do |crit|
+        crit.documents = [ band1, band2, band3 ]
+      end
+    end
+
+    let(:criteria2) do
+      Band.where(origin: "tally2").tap do |crit|
+        crit.documents = [ band4, band5, band6 ]
+      end
+    end
+
+    let(:context) do
+      described_class.new(criteria)
+    end
+
+    let(:context2) do
+      described_class.new(criteria2)
+    end
+
+    context "when tallying a string" do
+      let(:tally) do
+        context.tally(:name)
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq("Depeche Mode" => 1, "New Order" => 1, "10,000 Maniacs" => 1)
+      end
+    end
+
+    context "using an aliased field" do
+      let(:tally) do
+        context.tally(:years)
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(30 => 3)
+      end
+    end
+
+    context "when tallying a demongoizable field" do
+      let(:tally) do
+        context.tally(:sales)
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(BigDecimal("1E2") => 2, BigDecimal("2E3") => 1)
+      end
+    end
+
+    context "when tallying a localized field" do
+      let(:d1) { Dictionary.new(description: 'en1') }
+      let(:d2) { Dictionary.new(description: 'en1') }
+      let(:d3) { Dictionary.new(description: 'en1') }
+      let(:d4) { Dictionary.new(description: 'en2') }
+
+      before do
+        I18n.locale = :en
+        d1
+        d2
+        d3
+        d4
+        I18n.locale = :de
+        d1.description = 'de1'
+        d2.description = 'de1'
+        d3.description = 'de2'
+        d4.description = 'de3'
+        I18n.locale = :en
+      end
+
+      let(:criteria) do
+        Dictionary.all.tap do |crit|
+          crit.documents = [ d1, d2, d3, d4 ]
+        end
+      end
+
+      context "when getting the demongoized field" do
+        let(:tallied) do
+          context.tally(:description)
+        end
+
+        it "returns the translation for the current locale" do
+          expect(tallied).to eq("en1" => 3, "en2" => 1)
+        end
+      end
+
+      context "when getting a specific locale" do
+        let(:tallied) do
+          context.tally("description.de")
+        end
+
+        it "returns the translation for the the specific locale" do
+          expect(tallied).to eq("de1" => 2, "de2" => 1, "de3" => 1)
+        end
+      end
+
+      context "when getting the full hash" do
+        let(:tallied) do
+          context.tally("description_translations")
+        end
+
+        it "returns the correct hash" do
+          expect(tallied).to eq(
+            {"de" => "de1", "en" => "en1" } => 2,
+            {"de" => "de2", "en" => "en1" } => 1,
+            {"de" => "de3", "en" => "en2" } => 1
+          )
+        end
+      end
+    end
+
+    context "when tallying an embedded localized field" do
+
+      let(:person1) { Person.create!(addresses: [ address1a, address1b ]) }
+      let(:person2) { Person.create!(addresses: [ address2a, address2b ]) }
+
+      let(:address1a) { Address.new(name: "en1") }
+      let(:address1b) { Address.new(name: "en2") }
+      let(:address2a) { Address.new(name: "en1") }
+      let(:address2b) { Address.new(name: "en3") }
+
+      before do
+        I18n.locale = :en
+        address1a
+        address1b
+        address2a
+        address2b
+        I18n.locale = :de
+        address1a.name = "de1"
+        address1b.name = "de2"
+        address2a.name = "de1"
+        address2b.name = "de3"
+        person1
+        person2
+        I18n.locale = :en
+      end
+
+      let(:criteria) do
+        Person.all.tap do |crit|
+          crit.documents = [ person1, person2 ]
+        end
+      end
+
+      context "when getting the demongoized field" do
+        let(:tallied) do
+          context.tally("addresses.name")
+        end
+
+        it "returns the translation for the current locale" do
+          expect(tallied).to eq(
+            [ "en1", "en2" ] => 1,
+            [ "en1", "en3" ] => 1,
+          )
+        end
+      end
+
+      context "when getting a specific locale" do
+        let(:tallied) do
+          context.tally("addresses.name.de")
+        end
+
+        it "returns the translation for the the specific locale" do
+          expect(tallied).to eq(
+            [ "de1", "de2" ] => 1,
+            [ "de1", "de3" ] => 1,
+          )
+        end
+      end
+
+      context "when getting the full hash" do
+        let(:tallied) do
+          context.tally("addresses.name_translations")
+        end
+
+        it "returns the correct hash" do
+          expect(tallied).to eq(
+            [{ "de" => "de1", "en" => "en1" }, { "de" => "de2", "en" => "en2" }] => 1,
+            [{ "de" => "de1", "en" => "en1" }, { "de" => "de3", "en" => "en3" }] => 1,
+          )
+        end
+      end
+    end
+
+    context "when tallying an embedded field" do
+      let(:tally) do
+        context.tally("label.name")
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq("Atlantic" => 2, "Columbia" => 1)
+      end
+    end
+
+    context "when tallying an element in an embeds_many field" do
+
+      let(:tally) do
+        context2.tally("fanatics.age")
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(
+          [1, 2] => 2,
+          [1, 3] => 1
+        )
+      end
+    end
+
+    context "when tallying an embeds_many field" do
+
+      let(:tally) do
+        context2.tally("fanatics")
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(
+          fans1 => 1,
+          fans2 => 1,
+          fans3 => 1,
+        )
+      end
+    end
+
+    context "when tallying a field of type array" do
+
+      let(:tally) do
+        context2.tally("genres")
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(
+          [1, 2] => 2,
+          [1, 3] => 1
+        )
+      end
+    end
+
+    context "when tallying an element from an array of hashes" do
+
+      let(:tally) do
+        context.tally("genres.x")
+      end
+
+      it "returns the correct hash without the nil keys" do
+        expect(tally).to eq(
+          [1, 2] => 2,
+          [1, 3] => 1
+        )
+      end
+    end
+
+    context "when tallying an element from an array of hashes; with duplicate" do
+
+      let(:band4) { Band.new(origin: "tally", genres: [ { x: 1 }, {x: 1} ] ) }
+
+      let(:criteria) do
+        Band.where(origin: "tally").all.tap do |crit|
+          crit.documents = [ band1, band2, band3, band4 ]
+        end
+      end
+
+      let(:tally) do
+        context.tally("genres.x")
+      end
+
+      it "returns the correct hash without the nil keys" do
+        expect(tally).to eq(
+          [1, 2] => 2,
+          [1, 3] => 1,
+          [1, 1] => 1,
+        )
+      end
+    end
+
+    context "when tallying an aliased field of type array" do
+
+      let(:person1) { Person.new(array: [ 1, 2 ]) }
+      let(:person2) { Person.new(array: [ 1, 3 ]) }
+
+      let(:criteria) do
+        Person.all.tap do |crit|
+          crit.documents = [ person1, person2 ]
+        end
+      end
+
+      let(:tally) do
+        context.tally("array")
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(
+          [1, 2] => 1,
+          [1, 3] => 1
+        )
+      end
+    end
+
+    context "when going multiple levels deep in arrays" do
+
+      let(:tally) do
+        context.tally("genres.y.z")
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(
+          [1, 2] => 2,
+          [1, 3] => 1
+        )
+      end
+    end
+
+    context "when going multiple levels deep in an array" do
+
+      let(:tally) do
+        context.tally("genres.y.z")
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(
+          [1, 2] => 2,
+          [1, 3] => 1
+        )
+      end
+    end
+
+    context "when tallying deeply nested arrays/embedded associations" do
+
+      let(:person1) { Person.new(addresses: [ Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 2 } } ]))) ]) }
+      let(:person2) { Person.new(addresses: [ Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 2 } } ]))) ]) }
+      let(:person3) { Person.new(addresses: [ Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 3 } } ]))) ]) }
+
+      let(:criteria) do
+        Person.all.tap do |crit|
+          crit.documents = [ person1, person2, person3 ]
+        end
+      end
+
+      let(:tally) do
+        context.tally("addresses.code.deepest.array.y.z")
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(
+          [ [ 1, 2 ] ] => 2,
+          [ [ 1, 3 ] ] => 1
+        )
+      end
+    end
+
+    context "when tallying deeply nested arrays/embedded associations" do
+
+      let(:person1) do
+        Person.new(addresses: [ Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 2 } } ]))),
+                                    Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 2 } } ]))) ])
+      end
+
+      let(:person2) do
+        Person.new(addresses: [ Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 2 } } ]))),
+                                    Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 2 } } ]))) ])
+      end
+
+      let(:person3) do
+        Person.new(addresses: [ Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 3 } } ]))),
+                                    Address.new(code: Code.new(deepest: Deepest.new(array: [ { y: { z: 1 } }, { y: { z: 3 } } ]))) ])
+      end
+
+      let(:criteria) do
+        Person.all.tap do |crit|
+          crit.documents = [ person1, person2, person3 ]
+        end
+      end
+
+      let(:tally) do
+        context.tally("addresses.code.deepest.array.y.z")
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(
+          [ [ 1, 2 ], [ 1, 2 ] ] => 2,
+          [ [ 1, 3 ], [ 1, 3 ] ] => 1
+        )
+      end
+    end
+
+    context "when some keys are missing" do
+
+      let(:criteria) do
+        Band.where(origin: "tally").all.tap do |crit|
+          crit.documents = [ band1, band2, band3 ]
+          3.times{ crit.documents << Band.new(origin: "tally") }
+        end
+      end
+
+      let(:tally) do
+        context.tally(:name)
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(
+          "Depeche Mode" => 1,
+          "New Order" => 1,
+          "10,000 Maniacs" => 1,
+          nil => 3
+        )
+      end
+    end
+
+    context "when the first element is an embeds_one" do
+      let(:person1) { Person.create!(name: Name.new(translations: [ Translation.new(language: 1), Translation.new(language: 2) ])) }
+      let(:person2) { Person.create!(name: Name.new(translations: [ Translation.new(language: 1), Translation.new(language: 2) ])) }
+      let(:person3) { Person.create!(name: Name.new(translations: [ Translation.new(language: 1), Translation.new(language: 3) ])) }
+
+      let(:criteria) do
+        Person.all.tap do |crit|
+          crit.documents = [ person1, person2, person3 ]
+        end
+      end
+
+      let(:tally) do
+        context.tally("name.translations.language")
+      end
+
+      it "returns the correct hash" do
+        expect(tally).to eq(
+          [1, 2] => 2,
+          [1, 3] => 1
+        )
       end
     end
   end