2011-06-06 14:17:44 -04:00
|
|
|
require 'cases/helper'
|
2011-02-10 12:56:50 -05:00
|
|
|
require 'models/topic'
|
2015-01-05 18:25:19 -05:00
|
|
|
require 'models/author'
|
|
|
|
require 'models/post'
|
2011-02-10 12:56:50 -05:00
|
|
|
|
|
|
|
module ActiveRecord
|
|
|
|
class BindParameterTest < ActiveRecord::TestCase
|
2015-01-05 18:25:19 -05:00
|
|
|
fixtures :topics, :authors, :posts
|
2011-02-10 16:34:33 -05:00
|
|
|
|
2011-02-10 12:56:50 -05:00
|
|
|
class LogListener
|
|
|
|
attr_accessor :calls
|
|
|
|
|
|
|
|
def initialize
|
|
|
|
@calls = []
|
|
|
|
end
|
|
|
|
|
|
|
|
def call(*args)
|
|
|
|
calls << args
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def setup
|
|
|
|
super
|
|
|
|
@connection = ActiveRecord::Base.connection
|
2015-01-27 14:07:06 -05:00
|
|
|
@subscriber = LogListener.new
|
|
|
|
@pk = ConnectionAdapters::Column.new(Topic.primary_key, nil, Topic.type_for_attribute(Topic.primary_key))
|
2014-03-29 09:52:56 -04:00
|
|
|
@subscription = ActiveSupport::Notifications.subscribe('sql.active_record', @subscriber)
|
2011-02-10 12:56:50 -05:00
|
|
|
end
|
|
|
|
|
2014-03-14 00:35:58 -04:00
|
|
|
teardown do
|
2014-03-29 09:52:56 -04:00
|
|
|
ActiveSupport::Notifications.unsubscribe(@subscription)
|
2011-02-10 12:56:50 -05:00
|
|
|
end
|
|
|
|
|
2013-11-08 10:57:51 -05:00
|
|
|
if ActiveRecord::Base.connection.supports_statement_cache?
|
2015-01-05 18:25:19 -05:00
|
|
|
def test_bind_from_join_in_subquery
|
|
|
|
subquery = Author.joins(:thinking_posts).where(name: 'David')
|
|
|
|
scope = Author.from(subquery, 'authors').where(id: 1)
|
|
|
|
assert_equal 1, scope.count
|
|
|
|
end
|
|
|
|
|
2013-11-08 10:57:51 -05:00
|
|
|
def test_binds_are_logged
|
2014-11-17 18:04:40 -05:00
|
|
|
sub = @connection.substitute_at(@pk)
|
2015-01-27 15:42:02 -05:00
|
|
|
binds = [Relation::QueryAttribute.new("id", 1, Type::Value.new)]
|
2014-11-17 17:56:06 -05:00
|
|
|
sql = "select * from topics where id = #{sub.to_sql}"
|
2011-02-10 12:56:50 -05:00
|
|
|
|
2013-11-08 10:57:51 -05:00
|
|
|
@connection.exec_query(sql, 'SQL', binds)
|
2011-02-10 12:56:50 -05:00
|
|
|
|
2014-03-29 09:52:56 -04:00
|
|
|
message = @subscriber.calls.find { |args| args[4][:sql] == sql }
|
2013-11-08 10:57:51 -05:00
|
|
|
assert_equal binds, message[4][:binds]
|
|
|
|
end
|
2011-02-10 16:34:33 -05:00
|
|
|
|
2013-11-08 10:57:51 -05:00
|
|
|
def test_find_one_uses_binds
|
|
|
|
Topic.find(1)
|
2015-01-27 15:42:02 -05:00
|
|
|
message = @subscriber.calls.find { |args| args[4][:binds].any? { |attr| attr.value == 1 } }
|
2013-11-08 10:57:51 -05:00
|
|
|
assert message, 'expected a message with binds'
|
|
|
|
end
|
2011-02-10 16:34:33 -05:00
|
|
|
|
2015-01-27 15:42:02 -05:00
|
|
|
def test_logs_bind_vars_after_type_cast
|
2013-11-08 10:57:51 -05:00
|
|
|
payload = {
|
|
|
|
:name => 'SQL',
|
|
|
|
:sql => 'select * from topics where id = ?',
|
2015-01-27 15:42:02 -05:00
|
|
|
:binds => [Relation::QueryAttribute.new("id", "10", Type::Integer.new)]
|
2013-11-08 10:57:51 -05:00
|
|
|
}
|
|
|
|
event = ActiveSupport::Notifications::Event.new(
|
|
|
|
'foo',
|
|
|
|
Time.now,
|
|
|
|
Time.now,
|
|
|
|
123,
|
|
|
|
payload)
|
2011-02-10 16:34:33 -05:00
|
|
|
|
|
|
|
logger = Class.new(ActiveRecord::LogSubscriber) {
|
|
|
|
attr_reader :debugs
|
|
|
|
def initialize
|
|
|
|
super
|
|
|
|
@debugs = []
|
|
|
|
end
|
|
|
|
|
|
|
|
def debug str
|
|
|
|
@debugs << str
|
|
|
|
end
|
|
|
|
}.new
|
|
|
|
|
|
|
|
logger.sql event
|
2014-05-23 13:59:30 -04:00
|
|
|
assert_match([[@pk.name, 10]].inspect, logger.debugs.first)
|
2013-11-08 10:57:51 -05:00
|
|
|
end
|
2012-05-13 03:27:52 -04:00
|
|
|
end
|
2011-02-10 12:56:50 -05:00
|
|
|
end
|
|
|
|
end
|