require "cases/helper" require 'models/company' require 'models/topic' require 'models/edge' Company.has_many :accounts class NumericData < ActiveRecord::Base self.table_name = 'numeric_data' end class CalculationsTest < ActiveRecord::TestCase fixtures :companies, :accounts, :topics def test_should_sum_field assert_equal 318, Account.sum(:credit_limit) end def test_should_average_field value = Account.average(:credit_limit) assert_kind_of BigDecimal, value assert_equal BigDecimal.new('53.0'), value end def test_should_return_nil_as_average assert_nil NumericData.average(:bank_balance) end def test_type_cast_calculated_value_should_convert_db_averages_of_fixnum_class_to_decimal assert_equal 0, NumericData.send(:type_cast_calculated_value, 0, nil, 'avg') assert_equal 53.0, NumericData.send(:type_cast_calculated_value, 53, nil, 'avg') end def test_should_get_maximum_of_field assert_equal 60, Account.maximum(:credit_limit) end def test_should_get_maximum_of_field_with_include assert_equal 50, Account.maximum(:credit_limit, :include => :firm, :conditions => "companies.name != 'Summit'") end def test_should_get_maximum_of_field_with_scoped_include Account.with_scope :find => { :include => :firm, :conditions => "companies.name != 'Summit'" } do assert_equal 50, Account.maximum(:credit_limit) end end def test_should_get_minimum_of_field assert_equal 50, Account.minimum(:credit_limit) end def test_should_group_by_field c = Account.sum(:credit_limit, :group => :firm_id) [1,6,2].each { |firm_id| assert c.keys.include?(firm_id) } end def test_should_group_by_summed_field c = Account.sum(:credit_limit, :group => :firm_id) assert_equal 50, c[1] assert_equal 105, c[6] assert_equal 60, c[2] end def test_should_order_by_grouped_field c = Account.sum(:credit_limit, :group => :firm_id, :order => "firm_id") assert_equal [1, 2, 6, 9], c.keys.compact end def test_should_order_by_calculation c = Account.sum(:credit_limit, :group => :firm_id, :order => "sum_credit_limit desc, firm_id") assert_equal [105, 60, 53, 50, 50], c.keys.collect { |k| c[k] } assert_equal [6, 2, 9, 1], c.keys.compact end def test_should_limit_calculation c = Account.sum(:credit_limit, :conditions => "firm_id IS NOT NULL", :group => :firm_id, :order => "firm_id", :limit => 2) assert_equal [1, 2], c.keys.compact end def test_should_limit_calculation_with_offset c = Account.sum(:credit_limit, :conditions => "firm_id IS NOT NULL", :group => :firm_id, :order => "firm_id", :limit => 2, :offset => 1) assert_equal [2, 6], c.keys.compact end def test_should_group_by_summed_field_having_condition c = Account.sum(:credit_limit, :group => :firm_id, :having => 'sum(credit_limit) > 50') assert_nil c[1] assert_equal 105, c[6] assert_equal 60, c[2] end def test_should_group_by_summed_field_having_sanitized_condition c = Account.sum(:credit_limit, :group => :firm_id, :having => ['sum(credit_limit) > ?', 50]) assert_nil c[1] assert_equal 105, c[6] assert_equal 60, c[2] end def test_should_group_by_summed_association c = Account.sum(:credit_limit, :group => :firm) assert_equal 50, c[companies(:first_firm)] assert_equal 105, c[companies(:rails_core)] assert_equal 60, c[companies(:first_client)] end def test_should_sum_field_with_conditions assert_equal 105, Account.sum(:credit_limit, :conditions => 'firm_id = 6') end def test_should_return_zero_if_sum_conditions_return_nothing assert_equal 0, Account.sum(:credit_limit, :conditions => '1 = 2') assert_equal 0, companies(:rails_core).companies.sum(:id, :conditions => '1 = 2') end def test_sum_should_return_valid_values_for_decimals NumericData.create(:bank_balance => 19.83) assert_equal 19.83, NumericData.sum(:bank_balance) end def test_should_group_by_summed_field_with_conditions c = Account.sum(:credit_limit, :conditions => 'firm_id > 1', :group => :firm_id) assert_nil c[1] assert_equal 105, c[6] assert_equal 60, c[2] end def test_should_group_by_summed_field_with_conditions_and_having c = Account.sum(:credit_limit, :conditions => 'firm_id > 1', :group => :firm_id, :having => 'sum(credit_limit) > 60') assert_nil c[1] assert_equal 105, c[6] assert_nil c[2] end def test_should_group_by_fields_with_table_alias c = Account.sum(:credit_limit, :group => 'accounts.firm_id') assert_equal 50, c[1] assert_equal 105, c[6] assert_equal 60, c[2] end def test_should_calculate_with_invalid_field assert_equal 6, Account.calculate(:count, '*') assert_equal 6, Account.calculate(:count, :all) end def test_should_calculate_grouped_with_invalid_field c = Account.count(:all, :group => 'accounts.firm_id') assert_equal 1, c[1] assert_equal 2, c[6] assert_equal 1, c[2] end def test_should_calculate_grouped_association_with_invalid_field c = Account.count(:all, :group => :firm) assert_equal 1, c[companies(:first_firm)] assert_equal 2, c[companies(:rails_core)] assert_equal 1, c[companies(:first_client)] end def test_should_group_by_association_with_non_numeric_foreign_key ActiveRecord::Base.connection.expects(:select_all).returns([{"count_all" => 1, "firm_id" => "ABC"}]) firm = mock() firm.expects(:id).returns("ABC") firm.expects(:class).returns(Firm) Company.expects(:find).with(["ABC"]).returns([firm]) column = mock() column.expects(:name).at_least_once.returns(:firm_id) column.expects(:type_cast).with("ABC").returns("ABC") Account.expects(:columns).at_least_once.returns([column]) c = Account.count(:all, :group => :firm) first_key = c.keys.first assert_equal Firm, first_key.class assert_equal 1, c[first_key] end def test_should_calculate_grouped_association_with_foreign_key_option Account.belongs_to :another_firm, :class_name => 'Firm', :foreign_key => 'firm_id' c = Account.count(:all, :group => :another_firm) assert_equal 1, c[companies(:first_firm)] assert_equal 2, c[companies(:rails_core)] assert_equal 1, c[companies(:first_client)] end def test_should_not_modify_options_when_using_includes options = {:conditions => 'companies.id > 1', :include => :firm} options_copy = options.dup Account.count(:all, options) assert_equal options_copy, options end def test_should_calculate_grouped_by_function c = Company.count(:all, :group => "UPPER(#{QUOTED_TYPE})") assert_equal 2, c[nil] assert_equal 1, c['DEPENDENTFIRM'] assert_equal 3, c['CLIENT'] assert_equal 2, c['FIRM'] end def test_should_calculate_grouped_by_function_with_table_alias c = Company.count(:all, :group => "UPPER(companies.#{QUOTED_TYPE})") assert_equal 2, c[nil] assert_equal 1, c['DEPENDENTFIRM'] assert_equal 3, c['CLIENT'] assert_equal 2, c['FIRM'] end def test_should_not_overshadow_enumerable_sum assert_equal 6, [1, 2, 3].sum(&:abs) end def test_should_sum_scoped_field assert_equal 15, companies(:rails_core).companies.sum(:id) end def test_should_sum_scoped_field_with_conditions assert_equal 8, companies(:rails_core).companies.sum(:id, :conditions => 'id > 7') end def test_should_group_by_scoped_field c = companies(:rails_core).companies.sum(:id, :group => :name) assert_equal 7, c['Leetsoft'] assert_equal 8, c['Jadedpixel'] end def test_should_group_by_summed_field_with_conditions_and_having c = companies(:rails_core).companies.sum(:id, :group => :name, :having => 'sum(id) > 7') assert_nil c['Leetsoft'] assert_equal 8, c['Jadedpixel'] end def test_should_reject_invalid_options assert_nothing_raised do [:count, :sum].each do |func| # empty options are valid Company.send(:validate_calculation_options, func) # these options are valid for all calculations [:select, :conditions, :joins, :order, :group, :having, :distinct].each do |opt| Company.send(:validate_calculation_options, func, opt => true) end end # :include is only valid on :count Company.send(:validate_calculation_options, :count, :include => true) end assert_raise(ArgumentError) { Company.send(:validate_calculation_options, :sum, :foo => :bar) } assert_raise(ArgumentError) { Company.send(:validate_calculation_options, :count, :foo => :bar) } end def test_should_count_selected_field_with_include assert_equal 6, Account.count(:distinct => true, :include => :firm) assert_equal 4, Account.count(:distinct => true, :include => :firm, :select => :credit_limit) end def test_should_count_scoped_select Account.update_all("credit_limit = NULL") assert_equal 0, Account.scoped(:select => "credit_limit").count end def test_should_count_scoped_select_with_options Account.update_all("credit_limit = NULL") Account.last.update_attribute('credit_limit', 49) Account.first.update_attribute('credit_limit', 51) assert_equal 1, Account.scoped(:select => "credit_limit").count(:conditions => ['credit_limit >= 50']) end def test_should_count_manual_select_with_include assert_equal 6, Account.count(:select => "DISTINCT accounts.id", :include => :firm) end def test_count_with_column_parameter assert_equal 5, Account.count(:firm_id) end def test_count_with_column_and_options_parameter assert_equal 2, Account.count(:firm_id, :conditions => "credit_limit = 50") end def test_count_with_no_parameters_isnt_deprecated assert_not_deprecated { Account.count } end def test_count_with_too_many_parameters_raises assert_raise(ArgumentError) { Account.count(1, 2, 3) } end def test_should_sum_expression assert_equal '636', Account.sum("2 * credit_limit") end def test_count_with_from_option assert_equal Company.count(:all), Company.count(:all, :from => 'companies') assert_equal Account.count(:all, :conditions => "credit_limit = 50"), Account.count(:all, :from => 'accounts', :conditions => "credit_limit = 50") assert_equal Company.count(:type, :conditions => {:type => "Firm"}), Company.count(:type, :conditions => {:type => "Firm"}, :from => 'companies') end def test_sum_with_from_option assert_equal Account.sum(:credit_limit), Account.sum(:credit_limit, :from => 'accounts') assert_equal Account.sum(:credit_limit, :conditions => "credit_limit > 50"), Account.sum(:credit_limit, :from => 'accounts', :conditions => "credit_limit > 50") end def test_average_with_from_option assert_equal Account.average(:credit_limit), Account.average(:credit_limit, :from => 'accounts') assert_equal Account.average(:credit_limit, :conditions => "credit_limit > 50"), Account.average(:credit_limit, :from => 'accounts', :conditions => "credit_limit > 50") end def test_minimum_with_from_option assert_equal Account.minimum(:credit_limit), Account.minimum(:credit_limit, :from => 'accounts') assert_equal Account.minimum(:credit_limit, :conditions => "credit_limit > 50"), Account.minimum(:credit_limit, :from => 'accounts', :conditions => "credit_limit > 50") end def test_maximum_with_from_option assert_equal Account.maximum(:credit_limit), Account.maximum(:credit_limit, :from => 'accounts') assert_equal Account.maximum(:credit_limit, :conditions => "credit_limit > 50"), Account.maximum(:credit_limit, :from => 'accounts', :conditions => "credit_limit > 50") end def test_from_option_with_specified_index if Edge.connection.adapter_name == 'MySQL' assert_equal Edge.count(:all), Edge.count(:all, :from => 'edges USE INDEX(unique_edge_index)') assert_equal Edge.count(:all, :conditions => 'sink_id < 5'), Edge.count(:all, :from => 'edges USE INDEX(unique_edge_index)', :conditions => 'sink_id < 5') end end def test_from_option_with_table_different_than_class assert_equal Account.count(:all), Company.count(:all, :from => 'accounts') end end