mongoid 7.5.4 → 8.1.9

data/spec/mongoid/contextual/memory_spec.rb

@@ -4,6 +4,36 @@ require "spec_helper"
 
 describe Mongoid::Contextual::Memory do
 
+  shared_examples "raises an error when no documents" do
+    context "when there are no matching documents" do
+      let(:criteria) do
+        Address.all.tap do |crit|
+          crit.documents = []
+        end
+      end
+
+      it "returns nil" do
+        expect do
+          context.send(method)
+        end.to raise_error(Mongoid::Errors::DocumentNotFound, /Could not find a document of class Address./)
+      end
+    end
+  end
+
+  shared_examples "returns nil when no documents" do
+    context "when there are no matching documents" do
+      let(:criteria) do
+        Address.all.tap do |crit|
+          crit.documents = []
+        end
+      end
+
+      it "returns nil" do
+        expect(context.send(method)).to be_nil
+      end
+    end
+  end
+
   [ :blank?, :empty? ].each do |method|
 
     describe "##{method}" do
@@ -492,6 +522,8 @@ describe Mongoid::Contextual::Memory do
       end
 
       context "when getting a localized field" do
+        with_default_i18n_configs
+
         before do
           I18n.locale = :en
           d = Dictionary.create!(description: 'english-text')
@@ -500,10 +532,6 @@ describe Mongoid::Contextual::Memory do
           d.save!
         end
 
-        after do
-          I18n.locale = :en
-        end
-
         let(:criteria) do
           Dictionary.all.tap do |crit|
             crit.documents = [ Dictionary.first ]
@@ -547,16 +575,10 @@ describe Mongoid::Contextual::Memory do
 
         context 'when fallbacks are enabled with a locale list' do
           require_fallbacks
+          with_default_i18n_configs
 
-          around(:all) do |example|
-            prev_fallbacks = I18n.fallbacks.dup
+          before do
             I18n.fallbacks[:he] = [ :en ]
-            example.run
-            I18n.fallbacks = prev_fallbacks
-          end
-
-          after do
-            I18n.locale = :en
           end
 
           let(:distinct) do
@@ -565,13 +587,15 @@ describe Mongoid::Contextual::Memory do
 
           it "correctly uses the fallback" do
             I18n.locale = :en
-            d = Dictionary.create!(description: 'english-text')
+            Dictionary.create!(description: 'english-text')
             I18n.locale = :he
             distinct.should == "english-text"
           end
         end
 
         context "when the localized field is embedded" do
+          with_default_i18n_configs
+
           let(:person) do
             p = Passport.new
             I18n.locale = :en
@@ -582,10 +606,6 @@ describe Mongoid::Contextual::Memory do
             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 ]
@@ -746,52 +766,158 @@ describe Mongoid::Contextual::Memory do
       Address.new(street: "friedel")
     end
 
-    context "when there are matching documents" do
+    let(:criteria) do
+      Address.where(street: "hobrecht").tap do |crit|
+        crit.documents = [ hobrecht, friedel ]
+      end
+    end
+
+    context "when not passing options" do
 
-      let(:criteria) do
-        Address.where(street: "hobrecht").tap do |crit|
-          crit.documents = [ hobrecht, friedel ]
+      context "when there are matching documents" do
+
+        let(:context) do
+          described_class.new(criteria)
+        end
+
+        it "returns true" do
+          expect(context).to be_exists
         end
       end
 
-      let(:context) do
-        described_class.new(criteria)
+      context "when there are no matching documents" do
+
+        let(:criteria) do
+          Address.where(street: "pfluger").tap do |crit|
+            crit.documents = [ hobrecht, friedel ]
+          end
+        end
+
+        let(:context) do
+          described_class.new(criteria)
+        end
+
+        it "returns false" do
+          expect(context).to_not be_exists
+        end
       end
 
-      it "returns true" do
-        expect(context).to be_exists
+      context 'when there is a collation on the criteria' do
+
+        let(:criteria) do
+          Address.where(street: "pfluger").tap do |crit|
+            crit.documents = [ hobrecht, friedel ]
+          end.collation(locale: 'en_US', strength: 2)
+        end
+
+        it "raises an exception" do
+          expect {
+            context
+          }.to raise_exception(Mongoid::Errors::InMemoryCollationNotSupported)
+        end
       end
     end
 
