aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/test/cases/bind_parameter_test.rb
blob: 32726e8d8b9336c24f53621771d52ca138d68ee5 (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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
require 'cases/helper'
require 'models/topic'

module ActiveRecord
  class BindParameterTest < ActiveRecord::TestCase
    fixtures :topics

    class LogListener
      attr_accessor :calls

      def initialize
        @calls = []
      end

      def call(*args)
        calls << args
      end
    end

    def setup
      super
      @connection = ActiveRecord::Base.connection
      @listener   = LogListener.new
      @pk         = Topic.columns.find { |c| c.primary }
      ActiveSupport::Notifications.subscribe('sql.active_record', @listener)
    end

    def teardown
      ActiveSupport::Notifications.unsubscribe(@listener)
    end

    def test_binds_are_logged
      return skip_bind_parameter_test unless supports_statement_cache?

      sub   = @connection.substitute_at(@pk, 0)
      binds = [[@pk, 1]]
      sql   = "select * from topics where id = #{sub}"

      @connection.exec_query(sql, 'SQL', binds)

      message = @listener.calls.find { |args| args[4][:sql] == sql }
      assert_equal binds, message[4][:binds]
    end

    def test_find_one_uses_binds
      return skip_bind_parameter_test unless supports_statement_cache?

      Topic.find(1)
      binds = [[@pk, 1]]
      message = @listener.calls.find { |args| args[4][:binds] == binds }
      assert message, 'expected a message with binds'
    end

    def test_logs_bind_vars
      return skip_bind_parameter_test unless supports_statement_cache?

      pk = Topic.columns.find { |x| x.primary }

      payload = {
        :name  => 'SQL',
        :sql   => 'select * from topics where id = ?',
        :binds => [[pk, 10]]
      }
      event  = ActiveSupport::Notifications::Event.new(
        'foo',
        Time.now,
        Time.now,
        123,
        payload)

        logger = Class.new(ActiveRecord::LogSubscriber) {
          attr_reader :debugs
          def initialize
            super
            @debugs = []
          end

          def debug str
            @debugs << str
          end
        }.new

        logger.sql event
        assert_match([[pk.name, 10]].inspect, logger.debugs.first)
    end

    private

    def skip_bind_parameter_test
      skip('prepared statement caching is not supported')
    end

    def supports_statement_cache?
      @connection.supports_statement_cache?
    end
  end
end