finder_test.rb 38.9 KB
Newer Older
1
require "cases/helper"
J
Jeremy Kemper 已提交
2
require 'models/author'
3
require 'models/categorization'
J
Jeremy Kemper 已提交
4 5 6 7 8 9 10
require 'models/comment'
require 'models/company'
require 'models/topic'
require 'models/reply'
require 'models/entrant'
require 'models/developer'
require 'models/post'
11
require 'models/customer'
12
require 'models/job'
13
require 'models/categorization'
D
David Heinemeier Hansson 已提交
14

15 16 17 18 19 20 21 22 23 24 25 26 27
class DynamicFinderMatchTest < ActiveRecord::TestCase
  def test_find_no_match
    assert_nil ActiveRecord::DynamicFinderMatch.match("not_a_finder")
  end

  def test_find_by
    match = ActiveRecord::DynamicFinderMatch.match("find_by_age_and_sex_and_location")
    assert_not_nil match
    assert match.finder?
    assert_equal :find_initial, match.finder
    assert_equal %w(age sex location), match.attribute_names
  end

28 29 30 31 32 33 34 35 36
  def find_by_bang
    match = ActiveRecord::DynamicFinderMatch.match("find_by_age_and_sex_and_location!")
    assert_not_nil match
    assert match.finder?
    assert match.bang?
    assert_equal :find_initial, match.finder
    assert_equal %w(age sex location), match.attribute_names
  end

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
  def test_find_all_by
    match = ActiveRecord::DynamicFinderMatch.match("find_all_by_age_and_sex_and_location")
    assert_not_nil match
    assert match.finder?
    assert_equal :find_every, match.finder
    assert_equal %w(age sex location), match.attribute_names
  end

  def test_find_or_initialize_by
    match = ActiveRecord::DynamicFinderMatch.match("find_or_initialize_by_age_and_sex_and_location")
    assert_not_nil match
    assert !match.finder?
    assert match.instantiator?
    assert_equal :find_initial, match.finder
    assert_equal :new, match.instantiator
    assert_equal %w(age sex location), match.attribute_names
  end

  def test_find_or_create_by
    match = ActiveRecord::DynamicFinderMatch.match("find_or_create_by_age_and_sex_and_location")
    assert_not_nil match
    assert !match.finder?
    assert match.instantiator?
    assert_equal :find_initial, match.finder
    assert_equal :create, match.instantiator
    assert_equal %w(age sex location), match.attribute_names
  end
end

66
class FinderTest < ActiveRecord::TestCase
67
  fixtures :companies, :topics, :entrants, :developers, :developers_projects, :posts, :comments, :accounts, :authors, :customers
D
David Heinemeier Hansson 已提交
68 69

  def test_find
70
    assert_equal(topics(:first).title, Topic.find(1).title)
D
David Heinemeier Hansson 已提交
71
  end
72

73 74 75 76 77
  # find should handle strings that come from URLs
  # (example: Category.find(params[:id]))
  def test_find_with_string
    assert_equal(Topic.find(1).title,Topic.find("1").title)
  end
78

D
David Heinemeier Hansson 已提交
79
  def test_exists
80 81 82 83 84 85
    assert Topic.exists?(1)
    assert Topic.exists?("1")
    assert Topic.exists?(:author_name => "David")
    assert Topic.exists?(:author_name => "Mary", :approved => true)
    assert Topic.exists?(["parent_id = ?", 1])
    assert !Topic.exists?(45)
86 87 88 89 90 91 92 93 94

    begin
      assert !Topic.exists?("foo")
    rescue ActiveRecord::StatementInvalid
      # PostgreSQL complains about string comparison with integer field
    rescue Exception
      flunk
    end

95
    assert_raise(NoMethodError) { Topic.exists?([1,2]) }
D
David Heinemeier Hansson 已提交
96
  end
97

98 99 100 101 102 103 104 105 106 107 108 109 110 111 112
  def test_exists_with_aggregate_having_three_mappings
    existing_address = customers(:david).address
    assert Customer.exists?(:address => existing_address)
  end

  def test_exists_with_aggregate_having_three_mappings_with_one_difference
    existing_address = customers(:david).address
    assert !Customer.exists?(:address =>
      Address.new(existing_address.street, existing_address.city, existing_address.country + "1"))
    assert !Customer.exists?(:address =>
      Address.new(existing_address.street, existing_address.city + "1", existing_address.country))
    assert !Customer.exists?(:address =>
      Address.new(existing_address.street + "1", existing_address.city, existing_address.country))
  end

D
David Heinemeier Hansson 已提交
113 114 115 116
  def test_find_by_array_of_one_id
    assert_kind_of(Array, Topic.find([ 1 ]))
    assert_equal(1, Topic.find([ 1 ]).length)
  end
117

D
David Heinemeier Hansson 已提交
118
  def test_find_by_ids
119 120 121 122 123
    assert_equal 2, Topic.find(1, 2).size
    assert_equal topics(:second).title, Topic.find([2]).first.title
  end

  def test_find_by_ids_with_limit_and_offset
124
    assert_equal 2, Entrant.find([1,3,2], :limit => 2).size
125
    assert_equal 1, Entrant.find([1,3,2], :limit => 3, :offset => 2).size
126 127 128 129 130 131 132

    # Also test an edge case: If you have 11 results, and you set a
    #   limit of 3 and offset of 9, then you should find that there
    #   will be only 2 results, regardless of the limit.
    devs = Developer.find :all
    last_devs = Developer.find devs.map(&:id), :limit => 3, :offset => 9
    assert_equal 2, last_devs.size
D
David Heinemeier Hansson 已提交
133 134
  end

135 136 137 138
  def test_find_an_empty_array
    assert_equal [], Topic.find([])
  end

D
David Heinemeier Hansson 已提交
139
  def test_find_by_ids_missing_one
140
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find(1, 2, 45) }
D
David Heinemeier Hansson 已提交
141
  end
142

D
David Heinemeier Hansson 已提交
143 144
  def test_find_all_with_limit
    entrants = Entrant.find(:all, :order => "id ASC", :limit => 2)
