rails-sti-test.rb

begin
  require "bundler/inline"
rescue LoadError => e
  $stderr.puts "Bundler version 1.10 or later is required. Please update your Bundler"
  raise e
end

gemfile(true) do
  source "https://rubygems.org"
  # Activate the gem you are reporting the issue against.
  gem "activerecord", "4.2.7.1"
  gem "sqlite3"
end

require "active_record"
require "minitest/autorun"
require "logger"

# Ensure backward compatibility with Minitest 4
Minitest::Test = MiniTest::Unit::TestCase unless defined?(Minitest::Test)

# 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, force: true do |t|
    t.boolean :deleted
    t.string :type
    t.integer :user_id
  end

  create_table :users, force: true do |t|
  end
end

class Post < ActiveRecord::Base
end

class VideoPost < Post
end

class User < ActiveRecord::Base
  has_many :video_posts
end

class BugTest < Minitest::Test
  def setup
    Post.delete_all
    User.delete_all
  end

  def test_sti_query_directly
    video_post1 = VideoPost.new deleted: false
    video_post2 = VideoPost.new deleted: true
    user = User.create! video_posts: [video_post1, video_post2]

    assert_equal [video_post2], user.video_posts.where(deleted: true)
  end

  def test_query_through_sti_association_using_baseclass
    video_post1 = VideoPost.new deleted: false
    user1 = User.create! video_posts: [video_post1]
    video_post2 = VideoPost.new deleted: true
    user2 = User.create! video_posts: [video_post2]

    assert_equal [user2], User.includes(:video_posts).where(posts: {deleted: true})
  end

  def test_query_through_sti_association_using_subclass
    video_post1 = VideoPost.new deleted: false
    user1 = User.create! video_posts: [video_post1]
    video_post2 = VideoPost.new deleted: true
    user2 = User.create! video_posts: [video_post2]

    assert_equal [user2], User.includes(:video_posts).where(video_posts: {deleted: true})
  end
end