-    context "when there are no matching documents" do
+    context "when passing an _id" do
 
-      let(:criteria) do
-        Address.where(street: "pfluger").tap do |crit|
-          crit.documents = [ hobrecht, friedel ]
+      context "when its of type BSON::ObjectId" do
+
+        context "when calling it on an empty criteria" do
+
+          it "returns true" do
+            expect(criteria.exists?(hobrecht._id)).to be true
+          end
+        end
+
+        context "when calling it on a criteria that includes the object" do
+
+          it "returns true" do
+            expect(criteria.where(street: hobrecht.street).exists?(hobrecht._id)).to be true
+          end
+        end
+
+        context "when calling it on a criteria that does not include the object" do
+
+          it "returns false" do
+            expect(criteria.where(street: "bogus").exists?(hobrecht._id)).to be false
+          end
+        end
+
+        context "when the id does not exist" do
+
+          it "returns false" do
+            expect(criteria.exists?(BSON::ObjectId.new)).to be false
+          end
         end
       end
 
-      let(:context) do
-        described_class.new(criteria)
+      context "when its of type String" do
+
+        context "when the id exists" do
+
+          it "returns true" do
+            expect(criteria.exists?(hobrecht._id.to_s)).to be true
+          end
+        end
+
+        context "when the id does not exist" do
+
+          it "returns false" do
+            expect(criteria.exists?(BSON::ObjectId.new.to_s)).to be false
+          end
+        end
+      end
+    end
+
+    context "when passing a hash" do
+
+      context "when calling it on an empty criteria" do
+
+        it "returns true" do
+          expect(criteria.exists?(street: hobrecht.street)).to be true
+        end
+      end
+
+      context "when calling it on a criteria that includes the object" do
+
+        it "returns true" do
+          expect(criteria.where(_id: hobrecht._id).exists?(street: hobrecht.street)).to be true
+        end
+      end
+
+      context "when calling it on a criteria that does not include the object" do
+
+        it "returns false" do
+          expect(criteria.where(_id: BSON::ObjectId.new).exists?(street: hobrecht.street)).to be false
+        end
       end
 
+      context "when the conditions don't match" do
+
+        it "returns false" do
+          expect(criteria.exists?(street: "bogus")).to be false
+        end
+      end
+    end
+
+    context "when passing false" do
+
       it "returns false" do
-        expect(context).to_not be_exists
+        expect(criteria.exists?(false)).to be false
       end
     end
 
-    context 'when there is a collation on the criteria' do
+    context "when passing nil" do
 
-      let(:criteria) do
-        Address.where(street: "pfluger").tap do |crit|
-          crit.documents = [ hobrecht, friedel ]
-        end.collation(locale: 'en_US', strength: 2)
+      it "returns false" do
+        expect(criteria.exists?(nil)).to be false
       end
+    end
 
-      it "raises an exception" do
-        expect {
-          context
-        }.to raise_exception(Mongoid::Errors::InMemoryCollationNotSupported)
+    context "when the limit is 0" do
+
+      it "returns false" do
+        expect(criteria.limit(0).exists?).to be false
       end
     end
   end
@@ -800,6 +926,8 @@ describe Mongoid::Contextual::Memory do
 
     describe "##{method}" do
 
+      let(:method) { method }
+
       let(:hobrecht) do
         Address.new(street: "hobrecht")
       end
@@ -830,9 +958,7 @@ describe Mongoid::Contextual::Memory do
         expect(context.send(method, 1)).to eq([ hobrecht ])
       end
 
-      it "returns the matching document when passing deprecated options" do
-        expect(context.send(method, id_sort: :none)).to eq(hobrecht)
-      end
+      include_examples "returns nil when no documents"
 
       context 'when there is a collation on the criteria' do
 
@@ -851,8 +977,39 @@ describe Mongoid::Contextual::Memory do
     end
   end
 