145

D
David Heinemeier Hansson 已提交
146
    assert_equal(2, entrants.size)
147
    assert_equal(entrants(:first).name, entrants.first.name)
D
David Heinemeier Hansson 已提交
148 149 150
  end

  def test_find_all_with_prepared_limit_and_offset
151
    entrants = Entrant.find(:all, :order => "id ASC", :limit => 2, :offset => 1)
D
David Heinemeier Hansson 已提交
152

153 154
    assert_equal(2, entrants.size)
    assert_equal(entrants(:second).name, entrants.first.name)
155

156 157 158
    entrants = Entrant.find(:all, :order => "id ASC", :limit => 2, :offset => 2)
    assert_equal(1, entrants.size)
    assert_equal(entrants(:third).name, entrants.first.name)
D
David Heinemeier Hansson 已提交
159
  end
160

161 162 163 164
  def test_find_all_with_limit_and_offset_and_multiple_orderings
    developers = Developer.find(:all, :order => "salary ASC, id DESC", :limit => 3, :offset => 1)
    assert_equal ["David", "fixture_10", "fixture_9"], developers.collect {|d| d.name}
  end
165

166 167 168 169 170
  def test_find_with_limit_and_condition
    developers = Developer.find(:all, :order => "id DESC", :conditions => "salary = 100000", :limit => 3, :offset =>7)
    assert_equal(1, developers.size)
    assert_equal("fixture_3", developers.first.name)
  end
D
David Heinemeier Hansson 已提交
171 172 173

  def test_find_with_entire_select_statement
    topics = Topic.find_by_sql "SELECT * FROM topics WHERE author_name = 'Mary'"
174

D
David Heinemeier Hansson 已提交
175
    assert_equal(1, topics.size)
176
    assert_equal(topics(:second).title, topics.first.title)
D
David Heinemeier Hansson 已提交
177
  end
178

D
David Heinemeier Hansson 已提交
179 180
  def test_find_with_prepared_select_statement
    topics = Topic.find_by_sql ["SELECT * FROM topics WHERE author_name = ?", "Mary"]
181

D
David Heinemeier Hansson 已提交
182
    assert_equal(1, topics.size)
183
    assert_equal(topics(:second).title, topics.first.title)
D
David Heinemeier Hansson 已提交
184
  end
185

186 187 188 189
  def test_find_by_sql_with_sti_on_joined_table
    accounts = Account.find_by_sql("SELECT * FROM accounts INNER JOIN companies ON companies.id = accounts.firm_id")
    assert_equal [Account], accounts.collect(&:class).uniq
  end
190

D
David Heinemeier Hansson 已提交
191 192
  def test_find_first
    first = Topic.find(:first, :conditions => "title = 'The First Topic'")
193
    assert_equal(topics(:first).title, first.title)
D
David Heinemeier Hansson 已提交
194
  end
195

D
David Heinemeier Hansson 已提交
196 197 198 199
  def test_find_first_failing
    first = Topic.find(:first, :conditions => "title = 'The First Topic!'")
    assert_nil(first)
  end
200 201 202 203 204 205 206 207
  
  def test_first
    assert_equal topics(:second).title, Topic.first(:conditions => "title = 'The Second Topic of the day'").title
  end
  
  def test_first_failing
    assert_nil Topic.first(:conditions => "title = 'The Second Topic of the day!'")
  end
D
David Heinemeier Hansson 已提交
208 209 210 211 212

  def test_unexisting_record_exception_handling
    assert_raises(ActiveRecord::RecordNotFound) {
      Topic.find(1).parent
    }
213

214
    Topic.find(2).topic
D
David Heinemeier Hansson 已提交
215
  end
216

217
  def test_find_only_some_columns
218
    topic = Topic.find(1, :select => "author_name")
219
    assert_raises(ActiveRecord::MissingAttributeError) {topic.title}
220 221
    assert_equal "David", topic.author_name
    assert !topic.attribute_present?("title")
222
    #assert !topic.respond_to?("title")
223 224
    assert topic.attribute_present?("author_name")
    assert topic.respond_to?("author_name")
225
  end
J
Jeremy Kemper 已提交
226

227 228 229 230 231
  def test_find_on_blank_conditions
    [nil, " ", [], {}].each do |blank|
      assert_nothing_raised { Topic.find(:first, :conditions => blank) }
    end
  end
J
Jeremy Kemper 已提交
232

233 234 235 236 237
  def test_find_on_blank_bind_conditions
    [ [""], ["",{}] ].each do |blank|
      assert_nothing_raised { Topic.find(:first, :conditions => blank) }
    end
  end
D
David Heinemeier Hansson 已提交
238

239
  def test_find_on_array_conditions
240 241
    assert Topic.find(1, :conditions => ["approved = ?", false])
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => ["approved = ?", true]) }
D
David Heinemeier Hansson 已提交
242
  end
243

244 245 246 247
  def test_find_on_hash_conditions
    assert Topic.find(1, :conditions => { :approved => false })
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { :approved => true }) }
  end
248

249 250 251 252 253
  def test_find_on_hash_conditions_with_explicit_table_name
    assert Topic.find(1, :conditions => { 'topics.approved' => false })
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { 'topics.approved' => true }) }
  end

254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270
  def test_find_on_hash_conditions_with_hashed_table_name
    assert Topic.find(1, :conditions => {:topics => { :approved => false }})
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => {:topics => { :approved => true }}) }
  end

  def test_find_with_hash_conditions_on_joined_table
    firms = Firm.all :joins => :account, :conditions => {:accounts => { :credit_limit => 50 }}
    assert_equal 1, firms.size
    assert_equal companies(:first_firm), firms.first
  end

  def test_find_with_hash_conditions_on_joined_table_and_with_range
    firms = DependentFirm.all :joins => :account, :conditions => {:name => 'RailsCore', :accounts => { :credit_limit => 55..60 }}
    assert_equal 1, firms.size
    assert_equal companies(:rails_core), firms.first
  end

