finder_test.rb 44.2 KB
Newer Older
1
require "cases/helper"
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
require "models/post"
require "models/author"
require "models/categorization"
require "models/comment"
require "models/company"
require "models/tagging"
require "models/topic"
require "models/reply"
require "models/entrant"
require "models/project"
require "models/developer"
require "models/computer"
require "models/customer"
require "models/toy"
require "models/matey"
require "models/dog"
require "models/car"
require "models/tyre"
D
David Heinemeier Hansson 已提交
20

21
class FinderTest < ActiveRecord::TestCase
R
Fix...  
Ryuta Kamizono 已提交
22
  fixtures :companies, :topics, :entrants, :developers, :developers_projects, :posts, :comments, :accounts, :authors, :author_addresses, :customers, :categories, :categorizations, :cars
D
David Heinemeier Hansson 已提交
23

24 25
  def test_find_by_id_with_hash
    assert_raises(ActiveRecord::StatementInvalid) do
26
      Post.find_by_id(limit: 1)
27 28 29 30 31
    end
  end

  def test_find_by_title_and_id_with_hash
    assert_raises(ActiveRecord::StatementInvalid) do
32
      Post.find_by_title_and_id("foo", limit: 1)
33 34 35
    end
  end

D
David Heinemeier Hansson 已提交
36
  def test_find
37
    assert_equal(topics(:first).title, Topic.find(1).title)
D
David Heinemeier Hansson 已提交
38
  end
39

40
  def test_find_with_proc_parameter_and_block
S
Sean Griffin 已提交
41
    exception = assert_raises(RuntimeError) do
42 43
      Topic.all.find(-> { raise "should happen" }) { |e| e.title == "non-existing-title" }
    end
S
Sean Griffin 已提交
44
    assert_equal "should happen", exception.message
45

46
    assert_nothing_raised do
47 48 49 50
      Topic.all.find(-> { raise "should not happen" }) { |e| e.title == topics(:first).title }
    end
  end

51 52
  def test_find_with_ids_returning_ordered
    records = Topic.find([4,2,5])
53 54 55
    assert_equal "The Fourth Topic of the day", records[0].title
    assert_equal "The Second Topic of the day", records[1].title
    assert_equal "The Fifth Topic of the day", records[2].title
56 57

    records = Topic.find(4,2,5)
58 59 60
    assert_equal "The Fourth Topic of the day", records[0].title
    assert_equal "The Second Topic of the day", records[1].title
    assert_equal "The Fifth Topic of the day", records[2].title
61

62 63 64 65
    records = Topic.find(["4","2","5"])
    assert_equal "The Fourth Topic of the day", records[0].title
    assert_equal "The Second Topic of the day", records[1].title
    assert_equal "The Fifth Topic of the day", records[2].title
66

67 68 69 70
    records = Topic.find("4","2","5")
    assert_equal "The Fourth Topic of the day", records[0].title
    assert_equal "The Second Topic of the day", records[1].title
    assert_equal "The Fifth Topic of the day", records[2].title
71 72 73 74 75
  end

  def test_find_with_ids_and_order_clause
    # The order clause takes precedence over the informed ids
    records = Topic.order(:author_name).find([5,3,1])
76 77 78
    assert_equal "The Third Topic of the day", records[0].title
    assert_equal "The First Topic",            records[1].title
    assert_equal "The Fifth Topic of the day", records[2].title
79 80

    records = Topic.order(:id).find([5,3,1])
81 82 83
    assert_equal "The First Topic",            records[0].title
    assert_equal "The Third Topic of the day", records[1].title
    assert_equal "The Fifth Topic of the day", records[2].title
84 85
  end

86 87 88
  def test_find_with_ids_with_limit_and_order_clause
    # The order clause takes precedence over the informed ids
    records = Topic.limit(2).order(:id).find([5,3,1])
89
    assert_equal 2, records.size
90 91
    assert_equal "The First Topic",            records[0].title
    assert_equal "The Third Topic of the day", records[1].title
92 93 94 95 96
  end

  def test_find_with_ids_and_limit
    records = Topic.limit(3).find([3,2,5,1,4])
    assert_equal 3, records.size
97 98 99
    assert_equal "The Third Topic of the day",  records[0].title
    assert_equal "The Second Topic of the day", records[1].title
    assert_equal "The Fifth Topic of the day",  records[2].title
100 101
  end

102 103
  def test_find_with_ids_where_and_limit
    # Please note that Topic 1 is the only not approved so
104
    # if it were among the first 3 it would raise an ActiveRecord::RecordNotFound
105 106
    records = Topic.where(approved: true).limit(3).find([3,2,5,1,4])
    assert_equal 3, records.size
107 108 109
    assert_equal "The Third Topic of the day",  records[0].title
    assert_equal "The Second Topic of the day", records[1].title
    assert_equal "The Fifth Topic of the day",  records[2].title
110 111
  end

112
  def test_find_with_ids_and_offset
113 114
    records = Topic.offset(2).find([3,2,5,1,4])
    assert_equal 3, records.size
115 116 117
    assert_equal "The Fifth Topic of the day",  records[0].title
    assert_equal "The First Topic",             records[1].title
    assert_equal "The Fourth Topic of the day", records[2].title
118 119
  end

120 121 122 123 124 125
  def test_find_passing_active_record_object_is_deprecated
    assert_deprecated do
      Topic.find(Topic.last)
    end
  end

126
  def test_symbols_table_ref
127
    gc_disabled = GC.disable
128
    Post.where("author_id" => nil)  # warm up
129 130 131
    x = Symbol.all_symbols.count
    Post.where("title" => {"xxxqqqq" => "bar"})
    assert_equal x, Symbol.all_symbols.count
R
Ryuta Kamizono 已提交
132 133
  ensure
    GC.enable if gc_disabled == false
134 135
  end

136 137 138 139 140
  # 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
141

D
David Heinemeier Hansson 已提交
142
  def test_exists
143 144 145 146
    assert_equal true, Topic.exists?(1)
    assert_equal true, Topic.exists?("1")
    assert_equal true, Topic.exists?(title: "The First Topic")
    assert_equal true, Topic.exists?(heading: "The First Topic")
