7263da542b
Replaced by `#lease_connection` to better reflect what it does. `ActiveRecord::Base#connection` is deprecated in the same way but without a removal timeline nor a deprecation warning. Inside the Active Record test suite, we do remove `Base.connection` to ensure it's not used internally. Some callsites have been converted to use `with_connection`, some other have been more simply migrated to `lease_connection` and will serve as a list of callsites to convert for https://github.com/rails/rails/pull/50793
52 lines
1.7 KiB
Ruby
52 lines
1.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "cases/helper"
|
|
require "models/post"
|
|
|
|
class AnnotateTest < ActiveRecord::TestCase
|
|
fixtures :posts
|
|
|
|
def test_annotate_wraps_content_in_an_inline_comment
|
|
quoted_posts_id, quoted_posts = regexp_escape_table_name("posts.id"), regexp_escape_table_name("posts")
|
|
|
|
assert_queries_match(%r{SELECT #{quoted_posts_id} FROM #{quoted_posts} /\* foo \*/}i) do
|
|
posts = Post.select(:id).annotate("foo")
|
|
assert posts.first
|
|
end
|
|
end
|
|
|
|
def test_annotate_is_sanitized
|
|
quoted_posts_id, quoted_posts = regexp_escape_table_name("posts.id"), regexp_escape_table_name("posts")
|
|
|
|
assert_queries_match(%r{SELECT #{quoted_posts_id} FROM #{quoted_posts} /\* \* /foo/ \* \*/}i) do
|
|
posts = Post.select(:id).annotate("*/foo/*")
|
|
assert posts.first
|
|
end
|
|
|
|
assert_queries_match(%r{SELECT #{quoted_posts_id} FROM #{quoted_posts} /\* \*\* //foo// \*\* \*/}i) do
|
|
posts = Post.select(:id).annotate("**//foo//**")
|
|
assert posts.first
|
|
end
|
|
|
|
assert_queries_match(%r{SELECT #{quoted_posts_id} FROM #{quoted_posts} /\* \* \* //foo// \* \* \*/}i) do
|
|
posts = Post.select(:id).annotate("* *//foo//* *")
|
|
assert posts.first
|
|
end
|
|
|
|
assert_queries_match(%r{SELECT #{quoted_posts_id} FROM #{quoted_posts} /\* \* /foo/ \* \*/ /\* \* /bar \*/}i) do
|
|
posts = Post.select(:id).annotate("*/foo/*").annotate("*/bar")
|
|
assert posts.first
|
|
end
|
|
|
|
assert_queries_match(%r{SELECT #{quoted_posts_id} FROM #{quoted_posts} /\* \+ MAX_EXECUTION_TIME\(1\) \*/}i) do
|
|
posts = Post.select(:id).annotate("+ MAX_EXECUTION_TIME(1)")
|
|
assert posts.first
|
|
end
|
|
end
|
|
|
|
private
|
|
def regexp_escape_table_name(name)
|
|
Regexp.escape(Post.lease_connection.quote_table_name(name))
|
|
end
|
|
end
|