271 272 273 274 275 276 277 278
  def test_find_on_hash_conditions_with_explicit_table_name_and_aggregate
    david = customers(:david)
    assert Customer.find(david.id, :conditions => { 'customers.name' => david.name, :address => david.address })
    assert_raises(ActiveRecord::RecordNotFound) {
      Customer.find(david.id, :conditions => { 'customers.name' => david.name + "1", :address => david.address })
    }
  end

279 280 281 282
  def test_find_on_association_proxy_conditions
    assert_equal [1, 2, 3, 5, 6, 7, 8, 9, 10], Comment.find_all_by_post_id(authors(:david).posts).map(&:id).sort
  end

283 284 285 286
  def test_find_on_hash_conditions_with_range
    assert_equal [1,2], Topic.find(:all, :conditions => { :id => 1..2 }).map(&:id).sort
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { :id => 2..3 }) }
  end
287

288 289 290 291
  def test_find_on_hash_conditions_with_multiple_ranges
    assert_equal [1,2,3], Comment.find(:all, :conditions => { :id => 1..3, :post_id => 1..2 }).map(&:id).sort
    assert_equal [1], Comment.find(:all, :conditions => { :id => 1..1, :post_id => 1..10 }).map(&:id).sort
  end
292

293 294 295 296 297 298
  def test_find_on_multiple_hash_conditions
    assert Topic.find(1, :conditions => { :author_name => "David", :title => "The First Topic", :replies_count => 1, :approved => false })
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { :author_name => "David", :title => "The First Topic", :replies_count => 1, :approved => true }) }
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { :author_name => "David", :title => "HHC", :replies_count => 1, :approved => false }) }
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { :author_name => "David", :title => "The First Topic", :replies_count => 1, :approved => true }) }
  end
299

300 301 302 303 304 305 306 307

  def test_condition_interpolation
    assert_kind_of Firm, Company.find(:first, :conditions => ["name = '%s'", "37signals"])
    assert_nil Company.find(:first, :conditions => ["name = '%s'", "37signals!"])
    assert_nil Company.find(:first, :conditions => ["name = '%s'", "37signals!' OR 1=1"])
    assert_kind_of Time, Topic.find(:first, :conditions => ["id = %d", 1]).written_on
  end

308
  def test_condition_array_interpolation
D
David Heinemeier Hansson 已提交
309 310 311 312 313
    assert_kind_of Firm, Company.find(:first, :conditions => ["name = '%s'", "37signals"])
    assert_nil Company.find(:first, :conditions => ["name = '%s'", "37signals!"])
    assert_nil Company.find(:first, :conditions => ["name = '%s'", "37signals!' OR 1=1"])
    assert_kind_of Time, Topic.find(:first, :conditions => ["id = %d", 1]).written_on
  end
314

315 316 317 318 319
  def test_condition_hash_interpolation
    assert_kind_of Firm, Company.find(:first, :conditions => { :name => "37signals"})
    assert_nil Company.find(:first, :conditions => { :name => "37signals!"})
    assert_kind_of Time, Topic.find(:first, :conditions => {:id => 1}).written_on
  end
320

321 322 323 324 325
  def test_hash_condition_find_malformed
    assert_raises(ActiveRecord::StatementInvalid) {
      Company.find(:first, :conditions => { :id => 2, :dhh => true })
    }
  end
326

327 328
  def test_hash_condition_find_with_escaped_characters
    Company.create("name" => "Ain't noth'n like' \#stuff")
329 330 331 332 333 334 335 336 337 338 339 340 341
    assert Company.find(:first, :conditions => { :name => "Ain't noth'n like' \#stuff" })
  end

  def test_hash_condition_find_with_array
    p1, p2 = Post.find(:all, :limit => 2, :order => 'id asc')
    assert_equal [p1, p2], Post.find(:all, :conditions => { :id => [p1, p2] }, :order => 'id asc')
    assert_equal [p1, p2], Post.find(:all, :conditions => { :id => [p1, p2.id] }, :order => 'id asc')
  end

  def test_hash_condition_find_with_nil
    topic = Topic.find(:first, :conditions => { :last_read => nil } )
    assert_not_nil topic
    assert_nil topic.last_read
342
  end
D
David Heinemeier Hansson 已提交
343

344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385
  def test_hash_condition_find_with_aggregate_having_one_mapping
    balance = customers(:david).balance
    assert_kind_of Money, balance
    found_customer = Customer.find(:first, :conditions => {:balance => balance})
    assert_equal customers(:david), found_customer
  end

  def test_hash_condition_find_with_aggregate_attribute_having_same_name_as_field_and_key_value_being_aggregate
    gps_location = customers(:david).gps_location
    assert_kind_of GpsLocation, gps_location
    found_customer = Customer.find(:first, :conditions => {:gps_location => gps_location})
    assert_equal customers(:david), found_customer
  end

  def test_hash_condition_find_with_aggregate_having_one_mapping_and_key_value_being_attribute_value
    balance = customers(:david).balance
    assert_kind_of Money, balance
    found_customer = Customer.find(:first, :conditions => {:balance => balance.amount})
    assert_equal customers(:david), found_customer
  end

  def test_hash_condition_find_with_aggregate_attribute_having_same_name_as_field_and_key_value_being_attribute_value
    gps_location = customers(:david).gps_location
    assert_kind_of GpsLocation, gps_location
    found_customer = Customer.find(:first, :conditions => {:gps_location => gps_location.gps_location})
    assert_equal customers(:david), found_customer
  end

  def test_hash_condition_find_with_aggregate_having_three_mappings
    address = customers(:david).address
    assert_kind_of Address, address
    found_customer = Customer.find(:first, :conditions => {:address => address})
    assert_equal customers(:david), found_customer
  end

  def test_hash_condition_find_with_one_condition_being_aggregate_and_another_not
    address = customers(:david).address
    assert_kind_of Address, address
    found_customer = Customer.find(:first, :conditions => {:address => address, :name => customers(:david).name})
    assert_equal customers(:david), found_customer
  end

