ActiveRecord bug when joining has_many association and finding multiple records

activerecord-joins-find-multiple-bug.rb

unless File.exist?('Gemfile')
  File.write('Gemfile', <<-GEMFILE)
    source 'https://rubygems.org'
    gem 'rails', github: 'rails/rails'
    gem 'arel', github: 'rails/arel'
    gem 'sqlite3'
  GEMFILE

  system 'bundle'
end

require 'bundler'
Bundler.setup(:default)

require 'active_record'
require 'minitest/autorun'
require 'logger'

# This connection will do for database-independent bug reports.
ActiveRecord::Base.establish_connection(adapter: 'sqlite3', database: ':memory:')
ActiveRecord::Base.logger = Logger.new(STDOUT)

ActiveRecord::Schema.define do
  create_table :posts do |t|
  end

  create_table :comments do |t|
    t.integer :post_id
  end
end

class Post < ActiveRecord::Base
  has_many :comments
end

class Comment < ActiveRecord::Base
  belongs_to :post
end

class BugTest < Minitest::Test
  def test_association_stuff
    post1 = Post.create!
    post2 = Post.create!
    
    post1.comments << Comment.create!
    post1.comments << Comment.create!
    post2.comments << Comment.create!
    post2.comments << Comment.create!
    post2.comments << Comment.create!

    assert_equal 2, Post.joins(:comments).find(post1.id, post2.id).count
    assert_equal 2, Post.joins(:comments).where(id: [post1.id, post2.id]).count
  end
end