1 files changed, 10 insertions, 5 deletions
diff --git a/activerecord/test/cases/pk_test.rb b/activerecord/test/cases/pk_test.rb
index 2502c136cc..5bba1d5625 100644
--- a/activerecord/test/cases/pk_test.rb
+++ b/activerecord/test/cases/pk_test.rb
@@ -9,13 +9,18 @@ require 'models/mixed_case_monkey'
 class PrimaryKeysTest < ActiveRecord::TestCase
   fixtures :topics, :subscribers, :movies, :mixed_case_monkeys
 
-  def test_key
-    # test new record
+  def test_to_key_with_default_primary_key
     topic = Topic.new
-    assert topic.key.nil?
-    # test existing record
+    assert topic.to_key.nil?
     topic = Topic.find(1)
-    assert_equal topic.key, [1]
+    assert_equal topic.to_key, [1]
+  end
+
+  def test_to_key_with_customized_primary_key
+    keyboard = Keyboard.new
+    assert keyboard.to_key.nil?
+    keyboard.save
+    assert_equal keyboard.to_key, [keyboard.id]
   end
 
   def test_integer_key