count method should not take options if it is operated on has_many association which has finder_sql or counter_sql

[#2395 state:resolved]

Signed-off-by: José Valim <jose.valim@gmail.com>
This commit is contained in:
Neeraj Singh
2010-07-01 21:41:41 -04:00
committed by José Valim
parent f4fbc2c1f9
commit edb5401039
2 changed files with 31 additions and 2 deletions

View File

@@ -183,10 +183,13 @@ module ActiveRecord
# descendant's +construct_sql+ method will have set :counter_sql automatically.
# Otherwise, construct options and pass them with scope to the target class's +count+.
def count(column_name = nil, options = {})
if @reflection.options[:counter_sql]
column_name, options = nil, column_name if column_name.is_a?(Hash)
if @reflection.options[:counter_sql] && !options.blank?
raise ArgumentError, "If finder_sql/counter_sql is used then options cannot be passed"
elsif @reflection.options[:counter_sql]
@reflection.klass.count_by_sql(@counter_sql)
else
column_name, options = nil, column_name if column_name.is_a?(Hash)
if @reflection.options[:uniq]
# This is needed because 'SELECT count(DISTINCT *)..' is not valid SQL.

View File

@@ -11,6 +11,32 @@ require 'models/comment'
require 'models/person'
require 'models/reader'
require 'models/tagging'
require 'models/invoice'
require 'models/line_item'
class HasManyAssociationsTestForCountWithFinderSql < ActiveRecord::TestCase
class Invoice < ActiveRecord::Base
has_many :custom_line_items, :class_name => 'LineItem', :finder_sql => "SELECT line_items.* from line_items"
end
def test_should_fail
assert_raise(ArgumentError) do
Invoice.create.custom_line_items.count(:conditions => {:amount => 0})
end
end
end
class HasManyAssociationsTestForCountWithCountSql < ActiveRecord::TestCase
class Invoice < ActiveRecord::Base
has_many :custom_line_items, :class_name => 'LineItem', :counter_sql => "SELECT COUNT(*) line_items.* from line_items"
end
def test_should_fail
assert_raise(ArgumentError) do
Invoice.create.custom_line_items.count(:conditions => {:amount => 0})
end
end
end
class HasManyAssociationsTest < ActiveRecord::TestCase
fixtures :accounts, :categories, :companies, :developers, :projects,