147
    assert_equal true, Topic.exists?(author_name: "Mary", approved: true)
148
    assert_equal true, Topic.exists?(["parent_id = ?", 1])
149
    assert_equal true, Topic.exists?(id: [1, 9999])
150 151

    assert_equal false, Topic.exists?(45)
152
    assert_equal false, Topic.exists?(Topic.new.id)
153

154 155 156
    assert_raise(NoMethodError) { Topic.exists?([1,2]) }
  end

157
  def test_exists_with_polymorphic_relation
158 159
    post = Post.create!(title: "Post", body: "default", taggings: [Tagging.new(comment: "tagging comment")])
    relation = Post.tagged_with_comment("tagging comment")
160

161 162
    assert_equal true, relation.exists?(title: ["Post"])
    assert_equal true, relation.exists?(["title LIKE ?", "Post%"])
163 164 165 166 167 168 169
    assert_equal true, relation.exists?
    assert_equal true, relation.exists?(post.id)
    assert_equal true, relation.exists?(post.id.to_s)

    assert_equal false, relation.exists?(false)
  end

170 171 172 173 174 175
  def test_exists_passing_active_record_object_is_deprecated
    assert_deprecated do
      Topic.exists?(Topic.new)
    end
  end

176
  def test_exists_returns_false_when_parameter_has_invalid_type
177
    assert_equal false, Topic.exists?("foo")
178
    assert_equal false, Topic.exists?(("9"*53).to_i) # number that's bigger than int
D
David Heinemeier Hansson 已提交
179
  end
180

181
  def test_exists_does_not_select_columns_without_alias
