load a belongs to relationship

2 files changed, 5 insertions, 3 deletions

diff --git a/lib/humble/column.rb b/lib/humble/column.rb
index 8320e84..0fa73b8 100644
--- a/lib/humble/column.rb
+++ b/lib/humble/column.rb
@@ -52,8 +52,10 @@ module Humble
       @type = type
     end
 
-    def apply(id, entity)
-      #entity.public_send("#{column_name}=", id)
+    def apply(value, entity)
+      child_entity = @type.new
+      column = column_name.to_s.gsub(/_id/, '')
+      entity.public_send("#{column}=", child_entity)
     end
 
     def prepare(entity)
diff --git a/spec/integration/select_spec.rb b/spec/integration/select_spec.rb
index 9a1c4ea..8b453b5 100644
--- a/spec/integration/select_spec.rb
+++ b/spec/integration/select_spec.rb
@@ -37,7 +37,7 @@ describe "select items" do
       expect(result).to be_instance_of(Movie)
       expect(result.id).to eql(movie_id)
       expect(result.name).to eql('blood in, blood out')
-      #expect(result.studio).to be_instance_of(Studio)
+      expect(result.studio).to be_instance_of(Studio)
     end
   end
 end