D
David Heinemeier Hansson 已提交
386 387 388 389 390 391 392 393 394 395 396 397
  def test_bind_variables
    assert_kind_of Firm, Company.find(:first, :conditions => ["name = ?", "37signals"])
    assert_nil Company.find(:first, :conditions => ["name = ?", "37signals!"])
    assert_nil Company.find(:first, :conditions => ["name = ?", "37signals!' OR 1=1"])
    assert_kind_of Time, Topic.find(:first, :conditions => ["id = ?", 1]).written_on
    assert_raises(ActiveRecord::PreparedStatementInvalid) {
      Company.find(:first, :conditions => ["id=? AND name = ?", 2])
    }
    assert_raises(ActiveRecord::PreparedStatementInvalid) {
	   Company.find(:first, :conditions => ["id=?", 2, 3, 4])
    }
  end
398

D
David Heinemeier Hansson 已提交
399 400 401 402 403 404 405 406 407 408 409 410 411
  def test_bind_variables_with_quotes
    Company.create("name" => "37signals' go'es agains")
    assert Company.find(:first, :conditions => ["name = ?", "37signals' go'es agains"])
  end

  def test_named_bind_variables_with_quotes
    Company.create("name" => "37signals' go'es agains")
    assert Company.find(:first, :conditions => ["name = :name", {:name => "37signals' go'es agains"}])
  end

  def test_bind_arity
    assert_nothing_raised                                 { bind '' }
    assert_raises(ActiveRecord::PreparedStatementInvalid) { bind '', 1 }
412

D
David Heinemeier Hansson 已提交
413 414 415 416
    assert_raises(ActiveRecord::PreparedStatementInvalid) { bind '?' }
    assert_nothing_raised                                 { bind '?', 1 }
    assert_raises(ActiveRecord::PreparedStatementInvalid) { bind '?', 1, 1  }
  end
417

D
David Heinemeier Hansson 已提交
418 419 420
  def test_named_bind_variables
    assert_equal '1', bind(':a', :a => 1) # ' ruby-mode
    assert_equal '1 1', bind(':a :a', :a => 1)  # ' ruby-mode
421 422
    
    assert_nothing_raised { bind("'+00:00'", :foo => "bar") }
423

D
David Heinemeier Hansson 已提交
424 425 426 427 428 429
    assert_kind_of Firm, Company.find(:first, :conditions => ["name = :name", { :name => "37signals" }])
    assert_nil Company.find(:first, :conditions => ["name = :name", { :name => "37signals!" }])
    assert_nil Company.find(:first, :conditions => ["name = :name", { :name => "37signals!' OR 1=1" }])
    assert_kind_of Time, Topic.find(:first, :conditions => ["id = :id", { :id => 1 }]).written_on
  end

430
  def test_bind_enumerable
431 432
    quoted_abc = %(#{ActiveRecord::Base.connection.quote('a')},#{ActiveRecord::Base.connection.quote('b')},#{ActiveRecord::Base.connection.quote('c')})

D
David Heinemeier Hansson 已提交
433
    assert_equal '1,2,3', bind('?', [1, 2, 3])
434
    assert_equal quoted_abc, bind('?', %w(a b c))
D
David Heinemeier Hansson 已提交
435 436

    assert_equal '1,2,3', bind(':a', :a => [1, 2, 3])
437
    assert_equal quoted_abc, bind(':a', :a => %w(a b c)) # '
438 439 440

    require 'set'
    assert_equal '1,2,3', bind('?', Set.new([1, 2, 3]))
441
    assert_equal quoted_abc, bind('?', Set.new(%w(a b c)))
442 443

    assert_equal '1,2,3', bind(':a', :a => Set.new([1, 2, 3]))
444
    assert_equal quoted_abc, bind(':a', :a => Set.new(%w(a b c))) # '
D
David Heinemeier Hansson 已提交
445 446
  end

447 448 449 450 451 452 453
  def test_bind_empty_enumerable
    quoted_nil = ActiveRecord::Base.connection.quote(nil)
    assert_equal quoted_nil, bind('?', [])
    assert_equal " in (#{quoted_nil})", bind(' in (?)', [])
    assert_equal "foo in (#{quoted_nil})", bind('foo in (?)', [])
  end

454
  def test_bind_string
455
    assert_equal ActiveRecord::Base.connection.quote(''), bind('?', '')
456 457
  end

458 459 460 461 462 463 464 465
  def test_bind_record
    o = Struct.new(:quoted_id).new(1)
    assert_equal '1', bind('?', o)

    os = [o] * 3
    assert_equal '1,1,1', bind('?', os)
  end

466 467 468 469 470 471
  def test_named_bind_with_postgresql_type_casts
    l = Proc.new { bind(":a::integer '2009-01-01'::date", :a => '10') }
    assert_nothing_raised(&l)
    assert_equal "#{ActiveRecord::Base.quote_value('10')}::integer '2009-01-01'::date", l.call
  end

D
David Heinemeier Hansson 已提交
472
  def test_string_sanitation
473 474
    assert_not_equal "#{ActiveRecord::Base.connection.quoted_string_prefix}'something ' 1=1'", ActiveRecord::Base.sanitize("something ' 1=1")
    assert_equal "#{ActiveRecord::Base.connection.quoted_string_prefix}'something; select table'", ActiveRecord::Base.sanitize("something; select table")
D
David Heinemeier Hansson 已提交
475 476 477
  end

  def test_count
478 479 480
    assert_equal(0, Entrant.count(:conditions => "id > 3"))
    assert_equal(1, Entrant.count(:conditions => ["id > ?", 2]))
    assert_equal(2, Entrant.count(:conditions => ["id > ?", 1]))
D
David Heinemeier Hansson 已提交
481 482 483 484 485 486 487 488 489
  end

  def test_count_by_sql
    assert_equal(0, Entrant.count_by_sql("SELECT COUNT(*) FROM entrants WHERE id > 3"))
    assert_equal(1, Entrant.count_by_sql(["SELECT COUNT(*) FROM entrants WHERE id > ?", 2]))
    assert_equal(2, Entrant.count_by_sql(["SELECT COUNT(*) FROM entrants WHERE id > ?", 1]))
  end

  def test_find_by_one_attribute
490
    assert_equal topics(:first), Topic.find_by_title("The First Topic")
D
David Heinemeier Hansson 已提交
491 492
    assert_nil Topic.find_by_title("The First Topic!")
  end
J
Jeremy Kemper 已提交
493

494 495 496 497 498
  def test_find_by_one_attribute_bang
    assert_equal topics(:first), Topic.find_by_title!("The First Topic")
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find_by_title!("The First Topic!") }
  end

