Fixed preloading belongs_to associations which reference a custom foreign key. References #9640.

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@8762 5ecf4fe2-1ee6-0310-87b1-e25e094e27de

6 files changed, 47 insertions, 1 deletions

diff --git a/activerecord/test/cases/associations/eager_test.rb b/activerecord/test/cases/associations/eager_test.rb
index 9c258b574f..b606092d7e 100644
--- a/activerecord/test/cases/associations/eager_test.rb
+++ b/activerecord/test/cases/associations/eager_test.rb
@@ -7,10 +7,13 @@ require 'models/category'
 require 'models/company'
 require 'models/person'
 require 'models/reader'
+require 'models/owner'
+require 'models/pet'
 
 class EagerAssociationTest < ActiveRecord::TestCase
   fixtures :posts, :comments, :authors, :categories, :categories_posts,
-            :companies, :accounts, :tags, :taggings, :people, :readers
+            :companies, :accounts, :tags, :taggings, :people, :readers,
+            :owners, :pets
 
   def test_loading_with_one_association
     posts = Post.find(:all, :include => :comments)
@@ -73,6 +76,11 @@ class EagerAssociationTest < ActiveRecord::TestCase
     assert_nil Post.find(posts(:authorless).id, :include => :author).author
   end
 
+  def test_eager_association_loading_with_belongs_to_and_foreign_keys
+    pets = Pet.find(:all, :include => :owner)
+    assert_equal 3, pets.length
+  end
+
   def test_eager_association_loading_with_belongs_to
     comments = Comment.find(:all, :include => :post)
     assert_equal 10, comments.length
diff --git a/activerecord/test/fixtures/owners.yml b/activerecord/test/fixtures/owners.yml
new file mode 100644
index 0000000000..d5493a84b7
--- /dev/null
+++ b/activerecord/test/fixtures/owners.yml
@@ -0,0 +1,7 @@
+blackbeard:
+  owner_id: 1
+  name: blackbeard
+
+ashley:
+  owner_id: 2
+  name: ashley
diff --git a/activerecord/test/fixtures/pets.yml b/activerecord/test/fixtures/pets.yml
new file mode 100644
index 0000000000..a1601a53f0
--- /dev/null
+++ b/activerecord/test/fixtures/pets.yml
@@ -0,0 +1,14 @@
+parrot:
+  pet_id: 1
+  name: parrot
+  owner_id: 1
+
+chew:
+  pet_id: 2
+  name: chew
+  owner_id: 2
+
+mochi:
+  pet_id: 3
+  name: mochi
+  owner_id: 2
diff --git a/activerecord/test/models/owner.rb b/activerecord/test/models/owner.rb
new file mode 100644
index 0000000000..dbaf2ce688
--- /dev/null
+++ b/activerecord/test/models/owner.rb
@@ -0,0 +1,4 @@
+class Owner < ActiveRecord::Base
+  set_primary_key :owner_id
+  has_many :pets
+end
\ No newline at end of file
diff --git a/activerecord/test/models/pet.rb b/activerecord/test/models/pet.rb
new file mode 100644
index 0000000000..889ce46f33
--- /dev/null
+++ b/activerecord/test/models/pet.rb
@@ -0,0 +1,4 @@
+class Pet < ActiveRecord::Base
+  set_primary_key :pet_id
+  belongs_to :owner
+end
\ No newline at end of file
diff --git a/activerecord/test/schema/schema.rb b/activerecord/test/schema/schema.rb
index bdf128aecc..8ab2022f8b 100644
--- a/activerecord/test/schema/schema.rb
+++ b/activerecord/test/schema/schema.rb
@@ -358,4 +358,13 @@ ActiveRecord::Schema.define do
   create_table 'warehouse-things', :force => true do |t|
     t.integer :value
   end
+
+  create_table :owners, :primary_key => :owner_id ,:force => true do |t|
+    t.string :name
+  end
+
+  create_table :pets, :primary_key => :pet_id ,:force => true do |t|
+    t.string :name
+    t.integer :owner_id, :integer
+  end
 end