182
    assert_sql(/SELECT\W+1 AS one FROM ["`]topics["`]/i) do
183 184 185 186
      Topic.exists?
    end
  end

187
  def test_exists_returns_true_with_one_record_and_no_args
188
    assert_equal true, Topic.exists?
189
  end
190

E
Egor Lynko 已提交
191
  def test_exists_returns_false_with_false_arg
192
    assert_equal false, Topic.exists?(false)
E
Egor Lynko 已提交
193 194
  end

195 196 197
  # exists? should handle nil for id's that come from URLs and always return false
  # (example: Topic.exists?(params[:id])) where params[:id] is nil
  def test_exists_with_nil_arg
198 199 200 201 202
    assert_equal false, Topic.exists?(nil)
    assert_equal true, Topic.exists?

    assert_equal false, Topic.first.replies.exists?(nil)
    assert_equal true, Topic.first.replies.exists?
203 204
  end

205 206
  # ensures +exists?+ runs valid SQL by excluding order value
  def test_exists_with_order
207
    assert_equal true, Topic.order(:id).distinct.exists?
208 209
  end

210
  def test_exists_with_includes_limit_and_empty_result
211
    assert_equal false, Topic.includes(:replies).limit(0).exists?
212
    assert_equal false, Topic.includes(:replies).limit(1).where("0 = 1").exists?
213 214
  end

215 216
  def test_exists_with_distinct_association_includes_and_limit
    author = Author.first
217 218
    assert_equal false, author.unique_categorized_posts.includes(:special_comments).limit(0).exists?
    assert_equal true, author.unique_categorized_posts.includes(:special_comments).limit(1).exists?
219 220 221 222
  end

  def test_exists_with_distinct_association_includes_limit_and_order
    author = Author.first
223 224
    assert_equal false, author.unique_categorized_posts.includes(:special_comments).order("comments.tags_count DESC").limit(0).exists?
    assert_equal true, author.unique_categorized_posts.includes(:special_comments).order("comments.tags_count DESC").limit(1).exists?
225 226
  end

227
  def test_exists_with_empty_table_and_no_args_given
228
    Topic.delete_all
229
    assert_equal false, Topic.exists?
230
  end
231

232 233
  def test_exists_with_aggregate_having_three_mappings
    existing_address = customers(:david).address
234
    assert_equal true, Customer.exists?(address: existing_address)
235 236 237 238
  end

  def test_exists_with_aggregate_having_three_mappings_with_one_difference
    existing_address = customers(:david).address
239 240 241
    assert_equal false, Customer.exists?(address:       Address.new(existing_address.street, existing_address.city, existing_address.country + "1"))
    assert_equal false, Customer.exists?(address:       Address.new(existing_address.street, existing_address.city + "1", existing_address.country))
    assert_equal false, Customer.exists?(address:       Address.new(existing_address.street + "1", existing_address.city, existing_address.country))
242 243
  end

244
  def test_exists_does_not_instantiate_records
245 246 247
    assert_not_called(Developer, :instantiate) do
      Developer.exists?
    end
248 249
  end

D
David Heinemeier Hansson 已提交
250 251 252 253
  def test_find_by_array_of_one_id
    assert_kind_of(Array, Topic.find([ 1 ]))
    assert_equal(1, Topic.find([ 1 ]).length)
  end
254

D
David Heinemeier Hansson 已提交
255
  def test_find_by_ids
256 257 258 259 260
    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
261
    assert_equal 2, Entrant.limit(2).find([1,3,2]).size
262 263
    entrants = Entrant.limit(3).offset(2).find([1,3,2])
    assert_equal 1, entrants.size
264
    assert_equal "Ruby Guru", entrants.first.name
265 266 267 268

    # 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.
J
Jon Leighton 已提交
269
    devs = Developer.all
270
    last_devs = Developer.limit(3).offset(9).find devs.map(&:id)
271
    assert_equal 2, last_devs.size
272 273
    assert_equal "fixture_10", last_devs[0].name
    assert_equal "Jamis", last_devs[1].name
D
David Heinemeier Hansson 已提交
274 275
  end

276
  def test_find_with_large_number
277
    assert_raises(ActiveRecord::RecordNotFound) { Topic.find("9999999999999999999999999999999") }
278 279 280
  end

  def test_find_by_with_large_number
281
    assert_nil Topic.find_by(id: "9999999999999999999999999999999")
282 283 284
  end

  def test_find_by_id_with_large_number
285
    assert_nil Topic.find_by_id("9999999999999999999999999999999")
286 287
  end

288
  def test_find_on_relation_with_large_number
289
    assert_nil Topic.where("1=1").find_by(id: 9999999999999999999999999999999)
290 291 292 293
  end

  def test_find_by_bang_on_relation_with_large_number
    assert_raises(ActiveRecord::RecordNotFound) do
294
      Topic.where("1=1").find_by!(id: 9999999999999999999999999999999)
295 296 297
    end
  end

298 299 300 301
  def test_find_an_empty_array
    assert_equal [], Topic.find([])
  end

302 303 304 305
  def test_find_doesnt_have_implicit_ordering
    assert_sql(/^((?!ORDER).)*$/) { Topic.find(1) }
  end

D
David Heinemeier Hansson 已提交
306
  def test_find_by_ids_missing_one
307
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(1, 2, 45) }
D
David Heinemeier Hansson 已提交
308
  end
309

310
  def test_find_with_group_and_sanitized_having_method
311
    developers = Developer.group(:salary).having("sum(salary) > ?", 10000).select("salary").to_a
312 313 314 315 316
    assert_equal 3, developers.size
    assert_equal 3, developers.map(&:salary).uniq.size
    assert developers.all? { |developer| developer.salary > 10000 }
  end

D
David Heinemeier Hansson 已提交
317 318
  def test_find_with_entire_select_statement
    topics = Topic.find_by_sql "SELECT * FROM topics WHERE author_name = 'Mary'"
319

D
David Heinemeier Hansson 已提交
320
    assert_equal(1, topics.size)
321
    assert_equal(topics(:second).title, topics.first.title)
D
David Heinemeier Hansson 已提交
322
  end
323

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

D
David Heinemeier Hansson 已提交
327
    assert_equal(1, topics.size)
328
    assert_equal(topics(:second).title, topics.first.title)
D
David Heinemeier Hansson 已提交
329
  end
330

331 332 333 334
  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
335

336 337 338 339 340 341
  def test_find_by_association_subquery
    author = authors(:david)
    assert_equal author.post, Post.find_by(author: Author.where(id: author))
    assert_equal author.post, Post.find_by(author_id: Author.where(id: author))
  end

342 343 344 345 346 347 348 349 350 351 352 353 354 355 356
  def test_take
    assert_equal topics(:first), Topic.take
  end

  def test_take_failing
    assert_nil Topic.where("title = 'This title does not exist'").take
  end

  def test_take_bang_present
    assert_nothing_raised do
      assert_equal topics(:second), Topic.where("title = 'The Second Topic of the day'").take!
    end
  end

  def test_take_bang_missing
357
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
358 359 360 361
      Topic.where("title = 'This title does not exist'").take!
    end
  end

362
  def test_first
363
    assert_equal topics(:second).title, Topic.where("title = 'The Second Topic of the day'").first.title
364
  end
365

366
  def test_first_failing
367
    assert_nil Topic.where("title = 'The Second Topic of the day!'").first
368
  end
D
David Heinemeier Hansson 已提交
369

370 371 372 373 374 375 376
  def test_first_bang_present
    assert_nothing_raised do
      assert_equal topics(:second), Topic.where("title = 'The Second Topic of the day'").first!
    end
  end

  def test_first_bang_missing
377
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
378 379 380 381
      Topic.where("title = 'This title does not exist'").first!
    end
  end

382 383 384 385 386 387
  def test_first_have_primary_key_order_by_default
    expected = topics(:first)
    expected.touch # PostgreSQL changes the default order if no order clause is used
    assert_equal expected, Topic.first
  end

388
  def test_model_class_responds_to_first_bang
389 390
    assert Topic.first!
    Topic.delete_all
391
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
392 393 394 395
      Topic.first!
    end
  end

396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412
  def test_second
    assert_equal topics(:second).title, Topic.second.title
  end

  def test_second_with_offset
    assert_equal topics(:fifth), Topic.offset(3).second
  end

  def test_second_have_primary_key_order_by_default
    expected = topics(:second)
    expected.touch # PostgreSQL changes the default order if no order clause is used
    assert_equal expected, Topic.second
  end

  def test_model_class_responds_to_second_bang
    assert Topic.second!
    Topic.delete_all
413
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434
      Topic.second!
    end
  end

  def test_third
    assert_equal topics(:third).title, Topic.third.title
  end

  def test_third_with_offset
    assert_equal topics(:fifth), Topic.offset(2).third
  end

  def test_third_have_primary_key_order_by_default
    expected = topics(:third)
    expected.touch # PostgreSQL changes the default order if no order clause is used
    assert_equal expected, Topic.third
  end

  def test_model_class_responds_to_third_bang
    assert Topic.third!
    Topic.delete_all
435
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456
      Topic.third!
    end
  end

  def test_fourth
    assert_equal topics(:fourth).title, Topic.fourth.title
  end

  def test_fourth_with_offset
    assert_equal topics(:fifth), Topic.offset(1).fourth
  end

  def test_fourth_have_primary_key_order_by_default
    expected = topics(:fourth)
    expected.touch # PostgreSQL changes the default order if no order clause is used
    assert_equal expected, Topic.fourth
  end

  def test_model_class_responds_to_fourth_bang
    assert Topic.fourth!
    Topic.delete_all
457
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478
      Topic.fourth!
    end
  end

  def test_fifth
    assert_equal topics(:fifth).title, Topic.fifth.title
  end

  def test_fifth_with_offset
    assert_equal topics(:fifth), Topic.offset(0).fifth
  end

  def test_fifth_have_primary_key_order_by_default
    expected = topics(:fifth)
    expected.touch # PostgreSQL changes the default order if no order clause is used
    assert_equal expected, Topic.fifth
  end

  def test_model_class_responds_to_fifth_bang
    assert Topic.fifth!
    Topic.delete_all
479
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
480 481 482 483
      Topic.fifth!
    end
  end

484 485
  def test_second_to_last
    assert_equal topics(:fourth).title, Topic.second_to_last.title
486 487 488

    # test with offset
    assert_equal topics(:fourth), Topic.offset(1).second_to_last
489 490 491
    assert_equal topics(:fourth), Topic.offset(2).second_to_last
    assert_equal topics(:fourth), Topic.offset(3).second_to_last
    assert_equal nil, Topic.offset(4).second_to_last
492 493 494
    assert_equal nil, Topic.offset(5).second_to_last

    #test with limit
495
    # assert_equal nil, Topic.limit(1).second # TODO: currently failing
496
    assert_equal nil, Topic.limit(1).second_to_last
497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514
  end

  def test_second_to_last_have_primary_key_order_by_default
    expected = topics(:fourth)
    expected.touch # PostgreSQL changes the default order if no order clause is used
    assert_equal expected, Topic.second_to_last
  end

  def test_model_class_responds_to_second_to_last_bang
    assert Topic.second_to_last!
    Topic.delete_all
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
      Topic.second_to_last!
    end
  end

  def test_third_to_last
    assert_equal topics(:third).title, Topic.third_to_last.title
515 516 517

    # test with offset
    assert_equal topics(:third), Topic.offset(1).third_to_last
518 519 520
    assert_equal topics(:third), Topic.offset(2).third_to_last
    assert_equal nil, Topic.offset(3).third_to_last
    assert_equal nil, Topic.offset(4).third_to_last
521 522 523
    assert_equal nil, Topic.offset(5).third_to_last

    # test with limit
524
    # assert_equal nil, Topic.limit(1).third # TODO: currently failing
525
    assert_equal nil, Topic.limit(1).third_to_last
526
    # assert_equal nil, Topic.limit(2).third # TODO: currently failing
527
    assert_equal nil, Topic.limit(2).third_to_last
528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543
  end

  def test_third_to_last_have_primary_key_order_by_default
    expected = topics(:third)
    expected.touch # PostgreSQL changes the default order if no order clause is used
    assert_equal expected, Topic.third_to_last
  end

  def test_model_class_responds_to_third_to_last_bang
    assert Topic.third_to_last!
    Topic.delete_all
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
      Topic.third_to_last!
    end
  end

544 545 546 547 548 549 550
  def test_last_bang_present
    assert_nothing_raised do
      assert_equal topics(:second), Topic.where("title = 'The Second Topic of the day'").last!
    end
  end

  def test_last_bang_missing
551
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
552 553 554 555
      Topic.where("title = 'This title does not exist'").last!
    end
  end

556
  def test_model_class_responds_to_last_bang
557
    assert_equal topics(:fifth), Topic.last!
558
    assert_raises_with_message ActiveRecord::RecordNotFound, "Couldn't find Topic" do
559 560 561 562 563
      Topic.delete_all
      Topic.last!
    end
  end

564
  def test_take_and_first_and_last_with_integer_should_use_sql_limit
565 566 567
    assert_sql(/LIMIT|ROWNUM <=/) { Topic.take(3).entries }
    assert_sql(/LIMIT|ROWNUM <=/) { Topic.first(2).entries }
    assert_sql(/LIMIT|ROWNUM <=/) { Topic.last(5).entries }
568 569 570 571 572 573
  end

  def test_last_with_integer_and_order_should_keep_the_order
    assert_equal Topic.order("title").to_a.last(2), Topic.order("title").last(2)
  end

574 575 576 577
  def test_last_with_integer_and_order_should_use_sql_limit
    relation = Topic.order("title")
    assert_queries(1) { relation.last(5) }
    assert !relation.loaded?
578 579
  end

580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597
  def test_last_with_integer_and_reorder_should_use_sql_limit
    relation = Topic.reorder("title")
    assert_queries(1) { relation.last(5) }
    assert !relation.loaded?
  end

  def test_last_on_loaded_relation_should_not_use_sql
    relation  = Topic.limit(10).load
    assert_no_queries do
      relation.last
      relation.last(2)
    end
  end

  def test_last_with_irreversible_order
    assert_deprecated do
      Topic.order("coalesce(author_name, title)").last
    end
598
  end
599 600

  def test_last_on_relation_with_limit_and_offset
601
    post = posts("sti_comments")
602 603 604 605 606 607 608 609 610 611

    comments = post.comments.order(id: :asc)
    assert_equal comments.limit(2).to_a.last, comments.limit(2).last
    assert_equal comments.limit(2).to_a.last(2), comments.limit(2).last(2)
    assert_equal comments.limit(2).to_a.last(3), comments.limit(2).last(3)

    comments = comments.offset(1)
    assert_equal comments.limit(2).to_a.last, comments.limit(2).last
    assert_equal comments.limit(2).to_a.last(2), comments.limit(2).last(2)
    assert_equal comments.limit(2).to_a.last(3), comments.limit(2).last(3)
612 613
  end

614 615
  def test_take_and_first_and_last_with_integer_should_return_an_array
    assert_kind_of Array, Topic.take(5)
616 617 618 619
    assert_kind_of Array, Topic.first(5)
    assert_kind_of Array, Topic.last(5)
  end

D
David Heinemeier Hansson 已提交
620
  def test_unexisting_record_exception_handling
621
    assert_raise(ActiveRecord::RecordNotFound) {
D
David Heinemeier Hansson 已提交
622 623
      Topic.find(1).parent
    }
624

625
    Topic.find(2).topic
D
David Heinemeier Hansson 已提交
626
  end
627

628
  def test_find_only_some_columns
629
    topic = Topic.select("author_name").find(1)
630
    assert_raise(ActiveModel::MissingAttributeError) {topic.title}
631
    assert_raise(ActiveModel::MissingAttributeError) {topic.title?}
632
    assert_nil topic.read_attribute("title")
633 634
    assert_equal "David", topic.author_name
    assert !topic.attribute_present?("title")
635
    assert !topic.attribute_present?(:title)
636
    assert topic.attribute_present?("author_name")
637
    assert_respond_to topic, "author_name"
638
  end
J
Jeremy Kemper 已提交
639

640
  def test_find_on_array_conditions
641 642
    assert Topic.where(["approved = ?", false]).find(1)
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where(["approved = ?", true]).find(1) }
D
David Heinemeier Hansson 已提交
643
  end
644

645
  def test_find_on_hash_conditions
646 647
    assert Topic.where(approved: false).find(1)
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where(approved: true).find(1) }
648
  end
649

650
  def test_find_on_hash_conditions_with_qualified_attribute_dot_notation_string
651 652
    assert Topic.where("topics.approved" => false).find(1)
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where("topics.approved" => true).find(1) }
653 654
  end

655 656 657 658 659
  def test_find_on_hash_conditions_with_qualified_attribute_dot_notation_symbol
    assert Topic.where('topics.approved': false).find(1)
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where('topics.approved': true).find(1) }
  end

660
  def test_find_on_hash_conditions_with_hashed_table_name
661 662
    assert Topic.where(topics: { approved: false }).find(1)
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where(topics: { approved: true }).find(1) }
663 664
  end

665
  def test_find_on_combined_explicit_and_hashed_table_names
666 667 668
    assert Topic.where("topics.approved" => false, topics: { author_name: "David" }).find(1)
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where("topics.approved" => true, topics: { author_name: "David" }).find(1) }
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where("topics.approved" => false, topics: { author_name: "Melanie" }).find(1) }
669 670
  end

671
  def test_find_with_hash_conditions_on_joined_table
672
    firms = Firm.joins(:account).where(accounts: { credit_limit: 50 })
673 674 675 676 677
    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
678
    firms = DependentFirm.joins(:account).where(name: "RailsCore", accounts: { credit_limit: 55..60 })
679 680 681 682
    assert_equal 1, firms.size
    assert_equal companies(:rails_core), firms.first
  end

683 684
  def test_find_on_hash_conditions_with_explicit_table_name_and_aggregate
    david = customers(:david)
685
    assert Customer.where("customers.name" => david.name, :address => david.address).find(david.id)
686
    assert_raise(ActiveRecord::RecordNotFound) {
687
      Customer.where("customers.name" => david.name + "1", :address => david.address).find(david.id)
688 689 690
    }
  end

691
  def test_find_on_association_proxy_conditions
J
Jon Leighton 已提交
692
    assert_equal [1, 2, 3, 5, 6, 7, 8, 9, 10, 12], Comment.where(post_id: authors(:david).posts).map(&:id).sort
693 694
  end

695
  def test_find_on_hash_conditions_with_range
696 697
    assert_equal [1,2], Topic.where(id: 1..2).to_a.map(&:id).sort
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where(id: 2..3).find(1) }
698
  end
699

700
  def test_find_on_hash_conditions_with_end_exclusive_range
701 702 703
    assert_equal [1,2,3], Topic.where(id: 1..3).to_a.map(&:id).sort
    assert_equal [1,2], Topic.where(id: 1...3).to_a.map(&:id).sort
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where(id: 2...3).find(3) }
704 705
  end

706
  def test_find_on_hash_conditions_with_multiple_ranges
707 708
    assert_equal [1,2,3], Comment.where(id: 1..3, post_id: 1..2).to_a.map(&:id).sort
    assert_equal [1], Comment.where(id: 1..1, post_id: 1..10).to_a.map(&:id).sort
709
  end
710

711
  def test_find_on_hash_conditions_with_array_of_integers_and_ranges
712
    assert_equal [1,2,3,5,6,7,8,9], Comment.where(id: [1..2, 3, 5, 6..8, 9]).to_a.map(&:id).sort
713 714
  end

715 716 717 718
  def test_find_on_hash_conditions_with_array_of_ranges
    assert_equal [1,2,6,7,8], Comment.where(id: [1..2, 6..8]).to_a.map(&:id).sort
  end

719
  def test_find_on_multiple_hash_conditions
720 721 722 723
    assert Topic.where(author_name: "David", title: "The First Topic", replies_count: 1, approved: false).find(1)
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where(author_name: "David", title: "The First Topic", replies_count: 1, approved: true).find(1) }
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where(author_name: "David", title: "HHC", replies_count: 1, approved: false).find(1) }
    assert_raise(ActiveRecord::RecordNotFound) { Topic.where(author_name: "David", title: "The First Topic", replies_count: 1, approved: true).find(1) }
724
  end
725

726
  def test_condition_interpolation
727
    assert_kind_of Firm, Company.where("name = '%s'", "37signals").first
728 729 730
    assert_nil Company.where(["name = '%s'", "37signals!"]).first
    assert_nil Company.where(["name = '%s'", "37signals!' OR 1=1"]).first
    assert_kind_of Time, Topic.where(["id = %d", 1]).first.written_on
731 732
  end

733
  def test_condition_array_interpolation
734 735 736 737
    assert_kind_of Firm, Company.where(["name = '%s'", "37signals"]).first
    assert_nil Company.where(["name = '%s'", "37signals!"]).first
    assert_nil Company.where(["name = '%s'", "37signals!' OR 1=1"]).first
    assert_kind_of Time, Topic.where(["id = %d", 1]).first.written_on
D
David Heinemeier Hansson 已提交
738
  end
739

740
  def test_condition_hash_interpolation
741 742 743
    assert_kind_of Firm, Company.where(name: "37signals").first
    assert_nil Company.where(name: "37signals!").first
    assert_kind_of Time, Topic.where(id: 1).first.written_on
744
  end
745

746
  def test_hash_condition_find_malformed
747
    assert_raise(ActiveRecord::StatementInvalid) {
748
      Company.where(id: 2, dhh: true).first
749 750
    }
  end
751

752 753
  def test_hash_condition_find_with_escaped_characters
    Company.create("name" => "Ain't noth'n like' \#stuff")
754
    assert Company.where(name: "Ain't noth'n like' \#stuff").first
755 756 757
  end

  def test_hash_condition_find_with_array
758 759 760
    p1, p2 = Post.limit(2).order("id asc").to_a
    assert_equal [p1, p2], Post.where(id: [p1, p2]).order("id asc").to_a
    assert_equal [p1, p2], Post.where(id: [p1, p2.id]).order("id asc").to_a
761 762 763
  end

  def test_hash_condition_find_with_nil
764
    topic = Topic.where(last_read: nil).first
765 766
    assert_not_nil topic
    assert_nil topic.last_read
767
  end
D
David Heinemeier Hansson 已提交
768

769 770 771
  def test_hash_condition_find_with_aggregate_having_one_mapping
    balance = customers(:david).balance
    assert_kind_of Money, balance
772
    found_customer = Customer.where(balance: balance).first
773 774 775 776 777 778
    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
779
    found_customer = Customer.where(gps_location: gps_location).first
780 781 782 783 784 785
    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
786
    found_customer = Customer.where(balance: balance.amount).first
787 788 789 790 791 792
    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
793
    found_customer = Customer.where(gps_location: gps_location.gps_location).first
794 795 796 797 798 799
    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
800
    found_customer = Customer.where(address: address).first
801 802 803 804 805 806
    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
807
    found_customer = Customer.where(address: address, name: customers(:david).name).first
808 809 810
    assert_equal customers(:david), found_customer
  end

811
  def test_condition_utc_time_interpolation_with_default_timezone_local
812
    with_env_tz "America/New_York" do
813
      with_timezone_config default: :local do
814
        topic = Topic.first
815
        assert_equal topic, Topic.where(["written_on = ?", topic.written_on.getutc]).first
816 817 818 819 820
      end
    end
  end

  def test_hash_condition_utc_time_interpolation_with_default_timezone_local
821
    with_env_tz "America/New_York" do
822
      with_timezone_config default: :local do
823
        topic = Topic.first
824
        assert_equal topic, Topic.where(written_on: topic.written_on.getutc).first
825 826 827 828 829
      end
    end
  end

  def test_condition_local_time_interpolation_with_default_timezone_utc
830
    with_env_tz "America/New_York" do
831
      with_timezone_config default: :utc do
832
        topic = Topic.first
833
        assert_equal topic, Topic.where(["written_on = ?", topic.written_on.getlocal]).first
834 835 836 837 838
      end
    end
  end

  def test_hash_condition_local_time_interpolation_with_default_timezone_utc
839
    with_env_tz "America/New_York" do
840
      with_timezone_config default: :utc do
841
        topic = Topic.first
842
        assert_equal topic, Topic.where(written_on: topic.written_on.getlocal).first
843 844 845 846
      end
    end
  end

D
David Heinemeier Hansson 已提交
847
  def test_bind_variables
848 849 850 851
    assert_kind_of Firm, Company.where(["name = ?", "37signals"]).first
    assert_nil Company.where(["name = ?", "37signals!"]).first
    assert_nil Company.where(["name = ?", "37signals!' OR 1=1"]).first
    assert_kind_of Time, Topic.where(["id = ?", 1]).first.written_on
852
    assert_raise(ActiveRecord::PreparedStatementInvalid) {
853
      Company.where(["id=? AND name = ?", 2]).first
D
David Heinemeier Hansson 已提交
854
    }
855
    assert_raise(ActiveRecord::PreparedStatementInvalid) {
856
      Company.where(["id=?", 2, 3, 4]).first
D
David Heinemeier Hansson 已提交
857 858
    }
  end
859

D
David Heinemeier Hansson 已提交
860 861
  def test_bind_variables_with_quotes
    Company.create("name" => "37signals' go'es agains")
862
    assert Company.where(["name = ?", "37signals' go'es agains"]).first
D
David Heinemeier Hansson 已提交
863 864 865 866
  end

  def test_named_bind_variables_with_quotes
    Company.create("name" => "37signals' go'es agains")
867
    assert Company.where(["name = :name", {name: "37signals' go'es agains"}]).first
D
David Heinemeier Hansson 已提交
868 869 870
  end

  def test_named_bind_variables
871 872 873 874
    assert_kind_of Firm, Company.where(["name = :name", { name: "37signals" }]).first
    assert_nil Company.where(["name = :name", { name: "37signals!" }]).first
    assert_nil Company.where(["name = :name", { name: "37signals!' OR 1=1" }]).first
    assert_kind_of Time, Topic.where(["id = :id", { id: 1 }]).first.written_on
D
David Heinemeier Hansson 已提交
875 876 877
  end

  def test_string_sanitation
878 879
    assert_not_equal "'something ' 1=1'", ActiveRecord::Base.sanitize("something ' 1=1")
    assert_equal "'something; select table'", ActiveRecord::Base.sanitize("something; select table")
D
David Heinemeier Hansson 已提交
880 881 882 883 884 885 886 887 888
  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
889
    assert_equal topics(:first), Topic.find_by_title("The First Topic")
D
David Heinemeier Hansson 已提交
890 891
    assert_nil Topic.find_by_title("The First Topic!")
  end
J
Jeremy Kemper 已提交
892

893 894
  def test_find_by_one_attribute_bang
    assert_equal topics(:first), Topic.find_by_title!("The First Topic")
895 896 897
    assert_raises_with_message(ActiveRecord::RecordNotFound, "Couldn't find Topic") do
      Topic.find_by_title!("The First Topic!")
    end
898 899
  end

900
  def test_find_by_on_attribute_that_is_a_reserved_word
901
    dog_alias = "Dog"
902 903 904 905 906
    dog = Dog.create(alias: dog_alias)

    assert_equal dog, Dog.find_by_alias(dog_alias)
  end

907 908 909 910 911
  def test_find_by_one_attribute_that_is_an_alias
    assert_equal topics(:first), Topic.find_by_heading("The First Topic")
    assert_nil Topic.find_by_heading("The First Topic!")
  end

N
Nikita Afanasenko 已提交
912
  def test_find_by_one_attribute_bang_with_blank_defined
913 914
    blank_topic = BlankTopic.create(title: "The Blank One")
    assert_equal blank_topic, BlankTopic.find_by_title!("The Blank One")
N
Nikita Afanasenko 已提交
915 916
  end

917
  def test_find_by_one_attribute_with_conditions
918
    assert_equal accounts(:rails_core_account), Account.where("firm_id = ?", 6).find_by_credit_limit(50)
919 920
  end

921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954
  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

955 956
  def test_dynamic_finder_on_one_attribute_with_conditions_returns_same_results_after_caching
    # ensure this test can run independently of order
A
Akira Matsuda 已提交
957
    class << Account; self; end.send(:remove_method, :find_by_credit_limit) if Account.public_methods.include?(:find_by_credit_limit)
958 959
    a = Account.where("firm_id = ?", 6).find_by_credit_limit(50)
    assert_equal a, Account.where("firm_id = ?", 6).find_by_credit_limit(50) # find_by_credit_limit has been cached
960 961
  end

962
  def test_find_by_one_attribute_with_several_options
963
    assert_equal accounts(:unknown), Account.order("id DESC").where("id != ?", 3).find_by_credit_limit(50)
964
  end
965

D
David Heinemeier Hansson 已提交
966
  def test_find_by_one_missing_attribute
967
    assert_raise(NoMethodError) { Topic.find_by_undertitle("The First Topic!") }
D
David Heinemeier Hansson 已提交
968
  end
969

970
  def test_find_by_invalid_method_syntax
971 972 973 974
    assert_raise(NoMethodError) { Topic.fail_to_find_by_title("The First Topic") }
    assert_raise(NoMethodError) { Topic.find_by_title?("The First Topic") }
    assert_raise(NoMethodError) { Topic.fail_to_find_or_create_by_title("Nonexistent Title") }
    assert_raise(NoMethodError) { Topic.find_or_create_by_title?("Nonexistent Title") }
975
  end
D
David Heinemeier Hansson 已提交
976 977

  def test_find_by_two_attributes
978
    assert_equal topics(:first), Topic.find_by_title_and_author_name("The First Topic", "David")
D
David Heinemeier Hansson 已提交
979 980 981
    assert_nil Topic.find_by_title_and_author_name("The First Topic", "Mary")
  end

982 983 984 985
  def test_find_by_two_attributes_but_passing_only_one
    assert_raise(ArgumentError) { Topic.find_by_title_and_author_name("The First Topic") }
  end

L
Lauro Caetano 已提交
986
  def test_find_last_with_offset
987
    devs = Developer.order("id")
L
Lauro Caetano 已提交
988 989 990 991

    assert_equal devs[2], Developer.offset(2).first
    assert_equal devs[-3], Developer.offset(2).last
    assert_equal devs[-3], Developer.offset(2).last
992
    assert_equal devs[-3], Developer.offset(2).order("id DESC").first
L
Lauro Caetano 已提交
993 994
  end

D
David Heinemeier Hansson 已提交
995 996 997 998 999
  def test_find_by_nil_attribute
    topic = Topic.find_by_last_read nil
    assert_not_nil topic
    assert_nil topic.last_read
  end
1000

D
David Heinemeier Hansson 已提交
1001 1002 1003 1004 1005 1006
  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_with_bad_sql
1007
    assert_raise(ActiveRecord::StatementInvalid) { Topic.find_by_sql "select 1 from badtable" }
D
David Heinemeier Hansson 已提交
1008 1009
  end

1010
  def test_find_all_with_join
1011
    developers_on_project_one = Developer.
1012 1013
      joins("LEFT JOIN developers_projects ON developers.id = developers_projects.developer_id").
      where("project_id=1").to_a
1014
    assert_equal 3, developers_on_project_one.length
1015
    developer_names = developers_on_project_one.map(&:name)
1016 1017
    assert developer_names.include?("David")
    assert developer_names.include?("Jamis")
1018
  end
1019

1020
  def test_joins_dont_clobber_id
1021
    first = Firm.
1022 1023
      joins("INNER JOIN companies clients ON clients.firm_id = companies.id").
      where("companies.id = 1").first
1024 1025 1026
    assert_equal 1, first.id
  end

1027
  def test_joins_with_string_array
1028 1029 1030 1031
    person_with_reader_and_post = Post.
      joins(["INNER JOIN categorizations ON categorizations.post_id = posts.id",
             "INNER JOIN categories ON categories.id = categorizations.category_id AND categories.type = 'SpecialCategory'"
            ])
1032 1033 1034
    assert_equal 1, person_with_reader_and_post.size
  end

1035 1036
  def test_find_by_id_with_conditions_with_or
    assert_nothing_raised do
J
Jon Leighton 已提交
1037
      Post.where("posts.id <= 3 OR posts.#{QUOTED_TYPE} = 'Post'").find([1,2,3])
1038 1039
    end
  end
1040

1041
  def test_find_ignores_previously_inserted_record
1042
    Post.create!(title: "test", body: "it out")
J
Jon Leighton 已提交
1043
    assert_equal [], Post.where(id: nil)
1044 1045
  end

1046 1047 1048 1049 1050
  def test_find_by_empty_ids
    assert_equal [], Post.find([])
  end

  def test_find_by_empty_in_condition
1051
    assert_equal [], Post.where("id in (?)", [])
1052 1053 1054
  end

  def test_find_by_records
1055 1056 1057
    p1, p2 = Post.limit(2).order("id asc").to_a
    assert_equal [p1, p2], Post.where(["id in (?)", [p1, p2]]).order("id asc")
    assert_equal [p1, p2], Post.where(["id in (?)", [p1, p2.id]]).order("id asc")
1058 1059
  end

1060 1061 1062 1063 1064 1065 1066 1067 1068
  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
1069
    assert_equal ["1","2","3","4","5","6","7","8","9", "10", "11"], Company.connection.select_values("SELECT id FROM companies ORDER BY id").map!(&:to_s)
1070
    assert_equal ["37signals","Summit","Microsoft", "Flamboyant Software", "Ex Nihilo", "RailsCore", "Leetsoft", "Jadedpixel", "Odegy", "Ex Nihilo Part Deux", "Apex"], Company.connection.select_values("SELECT name FROM companies ORDER BY id")
1071 1072
  end

1073 1074
  def test_select_rows
    assert_equal(
1075
      [["1", "1", nil, "37signals"],
1076 1077
       ["2", "1", "2", "Summit"],
       ["3", "1", "1", "Microsoft"]],
1078
      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?}})
1079
    assert_equal [["1", "37signals"], ["2", "Summit"], ["3", "Microsoft"]],
1080
      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?}}
1081 1082
  end

1083
  def test_find_with_order_on_included_associations_with_construct_finder_sql_for_association_limiting_and_is_distinct
R
Fix...  
Ryuta Kamizono 已提交
1084 1085
    assert_equal 2, Post.includes(authors: :author_address).
      where.not(author_addresses: { id: nil }).
1086
      order("author_addresses.id DESC").limit(2).to_a.size
1087

1088
    assert_equal 3, Post.includes(author: :author_address, authors: :author_address).
R
Fix...  
Ryuta Kamizono 已提交
1089
      where.not(author_addresses_authors: { id: nil }).
1090
      order("author_addresses_authors.id DESC").limit(3).to_a.size
1091 1092
  end

1093
  def test_find_with_nil_inside_set_passed_for_one_attribute
1094 1095
    client_of = Company.
      where(client_of: [2, 1, nil],
1096 1097
            name: ["37signals", "Summit", "Microsoft"]).
      order("client_of DESC").
1098
      map(&:client_of)
1099

1100 1101
    assert client_of.include?(nil)
    assert_equal [2, 1].sort, client_of.compact.sort
1102 1103
  end

1104
  def test_find_with_nil_inside_set_passed_for_attribute
1105 1106
    client_of = Company.
      where(client_of: [nil]).
1107
      order("client_of DESC").
1108
      map(&:client_of)
1109 1110 1111 1112

    assert_equal [], client_of.compact
  end

1113
  def test_with_limiting_with_custom_select
1114
    posts = Post.references(:authors).merge(
1115 1116
      includes: :author, select: 'posts.*, authors.id as "author_id"',
      limit: 3, order: "posts.id"
1117
    ).to_a
1118 1119
    assert_equal 3, posts.size
    assert_equal [0, 1, 1], posts.map(&:author_id).sort
1120
  end
1121

1122
  def test_find_one_message_with_custom_primary_key
1123 1124 1125
    table_with_custom_primary_key do |model|
      model.primary_key = :name
      e = assert_raises(ActiveRecord::RecordNotFound) do
1126
        model.find "Hello World!"
1127 1128 1129 1130 1131 1132 1133 1134 1135
      end
      assert_equal %Q{Couldn't find MercedesCar with 'name'=Hello World!}, e.message
    end
  end

  def test_find_some_message_with_custom_primary_key
    table_with_custom_primary_key do |model|
      model.primary_key = :name
      e = assert_raises(ActiveRecord::RecordNotFound) do
1136
        model.find "Hello", "World!"
1137 1138
      end
      assert_equal %Q{Couldn't find all MercedesCars with 'name': (Hello, World!) (found 0 results, but was looking for 2)}, e.message
1139 1140 1141
    end
  end

1142 1143 1144 1145 1146 1147
  def test_find_without_primary_key
    assert_raises(ActiveRecord::UnknownPrimaryKey) do
      Matey.find(1)
    end
  end

1148
  def test_finder_with_offset_string
1149
    assert_nothing_raised { Topic.offset("3").to_a }
1150 1151
  end

1152 1153 1154 1155 1156 1157 1158 1159 1160
  test "find_by with hash conditions returns the first matching record" do
    assert_equal posts(:eager_other), Post.find_by(id: posts(:eager_other).id)
  end

  test "find_by with non-hash conditions returns the first matching record" do
    assert_equal posts(:eager_other), Post.find_by("id = #{posts(:eager_other).id}")
  end

  test "find_by with multi-arg conditions returns the first matching record" do
1161
    assert_equal posts(:eager_other), Post.find_by("id = ?", posts(:eager_other).id)
1162 1163 1164 1165 1166 1167
  end

  test "find_by returns nil if the record is missing" do
    assert_equal nil, Post.find_by("1 = 0")
  end

1168 1169 1170 1171 1172
  test "find_by with associations" do
    assert_equal authors(:david), Post.find_by(author: authors(:david)).author
    assert_equal authors(:mary) , Post.find_by(author: authors(:mary) ).author
  end

1173 1174 1175 1176
  test "find_by doesn't have implicit ordering" do
    assert_sql(/^((?!ORDER).)*$/) { Post.find_by(id: posts(:eager_other).id) }
  end

1177 1178 1179 1180 1181 1182 1183 1184 1185
  test "find_by! with hash conditions returns the first matching record" do
    assert_equal posts(:eager_other), Post.find_by!(id: posts(:eager_other).id)
  end

  test "find_by! with non-hash conditions returns the first matching record" do
    assert_equal posts(:eager_other), Post.find_by!("id = #{posts(:eager_other).id}")
  end

  test "find_by! with multi-arg conditions returns the first matching record" do
1186
    assert_equal posts(:eager_other), Post.find_by!("id = ?", posts(:eager_other).id)
1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198
  end

  test "find_by! doesn't have implicit ordering" do
    assert_sql(/^((?!ORDER).)*$/) { Post.find_by!(id: posts(:eager_other).id) }
  end

  test "find_by! raises RecordNotFound if the record is missing" do
    assert_raises(ActiveRecord::RecordNotFound) do
      Post.find_by!("1 = 0")
    end
  end

1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218
  test "find on a scope does not perform statement caching" do
    honda = cars(:honda)
    zyke = cars(:zyke)
    tyre = honda.tyres.create!
    tyre2 = zyke.tyres.create!

    assert_equal tyre, honda.tyres.custom_find(tyre.id)
    assert_equal tyre2, zyke.tyres.custom_find(tyre2.id)
  end

  test "find_by on a scope does not perform statement caching" do
    honda = cars(:honda)
    zyke = cars(:zyke)
    tyre = honda.tyres.create!
    tyre2 = zyke.tyres.create!

    assert_equal tyre, honda.tyres.custom_find_by(id: tyre.id)
    assert_equal tyre2, zyke.tyres.custom_find_by(id: tyre2.id)
  end

D
David Heinemeier Hansson 已提交
1219
  protected
1220 1221 1222
    def table_with_custom_primary_key
      yield(Class.new(Toy) do
        def self.name
1223
          "MercedesCar"
1224 1225 1226
        end
      end)
    end
1227 1228 1229 1230 1231 1232

    def assert_raises_with_message(exception_class, message, &block)
      err = assert_raises(exception_class) { block.call }
      assert_match message, err.message
    end

D
David Heinemeier Hansson 已提交
1233
end