+  describe "#first!" do
+
+    let(:method) { :first! }
+
+    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.first!).to eq(hobrecht)
+    end
+
+    include_examples "raises an error when no documents"
+  end
+
   describe "#take" do
 
+    let(:method) { :take }
+
     let(:hobrecht) do
       Address.new(street: "hobrecht")
     end
@@ -883,6 +1040,8 @@ describe Mongoid::Contextual::Memory do
       expect(context.take(1)).to eq([ hobrecht ])
     end
 
+    include_examples "returns nil when no documents"
+
     context 'when there is a collation on the criteria' do
 
       let(:criteria) do
@@ -901,6 +1060,8 @@ describe Mongoid::Contextual::Memory do
 
   describe "#take!" do
 
+    let(:method) { :take! }
+
     let(:hobrecht) do
       Address.new(street: "hobrecht")
     end
@@ -923,19 +1084,7 @@ describe Mongoid::Contextual::Memory 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
+    include_examples "raises an error when no documents"
 
     context 'when there is a collation on the criteria' do
 
@@ -1056,6 +1205,8 @@ describe Mongoid::Contextual::Memory do
 
   describe "#last" do
 
+    let(:method) { :last }
+
     let(:hobrecht) do
       Address.new(street: "hobrecht")
     end
@@ -1086,9 +1237,7 @@ describe Mongoid::Contextual::Memory do
       expect(context.last(1)).to eq([ friedel ])
     end
 
-    it "returns the matching document when passing deprecated options" do
-      expect(context.last(id_sort: :none)).to eq(friedel)
-    end
+    include_examples "returns nil when no documents"
 
     context 'when there is a collation on the criteria' do
 
@@ -1106,6 +1255,102 @@ describe Mongoid::Contextual::Memory do
     end
   end
 
+  describe "#last!" do
+    let(:method) { :last! }
+
+    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 last matching document" do
+      expect(context.last!).to eq(friedel)
+    end
+
+    include_examples "raises an error when no documents"
+  end
+
+  [ :second,
+    :third,
+    :fourth,
+    :fifth,
+    :second_to_last,
+    :third_to_last
+  ].each do |meth|
+    describe "##{meth}" do
+      let(:method) { meth }
+
+      let(:addresses) do
+        [
+          Address.new,
+          Address.new,
+          Address.new,
+          Address.new,
+          Address.new,
+        ]
+      end
+
+      let(:criteria) do
+        Address.all.tap do |crit|
+          crit.documents = addresses
+        end
+      end
+
+      let(:context) do
+        described_class.new(criteria)
+      end
+
+      it "returns the matching document" do
+        expect(context.send(method)).to eq(addresses.send(method))
+      end
+
+      include_examples "returns nil when no documents"
+    end
+
+    describe "##{meth}!" do
+      let(:method) { "#{meth}!" }
+
+      let(:addresses) do
+        [
+          Address.new,
+          Address.new,
+          Address.new,
+          Address.new,
+          Address.new,
+        ]
+      end
+
+      let(:criteria) do
+        Address.all.tap do |crit|
+          crit.documents = addresses
+        end
+      end
+
+      let(:context) do
+        described_class.new(criteria)
+      end
+
+      it "returns the matching document" do
+        expect(context.send(method)).to eq(addresses.send(meth))
+      end
+
+      include_examples "raises an error when no documents"
+    end
+  end
+
   [ :length, :size ].each do |method|
 
     describe "##{method}" do
@@ -1251,11 +1496,11 @@ describe Mongoid::Contextual::Memory do
       config_override :legacy_pluck_distinct, true
 
       let(:hobrecht) do
-        Address.new(street: "hobrecht")
+        Address.new(street: "hobrecht", number: 213)
       end
 
       let(:friedel) do
-        Address.new(street: "friedel")
+        Address.new(street: "friedel", number: 11)
       end
 
       let(:criteria) do
@@ -1275,6 +1520,17 @@ describe Mongoid::Contextual::Memory do
         end
       end
 
+      context "when plucking multiple fields" do
+
+        let!(:plucked) do
+          context.pluck(:street, :number)
+        end
+
+        it "returns the values as an array" do
+          expect(plucked).to eq([ ["hobrecht", 213], ["friedel", 11] ])
+        end
+      end
+
       context "when plucking a mix of empty and non-empty values" do
 
         let(:empty_doc) do