499 500
  def test_find_by_one_attribute_caches_dynamic_finder
    # ensure this test can run independently of order
501 502
    class << Topic; self; end.send(:remove_method, :find_by_title) if Topic.public_methods.any? { |m| m.to_s == 'find_by_title' }
    assert !Topic.public_methods.any? { |m| m.to_s == 'find_by_title' }
503
    t = Topic.find_by_title("The First Topic")
504
    assert Topic.public_methods.any? { |m| m.to_s == 'find_by_title' }
505 506 507 508
  end

  def test_dynamic_finder_returns_same_results_after_caching
    # ensure this test can run independently of order
509
    class << Topic; self; end.send(:remove_method, :find_by_title) if Topic.public_method_defined?(:find_by_title)
510 511 512
    t = Topic.find_by_title("The First Topic")
    assert_equal t, Topic.find_by_title("The First Topic") # find_by_title has been cached
  end
D
David Heinemeier Hansson 已提交
513

514
  def test_find_by_one_attribute_with_order_option
515
    assert_equal accounts(:signals37), Account.find_by_credit_limit(50, :order => 'id')
516 517 518 519 520 521 522
    assert_equal accounts(:rails_core_account), Account.find_by_credit_limit(50, :order => 'id DESC')
  end

  def test_find_by_one_attribute_with_conditions
    assert_equal accounts(:rails_core_account), Account.find_by_credit_limit(50, :conditions => ['firm_id = ?', 6])
  end

523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556
  def test_find_by_one_attribute_that_is_an_aggregate
    address = customers(:david).address
    assert_kind_of Address, address
    found_customer = Customer.find_by_address(address)
    assert_equal customers(:david), found_customer
  end

  def test_find_by_one_attribute_that_is_an_aggregate_with_one_attribute_difference
    address = customers(:david).address
    assert_kind_of Address, address
    missing_address = Address.new(address.street, address.city, address.country + "1")
    assert_nil Customer.find_by_address(missing_address)
    missing_address = Address.new(address.street, address.city + "1", address.country)
    assert_nil Customer.find_by_address(missing_address)
    missing_address = Address.new(address.street + "1", address.city, address.country)
    assert_nil Customer.find_by_address(missing_address)
  end

  def test_find_by_two_attributes_that_are_both_aggregates
    balance = customers(:david).balance
    address = customers(:david).address
    assert_kind_of Money, balance
    assert_kind_of Address, address
    found_customer = Customer.find_by_balance_and_address(balance, address)
    assert_equal customers(:david), found_customer
  end

  def test_find_by_two_attributes_with_one_being_an_aggregate
    balance = customers(:david).balance
    assert_kind_of Money, balance
    found_customer = Customer.find_by_balance_and_name(balance, customers(:david).name)
    assert_equal customers(:david), found_customer
  end

557 558
  def test_dynamic_finder_on_one_attribute_with_conditions_caches_method
    # ensure this test can run independently of order
559 560
    class << Account; self; end.send(:remove_method, :find_by_credit_limit) if Account.public_methods.any? { |m| m.to_s == 'find_by_credit_limit' }
    assert !Account.public_methods.any? { |m| m.to_s == 'find_by_credit_limit' }
561
    a = Account.find_by_credit_limit(50, :conditions => ['firm_id = ?', 6])
562
    assert Account.public_methods.any? { |m| m.to_s == 'find_by_credit_limit' }
563 564 565 566
  end

  def test_dynamic_finder_on_one_attribute_with_conditions_returns_same_results_after_caching
    # ensure this test can run independently of order
567
    class << Account; self; end.send(:remove_method, :find_by_credit_limit) if Account.public_methods.any? { |m| m.to_s == 'find_by_credit_limit' }
568 569 570 571
    a = Account.find_by_credit_limit(50, :conditions => ['firm_id = ?', 6])
    assert_equal a, Account.find_by_credit_limit(50, :conditions => ['firm_id = ?', 6]) # find_by_credit_limit has been cached
  end

572 573 574
  def test_find_by_one_attribute_with_several_options
    assert_equal accounts(:unknown), Account.find_by_credit_limit(50, :order => 'id DESC', :conditions => ['id != ?', 3])
  end
575

D
David Heinemeier Hansson 已提交
576 577 578
  def test_find_by_one_missing_attribute
    assert_raises(NoMethodError) { Topic.find_by_undertitle("The First Topic!") }
  end
579

580 581 582 583 584 585
  def test_find_by_invalid_method_syntax
    assert_raises(NoMethodError) { Topic.fail_to_find_by_title("The First Topic") }
    assert_raises(NoMethodError) { Topic.find_by_title?("The First Topic") }
    assert_raises(NoMethodError) { Topic.fail_to_find_or_create_by_title("Nonexistent Title") }
    assert_raises(NoMethodError) { Topic.find_or_create_by_title?("Nonexistent Title") }
  end
D
David Heinemeier Hansson 已提交
586 587

  def test_find_by_two_attributes
588
    assert_equal topics(:first), Topic.find_by_title_and_author_name("The First Topic", "David")
D
David Heinemeier Hansson 已提交
589 590 591 592 593 594
    assert_nil Topic.find_by_title_and_author_name("The First Topic", "Mary")
  end

  def test_find_all_by_one_attribute
    topics = Topic.find_all_by_content("Have a nice day")
    assert_equal 2, topics.size
595
    assert topics.include?(topics(:first))
D
David Heinemeier Hansson 已提交
596 597 598

    assert_equal [], Topic.find_all_by_title("The First Topic!!")
  end
599

