Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix unscoping default_scope in STI associations #29834

Merged
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
Expand Up @@ -114,8 +114,18 @@ def reflection_scope
@reflection_scope ||= reflection.scope_for(klass)
end

def klass_scope
current_scope = klass.current_scope

if current_scope && current_scope.empty_scope?
klass.unscoped
else
klass.default_scoped
end
end

def build_scope
scope = klass.default_scoped
scope = klass_scope

if reflection.type
scope.where!(reflection.type => model.base_class.sti_name)
Expand Down
6 changes: 4 additions & 2 deletions activerecord/lib/active_record/reflection.rb
Expand Up @@ -219,8 +219,10 @@ def join_scopes(table, predicate_builder) # :nodoc:
end

def klass_join_scope(table, predicate_builder) # :nodoc:
if klass.current_scope && klass.current_scope.values.empty?
klass.unscoped
current_scope = klass.current_scope

if current_scope && current_scope.empty_scope?
build_scope(table, predicate_builder)
else
klass.default_scoped(build_scope(table, predicate_builder))
end
Expand Down
4 changes: 4 additions & 0 deletions activerecord/lib/active_record/relation.rb
Expand Up @@ -642,6 +642,10 @@ def inspect
"#<#{self.class.name} [#{entries.join(', ')}]>"
end

def empty_scope? # :nodoc:
@values == klass.unscoped.values
end

protected

def load_records(records)
Expand Down
18 changes: 18 additions & 0 deletions activerecord/test/cases/scoping/default_scoping_test.rb
Expand Up @@ -392,6 +392,24 @@ def test_unscoped_with_joins_should_not_have_default_scope
Comment.joins(:post).to_a
end

def test_sti_association_with_unscoped_not_affected_by_default_scope
post = posts(:thinking)
comments = [comments(:does_it_hurt)]

post.special_comments.update_all(deleted_at: Time.now)

assert_raises(ActiveRecord::RecordNotFound) { Post.joins(:special_comments).find(post.id) }
assert_equal [], post.special_comments

SpecialComment.unscoped do
assert_equal post, Post.joins(:special_comments).find(post.id)
assert_equal comments, Post.joins(:special_comments).find(post.id).special_comments
assert_equal comments, Post.eager_load(:special_comments).find(post.id).special_comments
assert_equal comments, Post.includes(:special_comments).find(post.id).special_comments
assert_equal comments, Post.preload(:special_comments).find(post.id).special_comments
end
end

def test_default_scope_select_ignored_by_aggregations
assert_equal DeveloperWithSelect.all.to_a.count, DeveloperWithSelect.count
end
Expand Down
1 change: 1 addition & 0 deletions activerecord/test/models/comment.rb
Expand Up @@ -54,6 +54,7 @@ def to_s
end

class SpecialComment < Comment
default_scope { where(deleted_at: nil) }
end

class SubSpecialComment < SpecialComment
Expand Down
1 change: 1 addition & 0 deletions activerecord/test/schema/schema.rb
Expand Up @@ -189,6 +189,7 @@
t.string :resource_id
t.string :resource_type
t.integer :developer_id
t.datetime :deleted_at
end

create_table :companies, force: true do |t|
Expand Down