Merge pull request #27274 from kamipo/primary_key_with_auto_increment_and_bigint

Make `:auto_increment` option works on `:bigint`

activerecord/lib/active_record/connection_adapters/postgresql/schema_definitions.rb

@@ -44,8 +44,12 @@ module ActiveRecord
         def primary_key(name, type = :primary_key, **options)
           if type == :uuid
             options[:default] = options.fetch(:default, "gen_random_uuid()")
-          elsif options.delete(:auto_increment) == true && type == :integer
+          elsif options.delete(:auto_increment) == true && %i(integer bigint).include?(type)
-            type = :serial
+            type = if type == :bigint || options[:limit] == 8
+              :bigserial
+            else
+              :serial
+            end
           end
 
           super

activerecord/lib/active_record/connection_adapters/sqlite3/schema_definitions.rb

@@ -3,7 +3,7 @@ module ActiveRecord
     module SQLite3
       module ColumnMethods
         def primary_key(name, type = :primary_key, **options)
-          if options.delete(:auto_increment) == true && type == :integer
+          if options.delete(:auto_increment) == true && %i(integer bigint).include?(type)
             type = :primary_key
           end
 

activerecord/test/cases/primary_keys_test.rb

@@ -224,7 +224,6 @@ class PrimaryKeyWithAutoIncrementTest < ActiveRecord::TestCase
 
   def setup
     @connection = ActiveRecord::Base.connection
-    @connection.create_table(:auto_increments, id: :integer, auto_increment: true, force: true)
   end
 
   def teardown
@@ -232,15 +231,26 @@ class PrimaryKeyWithAutoIncrementTest < ActiveRecord::TestCase
   end
 
   def test_primary_key_with_auto_increment
-    record1 = AutoIncrement.create!
+    @connection.create_table(:auto_increments, id: :integer, auto_increment: true, force: true)
-    assert_not_nil record1.id
+    assert_auto_incremented
-
-    record1.destroy
-
-    record2 = AutoIncrement.create!
-    assert_not_nil record2.id
-    assert_operator record2.id, :>, record1.id
   end
+
+  def test_primary_key_with_auto_increment_and_bigint
+    @connection.create_table(:auto_increments, id: :bigint, auto_increment: true, force: true)
+    assert_auto_incremented
+  end
+
+  private
+    def assert_auto_incremented
+      record1 = AutoIncrement.create!
+      assert_not_nil record1.id
+
+      record1.destroy
+
+      record2 = AutoIncrement.create!
+      assert_not_nil record2.id
+      assert_operator record2.id, :>, record1.id
+    end
 end
 
 class PrimaryKeyAnyTypeTest < ActiveRecord::TestCase