data_broker 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 92b08a7cb73e63a0d6eebc8cfe06f85ce03bba5f
-  data.tar.gz: 9a7ff3d396bf73b28fb40cba1a67eb251c33cddb
+  metadata.gz: efa47ef957de906755498b6ae0f153e142364672
+  data.tar.gz: 3d18d6b223a291474a1a75375dca874112cf927f
 SHA512:
-  metadata.gz: 017e64e604f99616a4bf08f6dcd887b2f6d24f59fcdaf6bcbace7b0d60ab77e40c4a2609f53981f96958f3a3aefb74b5ff1a2af6e03dc37173f2f55884fb96c5
-  data.tar.gz: 350da4a150a55351957821d047ab8da7bc37e8d69ae68c90c355de13f3302ecc570bcc0baa7c2cdd9b6748080ef4c3ecc848d0e18834e62f3bec7e1e88a24d58
+  metadata.gz: c0bd2f321438ba7e89a3b87f6f96ee23c63f347e5f186129744ec9177376b42ce6ded419f2296a2c0315194116266f3c9c514f00b28b271c0da987d11a39fbac
+  data.tar.gz: f4d8e9924a6d49cf3138b34b2c41b2a072cd8d7415154504c8aea0d72a25f00398427016e703bcab311c59c16ed5ecf3e2d2aef2dfc5e4d22b8a4a68a44f4923

data/lib/data_broker/mapper.rb

@@ -37,33 +37,43 @@ module DataBroker
         end
       end
 
-      def all
-        records_to_objects(self.record.all)
+      def all(eager_load: false)
+        records_to_objects(self.record.all, :eager_load => eager_load)
       end
 
-      def find(id)
+      def find(id, eager_load: true)
         record = self.eager_record.find_by(:id => id)
-        record.nil? ? nil : records_to_objects(record, :eager_load => true).first
+        if record.nil?
+          nil
+        else
+          records_to_objects(record, :eager_load => eager_load).first
+        end
       end
 
-      def where(conds)
+      def where(eager_load: true, **conds)
         conds = Hash[conds.map { |key, value| [self.mapping[key], value] }]
-        records_to_objects(self.eager_record.where(conds), :eager_load => true)
+        records_to_objects(
+          self.eager_record.where(conds), :eager_load => eager_load
+        )
       end
 
-      def save(model)
+      def save(model, eager_load: true)
         record = self.eager_record.find_by(:id => model.id) || self.record.new
         attributes = Hash[self.mapping.map { |key, value| [value, model[key]] }]
         attributes.delete(:id)
         record.attributes = attributes
 
         record.save!
-        records_to_objects(record.reload, :eager_load => true).first
+        records_to_objects(record.reload, :eager_load => eager_load).first
       end
 
-      def destroy(model)
+      def destroy(model, eager_load: false)
         record = self.record.find_by(:id => model.id)
-        record.nil? ? nil : records_to_objects(record.destroy).first
+        if record.nil?
+          nil
+        else
+          records_to_objects(record.destroy, :eager_load => false).first
+        end
       end
 
       def records_to_objects(records, eager_load: false)

data/lib/data_broker/version.rb

@@ -1,3 +1,3 @@
 module DataBroker
-  VERSION = "0.2.1"
+  VERSION = "0.2.2"
 end

data/spec/mapper_spec.rb

@@ -88,6 +88,14 @@ describe DataBroker::Mapper do
 
       expect(obj).to be_a(mapper.model)
     end
+
+    it "eager loads if requested" do
+      3.times { mapper.record.create }
+      expect(mapper).to receive(:load_parents!).exactly(3).times
+      expect(mapper).to receive(:load_children!).once
+
+      mapper.all(:eager_load => true)
+    end
   end
 
   describe "#find" do
@@ -130,6 +138,17 @@ describe DataBroker::Mapper do
       ids = obj.instance_variable_get(:@child_ids)
       expect(ids).to match_array([id])
     end
+
+    it "doesn't eager load if requested" do
+      expect(parent_mapper).to_not receive(:load_children!)
+      expect(parent_mapper).to_not receive(:load_parents!)
+
+      parent_id = parent_mapper.record.create.id
+      mapper.record.create(
+        :child_name => "Bobby", :parent_id => parent_id
+      ).id
+      parent_mapper.find(parent_id, :eager_load => false)
+    end
   end
 
   describe "#where" do
@@ -186,6 +205,16 @@ describe DataBroker::Mapper do
       ids = obj.instance_variable_get(:@child_ids)
       expect(ids).to match_array([id])
     end
+
+    it "doesn't eager load if requested" do
+      expect(parent_mapper).to_not receive(:load_children!)
+      expect(parent_mapper).to_not receive(:load_parents!)
+      parent_id = parent_mapper.record.create.id
+      mapper.record.create(
+        :child_name => "Bobby", :parent_id => parent_id
+      ).id
+      parent_mapper.where(:id => parent_id, :eager_load => false).first
+    end
   end
 
   describe "#save" do
@@ -210,6 +239,12 @@ describe DataBroker::Mapper do
 
       expect(rt.id).to eq(model.id)
     end
+
+    it "doesn't eager load if requested" do
+      expect(mapper).to_not receive(:load_children!)
+      expect(mapper).to_not receive(:load_parents!)
+      mapper.save(mapper.model.new, :eager_load => false)
+    end
   end
 
   describe "#destroy" do
@@ -231,5 +266,11 @@ describe DataBroker::Mapper do
 
       expect(mapper.find(model.id)).to be_nil
     end
+
+    it "eager loads if requests" do
+      expect(mapper).to receive(:load_children!).once
+      model = mapper.save(mapper.model.new)
+      mapper.destroy(model, :eager_load => true)
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: data_broker
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - Shane Emmons
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-01-14 00:00:00.000000000 Z
+date: 2014-01-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: virtus