600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625
  def test_find_all_by_one_attribute_that_is_an_aggregate
    balance = customers(:david).balance
    assert_kind_of Money, balance
    found_customers = Customer.find_all_by_balance(balance)
    assert_equal 1, found_customers.size
    assert_equal customers(:david), found_customers.first
  end

  def test_find_all_by_two_attributes_that_are_both_aggregates
    balance = customers(:david).balance
    address = customers(:david).address
    assert_kind_of Money, balance
    assert_kind_of Address, address
    found_customers = Customer.find_all_by_balance_and_address(balance, address)
    assert_equal 1, found_customers.size
    assert_equal customers(:david), found_customers.first
  end

  def test_find_all_by_two_attributes_with_one_being_an_aggregate
    balance = customers(:david).balance
    assert_kind_of Money, balance
    found_customers = Customer.find_all_by_balance_and_name(balance, customers(:david).name)
    assert_equal 1, found_customers.size
    assert_equal customers(:david), found_customers.first
  end

626 627
  def test_find_all_by_one_attribute_with_options
    topics = Topic.find_all_by_content("Have a nice day", :order => "id DESC")
628
    assert topics(:first), topics.last
629 630

    topics = Topic.find_all_by_content("Have a nice day", :order => "id")
631
    assert topics(:first), topics.first
632
  end
D
David Heinemeier Hansson 已提交
633

634
  def test_find_all_by_array_attribute
635
    assert_equal 2, Topic.find_all_by_title(["The First Topic", "The Second Topic of the day"]).size
636 637
  end

D
David Heinemeier Hansson 已提交
638 639 640
  def test_find_all_by_boolean_attribute
    topics = Topic.find_all_by_approved(false)
    assert_equal 1, topics.size
641
    assert topics.include?(topics(:first))
D
David Heinemeier Hansson 已提交
642 643

    topics = Topic.find_all_by_approved(true)
644
    assert_equal 3, topics.size
645
    assert topics.include?(topics(:second))
D
David Heinemeier Hansson 已提交
646
  end
647

D
David Heinemeier Hansson 已提交
648 649 650 651 652
  def test_find_by_nil_attribute
    topic = Topic.find_by_last_read nil
    assert_not_nil topic
    assert_nil topic.last_read
  end
653

D
David Heinemeier Hansson 已提交
654 655
  def test_find_all_by_nil_attribute
    topics = Topic.find_all_by_last_read nil
656 657
    assert_equal 3, topics.size
    assert topics.collect(&:last_read).all?(&:nil?)
D
David Heinemeier Hansson 已提交
658
  end
659

D
David Heinemeier Hansson 已提交
660 661 662 663 664 665 666 667 668 669 670
  def test_find_by_nil_and_not_nil_attributes
    topic = Topic.find_by_last_read_and_author_name nil, "Mary"
    assert_equal "Mary", topic.author_name
  end

  def test_find_all_by_nil_and_not_nil_attributes
    topics = Topic.find_all_by_last_read_and_author_name nil, "Mary"
    assert_equal 1, topics.size
    assert_equal "Mary", topics[0].author_name
  end

671 672 673 674 675
  def test_find_or_create_from_one_attribute
    number_of_companies = Company.count
    sig38 = Company.find_or_create_by_name("38signals")
    assert_equal number_of_companies + 1, Company.count
    assert_equal sig38, Company.find_or_create_by_name("38signals")
676
    assert !sig38.new_record?
677 678 679
  end

  def test_find_or_create_from_two_attributes
680 681 682 683
    number_of_topics = Topic.count
    another = Topic.find_or_create_by_title_and_author_name("Another topic","John")
    assert_equal number_of_topics + 1, Topic.count
    assert_equal another, Topic.find_or_create_by_title_and_author_name("Another topic", "John")
684
    assert !another.new_record?
685
  end
686

687 688 689 690 691 692 693 694
  def test_find_or_create_from_two_attributes_with_one_being_an_aggregate
    number_of_customers = Customer.count
    created_customer = Customer.find_or_create_by_balance_and_name(Money.new(123), "Elizabeth")
    assert_equal number_of_customers + 1, Customer.count
    assert_equal created_customer, Customer.find_or_create_by_balance(Money.new(123), "Elizabeth")
    assert !created_customer.new_record?
  end

695 696 697 698 699 700 701 702 703 704 705
  def test_find_or_create_from_one_attribute_and_hash
    number_of_companies = Company.count
    sig38 = Company.find_or_create_by_name({:name => "38signals", :firm_id => 17, :client_of => 23})
    assert_equal number_of_companies + 1, Company.count
    assert_equal sig38, Company.find_or_create_by_name({:name => "38signals", :firm_id => 17, :client_of => 23})
    assert !sig38.new_record?
    assert_equal "38signals", sig38.name
    assert_equal 17, sig38.firm_id
    assert_equal 23, sig38.client_of
  end

706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725
  def test_find_or_create_from_one_aggregate_attribute
    number_of_customers = Customer.count
    created_customer = Customer.find_or_create_by_balance(Money.new(123))
    assert_equal number_of_customers + 1, Customer.count
    assert_equal created_customer, Customer.find_or_create_by_balance(Money.new(123))
    assert !created_customer.new_record?
  end

  def test_find_or_create_from_one_aggregate_attribute_and_hash
    number_of_customers = Customer.count
    balance = Money.new(123)
    name = "Elizabeth"
    created_customer = Customer.find_or_create_by_balance({:balance => balance, :name => name})
    assert_equal number_of_customers + 1, Customer.count
    assert_equal created_customer, Customer.find_or_create_by_balance({:balance => balance, :name => name})
    assert !created_customer.new_record?
    assert_equal balance, created_customer.balance
    assert_equal name, created_customer.name
  end

726 727 728
  def test_find_or_initialize_from_one_attribute
    sig38 = Company.find_or_initialize_by_name("38signals")
    assert_equal "38signals", sig38.name
729
    assert sig38.new_record?
730
  end
J
Jeremy Kemper 已提交
731

732 733 734 735 736 737
  def test_find_or_initialize_from_one_aggregate_attribute
    new_customer = Customer.find_or_initialize_by_balance(Money.new(123))
    assert_equal 123, new_customer.balance.amount
    assert new_customer.new_record?
  end

