aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/test/cases/adapters/sqlite3/annotate_test.rb
blob: 6567a5eca3c58d79aee903e1e13eac470ec34e3c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# frozen_string_literal: true

require "cases/helper"
require "models/post"

class SQLite3AnnotateTest < ActiveRecord::SQLite3TestCase
  fixtures :posts

  def test_annotate_wraps_content_in_an_inline_comment
    assert_sql(%r{\ASELECT "posts"\."id" FROM "posts" /\* foo \*/}) do
      posts = Post.select(:id).annotate("foo")
      assert posts.first
    end
  end

  def test_annotate_is_sanitized
    assert_sql(%r{\ASELECT "posts"\."id" FROM "posts" /\* foo \*/}) do
      posts = Post.select(:id).annotate("*/foo/*")
      assert posts.first
    end

    assert_sql(%r{\ASELECT "posts"\."id" FROM "posts" /\* foo \*/}) do
      posts = Post.select(:id).annotate("**//foo//**")
      assert posts.first
    end

    assert_sql(%r{\ASELECT "posts"\."id" FROM "posts" /\* foo \*/ /\* bar \*/}) do
      posts = Post.select(:id).annotate("*/foo/*").annotate("*/bar")
      assert posts.first
    end

    assert_sql(%r{\ASELECT "posts"\."id" FROM "posts" /\* \+ MAX_EXECUTION_TIME\(1\) \*/}) do
      posts = Post.select(:id).annotate("+ MAX_EXECUTION_TIME(1)")
      assert posts.first
    end
  end
end