@@ -1304,7 +1560,7 @@ describe Mongoid::Contextual::Memory do
             context.pluck(:foo)
           end
 
-          it "returns a empty array" do
+          it "returns an empty array" do
             expect(plucked).to eq([nil, nil])
           end
         end
@@ -1315,7 +1571,7 @@ describe Mongoid::Contextual::Memory do
             context.pluck(:foo, :bar)
           end
 
-          it "returns a empty array" do
+          it "returns an empty array" do
             expect(plucked).to eq([[nil, nil], [nil, nil]])
           end
         end
@@ -1352,7 +1608,7 @@ describe Mongoid::Contextual::Memory do
         Band.create!(name: "Photek", likes: 1)
       end
 
-      let(:maniacs) do
+      let!(:maniacs) do
         Band.create!(name: "10,000 Maniacs", likes: 1, sales: "1E2")
       end
 
@@ -1537,6 +1793,7 @@ describe Mongoid::Contextual::Memory do
       end
 
       context 'when plucking a localized field' do
+        with_default_i18n_configs
 
         before do
           I18n.locale = :en
@@ -1546,10 +1803,6 @@ describe Mongoid::Contextual::Memory do
           d.save!
         end
 
-        after do
-          I18n.locale = :en
-        end
-
         let(:criteria) do
           Dictionary.all.tap do |crit|
             crit.documents = [ Dictionary.first ]
@@ -1607,16 +1860,10 @@ describe Mongoid::Contextual::Memory do
 
         context 'when fallbacks are enabled with a locale list' do
           require_fallbacks
+          with_default_i18n_configs
 
-          around(:all) do |example|
-            prev_fallbacks = I18n.fallbacks.dup
+          before do
             I18n.fallbacks[:he] = [ :en ]
-            example.run
-            I18n.fallbacks = prev_fallbacks
-          end
-
-          after do
-            I18n.locale = :en
           end
 
           let(:plucked) do
@@ -1625,13 +1872,15 @@ describe Mongoid::Contextual::Memory do
 
           it "correctly uses the fallback" do
             I18n.locale = :en
-            d = Dictionary.create!(description: 'english-text')
+            Dictionary.create!(description: 'english-text')
             I18n.locale = :he
             plucked.should == "english-text"
           end
         end
 
         context "when the localized field is embedded" do
+          with_default_i18n_configs
+
           before do
             p = Passport.new
             I18n.locale = :en
@@ -1642,10 +1891,6 @@ describe Mongoid::Contextual::Memory do
             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
@@ -1678,9 +1923,12 @@ describe Mongoid::Contextual::Memory do
           Band.where(name: maniacs.name).pluck(:sales)
         end
 
-        it "demongoizes the field" do
-          expect(plucked.first).to be_a(BigDecimal)
-          expect(plucked.first).to eq(BigDecimal("1E2"))
+        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
 
@@ -1690,7 +1938,7 @@ describe Mongoid::Contextual::Memory do
 
         let(:plucked) do
           Band.where(_id: band.id).tap do |crit|
-            crit.documents = [band]
+            crit.documents = [ band ]
           end.pluck("label.sales")
         end
 
@@ -1716,7 +1964,7 @@ describe Mongoid::Contextual::Memory do
 
         let(:plucked) do
           Band.where(_id: band.id).tap do |crit|
-            crit.documents = band
+            crit.documents = [ band ]
           end.pluck("label.qwerty")
         end
 
@@ -1725,7 +1973,7 @@ describe Mongoid::Contextual::Memory do
         end
       end
 
-      context "when tallying deeply nested arrays/embedded associations" do
+      context "when plucking deeply nested arrays/embedded associations" do
 
         let(:criteria) do
           Person.all.tap do |crit|
@@ -1750,6 +1998,520 @@ describe Mongoid::Contextual::Memory do
     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
+      with_default_i18n_configs
+
+      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
+      with_default_i18n_configs
+
+      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
+
   describe '#inc' do
 
     let(:criteria) do