738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753
  def test_find_or_initialize_from_one_attribute_should_not_set_attribute_even_when_protected
    c = Company.find_or_initialize_by_name({:name => "Fortune 1000", :rating => 1000})
    assert_equal "Fortune 1000", c.name
    assert_not_equal 1000, c.rating
    assert c.valid?
    assert c.new_record?
  end

  def test_find_or_create_from_one_attribute_should_set_not_attribute_even_when_protected
    c = Company.find_or_create_by_name({:name => "Fortune 1000", :rating => 1000})
    assert_equal "Fortune 1000", c.name
    assert_not_equal 1000, c.rating
    assert c.valid?
    assert !c.new_record?
  end

754 755 756 757 758
  def test_find_or_initialize_from_one_attribute_should_set_attribute_even_when_protected
    c = Company.find_or_initialize_by_name_and_rating("Fortune 1000", 1000)
    assert_equal "Fortune 1000", c.name
    assert_equal 1000, c.rating
    assert c.valid?
J
Jeremy Kemper 已提交
759
    assert c.new_record?
760 761 762 763 764 765 766
  end

  def test_find_or_create_from_one_attribute_should_set_attribute_even_when_protected
    c = Company.find_or_create_by_name_and_rating("Fortune 1000", 1000)
    assert_equal "Fortune 1000", c.name
    assert_equal 1000, c.rating
    assert c.valid?
J
Jeremy Kemper 已提交
767
    assert !c.new_record?
768
  end
769

770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785
  def test_find_or_initialize_should_set_protected_attributes_if_given_as_block
    c = Company.find_or_initialize_by_name(:name => "Fortune 1000") { |f| f.rating = 1000 }
    assert_equal "Fortune 1000", c.name
    assert_equal 1000.to_f, c.rating.to_f
    assert c.valid?
    assert c.new_record?
  end

  def test_find_or_create_should_set_protected_attributes_if_given_as_block
    c = Company.find_or_create_by_name(:name => "Fortune 1000") { |f| f.rating = 1000 }
    assert_equal "Fortune 1000", c.name
    assert_equal 1000.to_f, c.rating.to_f
    assert c.valid?
    assert !c.new_record?
  end
  
786
  def test_dynamic_find_or_initialize_from_one_attribute_caches_method
787 788
    class << Company; self; end.send(:remove_method, :find_or_initialize_by_name) if Company.public_methods.any? { |m| m.to_s == 'find_or_initialize_by_name' }
    assert !Company.public_methods.any? { |m| m.to_s == 'find_or_initialize_by_name' }
789
    sig38 = Company.find_or_initialize_by_name("38signals")
790
    assert Company.public_methods.any? { |m| m.to_s == 'find_or_initialize_by_name' }
791 792
  end

793 794 795 796
  def test_find_or_initialize_from_two_attributes
    another = Topic.find_or_initialize_by_title_and_author_name("Another topic","John")
    assert_equal "Another topic", another.title
    assert_equal "John", another.author_name
797
    assert another.new_record?
798
  end
799

800 801 802 803 804 805 806
  def test_find_or_initialize_from_one_aggregate_attribute_and_one_not
    new_customer = Customer.find_or_initialize_by_balance_and_name(Money.new(123), "Elizabeth")
    assert_equal 123, new_customer.balance.amount
    assert_equal "Elizabeth", new_customer.name
    assert new_customer.new_record?
  end

807 808 809 810 811 812 813
  def test_find_or_initialize_from_one_attribute_and_hash
    sig38 = Company.find_or_initialize_by_name({:name => "38signals", :firm_id => 17, :client_of => 23})
    assert_equal "38signals", sig38.name
    assert_equal 17, sig38.firm_id
    assert_equal 23, sig38.client_of
    assert sig38.new_record?
  end
814

815 816 817 818 819 820 821 822 823
  def test_find_or_initialize_from_one_aggregate_attribute_and_hash
    balance = Money.new(123)
    name = "Elizabeth"
    new_customer = Customer.find_or_initialize_by_balance({:balance => balance, :name => name})
    assert_equal balance, new_customer.balance
    assert_equal name, new_customer.name
    assert new_customer.new_record?
  end

D
David Heinemeier Hansson 已提交
824 825 826 827
  def test_find_with_bad_sql
    assert_raises(ActiveRecord::StatementInvalid) { Topic.find_by_sql "select 1 from badtable" }
  end

828 829 830 831 832
  def test_find_with_invalid_params
    assert_raises(ArgumentError) { Topic.find :first, :join => "It should be `joins'" }
    assert_raises(ArgumentError) { Topic.find :first, :conditions => '1 = 1', :join => "It should be `joins'" }
  end

833 834 835 836
  def test_dynamic_finder_with_invalid_params
    assert_raises(ArgumentError) { Topic.find_by_title 'No Title', :join => "It should be `joins'" }
  end

D
David Heinemeier Hansson 已提交
837
  def test_find_all_with_limit
838
    first_five_developers = Developer.find :all, :order => 'id ASC', :limit =>  5
D
David Heinemeier Hansson 已提交
839 840 841
    assert_equal 5, first_five_developers.length
    assert_equal 'David', first_five_developers.first.name
    assert_equal 'fixture_5', first_five_developers.last.name
842

843
    no_developers = Developer.find :all, :order => 'id ASC', :limit => 0
D
David Heinemeier Hansson 已提交
844 845
    assert_equal 0, no_developers.length
  end
846

D
David Heinemeier Hansson 已提交
847 848 849 850
  def test_find_all_with_limit_and_offset
    first_three_developers = Developer.find :all, :order => 'id ASC', :limit => 3, :offset => 0
    second_three_developers = Developer.find :all, :order => 'id ASC', :limit => 3, :offset => 3
    last_two_developers = Developer.find :all, :order => 'id ASC', :limit => 2, :offset => 8
851

D
David Heinemeier Hansson 已提交
852 853 854
    assert_equal 3, first_three_developers.length
    assert_equal 3, second_three_developers.length
    assert_equal 2, last_two_developers.length
