fix: lease_connection -> connection

Author: BuonOmo

test/cases/primary_keys_test.rb

@@ -97,4 +97,38 @@ def test_schema_dump_primary_key_integer_with_default_nil
       assert_match %r{create_table "int_defaults", id: :bigint, default: nil}, schema
     end
   end
+
+  class PrimaryKeyHiddenColumnTest < ActiveRecord::TestCase
+    class Rocket < ActiveRecord::Base
+    end
+
+    def setup
+      connection = ActiveRecord::Base.connection
+      connection.execute <<-SQL
+        CREATE TABLE rockets(
+          id SERIAL PRIMARY KEY USING HASH WITH (bucket_count=4)
+        )
+      SQL
+    end
+
+    def teardown
+      ActiveRecord::Base.connection.drop_table :rockets
+    end
+
+    def test_to_key_with_hidden_primary_key_part
+      rocket = Rocket.new
+      assert_nil rocket.to_key
+      rocket.save
+      assert_equal rocket.to_key, [rocket.id]
+    end
+
+    def test_read_attribute_with_hidden_primary_key_part
+      rocket = Rocket.create!
+      id = assert_not_deprecated(ActiveRecord.deprecator) do
+        rocket.read_attribute(:id)
+      end
+
+      assert_equal rocket.id, id
+    end
+  end
 end

test/cases/schema_dumper_test.rb

@@ -10,11 +10,6 @@ class SchemaDumperTest < ActiveRecord::TestCase
     include SchemaDumpingHelper
     self.use_transactional_tests = false
 
-    setup do
-      @schema_migration = ActiveRecord::Base.connection_pool.schema_migration
-      @schema_migration.create_table
-    end
-
     # See https://github.com/cockroachdb/activerecord-cockroachdb-adapter/issues/347
     def test_dump_index_rather_than_unique_constraints
       ActiveRecord::Base.with_connection do |conn|
@@ -207,7 +202,7 @@ def down
       $stdout = original
     end
 
-    if ActiveRecord::Base.lease_connection.supports_check_constraints?
+    if ActiveRecord::Base.connection.supports_check_constraints?
       def test_schema_dumps_check_constraints
         constraint_definition = dump_table_schema("products").split(/\n/).grep(/t.check_constraint.*products_price_check/).first.strip
         if current_adapter?(:Mysql2Adapter)