855

D
David Heinemeier Hansson 已提交
856 857 858 859 860
    assert_equal 'David', first_three_developers.first.name
    assert_equal 'fixture_4', second_three_developers.first.name
    assert_equal 'fixture_9', last_two_developers.first.name
  end

861 862 863 864 865 866 867 868 869 870
  def test_find_all_with_limit_and_offset_and_multiple_order_clauses
    first_three_posts = Post.find :all, :order => 'author_id, id', :limit => 3, :offset => 0
    second_three_posts = Post.find :all, :order => ' author_id,id ', :limit => 3, :offset => 3
    last_posts = Post.find :all, :order => ' author_id, id  ', :limit => 3, :offset => 6

    assert_equal [[0,3],[1,1],[1,2]], first_three_posts.map { |p| [p.author_id, p.id] }
    assert_equal [[1,4],[1,5],[1,6]], second_three_posts.map { |p| [p.author_id, p.id] }
    assert_equal [[2,7]], last_posts.map { |p| [p.author_id, p.id] }
  end

871
  def test_find_all_with_join
872
    developers_on_project_one = Developer.find(
873 874
      :all,
      :joins => 'LEFT JOIN developers_projects ON developers.id = developers_projects.developer_id',
875 876
      :conditions => 'project_id=1'
    )
877
    assert_equal 3, developers_on_project_one.length
878 879 880
    developer_names = developers_on_project_one.map { |d| d.name }
    assert developer_names.include?('David')
    assert developer_names.include?('Jamis')
881
  end
882

883 884 885 886 887 888 889 890 891
  def test_joins_dont_clobber_id
    first = Firm.find(
      :first,
      :joins => 'INNER JOIN companies AS clients ON clients.firm_id = companies.id',
      :conditions => 'companies.id = 1'
    )
    assert_equal 1, first.id
  end

892 893 894
  def test_find_by_id_with_conditions_with_or
    assert_nothing_raised do
      Post.find([1,2,3],
895
        :conditions => "posts.id <= 3 OR posts.#{QUOTED_TYPE} = 'Post'")
896 897
    end
  end
898

899 900
  # http://dev.rubyonrails.org/ticket/6778
  def test_find_ignores_previously_inserted_record
901
    post = Post.create!(:title => 'test', :body => 'it out')
902 903 904
    assert_equal [], Post.find_all_by_id(nil)
  end

905 906 907 908 909 910 911 912 913
  def test_find_by_empty_ids
    assert_equal [], Post.find([])
  end

  def test_find_by_empty_in_condition
    assert_equal [], Post.find(:all, :conditions => ['id in (?)', []])
  end

  def test_find_by_records
J
Jeremy Kemper 已提交
914 915 916
    p1, p2 = Post.find(:all, :limit => 2, :order => 'id asc')
    assert_equal [p1, p2], Post.find(:all, :conditions => ['id in (?)', [p1, p2]], :order => 'id asc')
    assert_equal [p1, p2], Post.find(:all, :conditions => ['id in (?)', [p1, p2.id]], :order => 'id asc')
917 918
  end

919 920 921 922 923 924 925 926 927
  def test_select_value
    assert_equal "37signals", Company.connection.select_value("SELECT name FROM companies WHERE id = 1")
    assert_nil Company.connection.select_value("SELECT name FROM companies WHERE id = -1")
    # make sure we didn't break count...
    assert_equal 0, Company.count_by_sql("SELECT COUNT(*) FROM companies WHERE name = 'Halliburton'")
    assert_equal 1, Company.count_by_sql("SELECT COUNT(*) FROM companies WHERE name = '37signals'")
  end

  def test_select_values
928 929
    assert_equal ["1","2","3","4","5","6","7","8","9"], Company.connection.select_values("SELECT id FROM companies ORDER BY id").map! { |i| i.to_s }
    assert_equal ["37signals","Summit","Microsoft", "Flamboyant Software", "Ex Nihilo", "RailsCore", "Leetsoft", "Jadedpixel", "Odegy"], Company.connection.select_values("SELECT name FROM companies ORDER BY id")
930 931
  end

932 933 934 935 936
  def test_select_rows
    assert_equal(
      [["1", nil, nil, "37signals"],
       ["2", "1", "2", "Summit"],
       ["3", "1", "1", "Microsoft"]],
937
      Company.connection.select_rows("SELECT id, firm_id, client_of, name FROM companies WHERE id IN (1,2,3) ORDER BY id").map! {|i| i.map! {|j| j.to_s unless j.nil?}})
938
    assert_equal [["1", "37signals"], ["2", "Summit"], ["3", "Microsoft"]],
939
      Company.connection.select_rows("SELECT id, name FROM companies WHERE id IN (1,2,3) ORDER BY id").map! {|i| i.map! {|j| j.to_s unless j.nil?}}
940 941
  end

942
  def test_find_with_order_on_included_associations_with_construct_finder_sql_for_association_limiting_and_is_distinct
943
    assert_equal 2, Post.find(:all, :include => { :authors => :author_address }, :order => ' author_addresses.id DESC ', :limit => 2).size
944

945 946
    assert_equal 3, Post.find(:all, :include => { :author => :author_address, :authors => :author_address},
                              :order => ' author_addresses_authors.id DESC ', :limit => 3).size
947 948
  end

949
  def test_with_limiting_with_custom_select
T
Tarmo Tänav 已提交
950
    posts = Post.find(:all, :include => :author, :select => ' posts.*, authors.id as "author_id"', :limit => 3, :order => 'posts.id')
951 952
    assert_equal 3, posts.size
    assert_equal [0, 1, 1], posts.map(&:author_id).sort
953
  end
954

D
David Heinemeier Hansson 已提交
955 956 957 958 959 960 961 962 963
  protected
    def bind(statement, *vars)
      if vars.first.is_a?(Hash)
        ActiveRecord::Base.send(:replace_named_bind_variables, statement, vars.first)
      else
        ActiveRecord::Base.send(:replace_bind_variables, statement, vars)
      end
    end
end