relations_test.rb 29.0 KB
Newer Older
1
require "cases/helper"
2
require 'models/tag'
3
require 'models/tagging'
4 5
require 'models/post'
require 'models/topic'
6
require 'models/comment'
7 8
require 'models/reply'
require 'models/author'
9
require 'models/comment'
10 11 12
require 'models/entrant'
require 'models/developer'
require 'models/company'
13
require 'models/bird'
14 15 16
require 'models/car'
require 'models/engine'
require 'models/tyre'
17
require 'models/minivan'
18

19

E
Emilio Tagua 已提交
20
class RelationTest < ActiveRecord::TestCase
21
  fixtures :authors, :topics, :entrants, :developers, :companies, :developers_projects, :accounts, :categories, :categorizations, :posts, :comments,
22 23 24 25 26 27 28
    :tags, :taggings, :cars, :minivans

  def test_do_not_double_quote_string_id
    van = Minivan.last
    assert van
    assert_equal van.id, Minivan.where(:minivan_id => van).to_a.first.minivan_id
  end
29

30 31 32 33 34 35
  def test_do_not_double_quote_string_id_with_array
    van = Minivan.last
    assert van
    assert_equal van, Minivan.where(:minivan_id => [van]).to_a.first
  end

A
Aaron Patterson 已提交
36 37 38 39 40 41 42 43 44
  def test_bind_values
    relation = Post.scoped
    assert_equal [], relation.bind_values

    relation2 = relation.bind 'foo'
    assert_equal %w{ foo }, relation2.bind_values
    assert_equal [], relation.bind_values
  end

45
  def test_two_scopes_with_includes_should_not_drop_any_include
46
    car = Car.incl_engines.incl_tyres.first
47 48
    assert_no_queries { car.tyres.length }
    assert_no_queries { car.engines.length }
49
  end
50

51 52 53 54 55
  def test_dynamic_finder
    x = Post.where('author_id = ?', 1)
    assert x.klass.respond_to?(:find_by_id), '@klass should handle dynamic finders'
  end

56 57 58 59 60
  def test_multivalue_where
    posts = Post.where('author_id = ? AND id = ?', 1, 1)
    assert_equal 1, posts.to_a.size
  end

61 62 63 64 65 66
  def test_scoped
    topics = Topic.scoped
    assert_kind_of ActiveRecord::Relation, topics
    assert_equal 4, topics.size
  end

67 68 69 70 71 72 73 74 75 76
  def test_to_json
    assert_nothing_raised  { Bird.scoped.to_json }
    assert_nothing_raised  { Bird.scoped.all.to_json }
  end

  def test_to_yaml
    assert_nothing_raised  { Bird.scoped.to_yaml }
    assert_nothing_raised  { Bird.scoped.all.to_yaml }
  end

77 78 79 80 81
  def test_to_xml
    assert_nothing_raised  { Bird.scoped.to_xml }
    assert_nothing_raised  { Bird.scoped.all.to_xml }
  end

82 83 84 85 86 87
  def test_scoped_all
    topics = Topic.scoped.all
    assert_kind_of Array, topics
    assert_no_queries { assert_equal 4, topics.size }
  end

P
Pratik Naik 已提交
88 89 90 91 92 93 94 95 96 97 98
  def test_loaded_all
    topics = Topic.scoped

    assert_queries(1) do
      2.times { assert_equal 4, topics.all.size }
    end

    assert topics.loaded?
  end

  def test_scoped_first
P
Pratik Naik 已提交
99
    topics = Topic.scoped.order('id ASC')
P
Pratik Naik 已提交
100 101 102 103 104 105 106 107 108

    assert_queries(1) do
      2.times { assert_equal "The First Topic", topics.first.title }
    end

    assert ! topics.loaded?
  end

  def test_loaded_first
P
Pratik Naik 已提交
109
    topics = Topic.scoped.order('id ASC')
P
Pratik Naik 已提交
110 111 112 113 114 115 116 117 118

    assert_queries(1) do
      topics.all # force load
      2.times { assert_equal "The First Topic", topics.first.title }
    end

    assert topics.loaded?
  end

119 120 121 122 123 124 125 126 127
  def test_reload
    topics = Topic.scoped

    assert_queries(1) do
      2.times { topics.to_a }
    end

    assert topics.loaded?

128 129
    original_size = topics.to_a.size
    Topic.create! :title => 'fake'
130

131 132 133
    assert_queries(1) { topics.reload }
    assert_equal original_size + 1, topics.size
    assert topics.loaded?
134 135
  end

136
  def test_finding_with_conditions
137 138
    assert_equal ["David"], Author.where(:name => 'David').map(&:name)
    assert_equal ['Mary'],  Author.where(["name = ?", 'Mary']).map(&:name)
139
    assert_equal ['Mary'],  Author.where("name = ?", 'Mary').map(&:name)
140 141 142
  end

  def test_finding_with_order
143
    topics = Topic.order('id')
144
    assert_equal 4, topics.to_a.size
145 146 147
    assert_equal topics(:first).title, topics.first.title
  end

148 149 150 151 152 153 154 155 156 157 158 159

  def test_finding_with_arel_order
    topics = Topic.order(Topic.arel_table[:id].asc)
    assert_equal 4, topics.to_a.size
    assert_equal topics(:first).title, topics.first.title
  end

  def test_finding_last_with_arel_order
    topics = Topic.order(Topic.arel_table[:id].asc)
    assert_equal topics(:fourth).title, topics.last.title
  end

160 161 162 163 164 165
  def test_finding_with_order_concatenated
    topics = Topic.order('author_name').order('title')
    assert_equal 4, topics.to_a.size
    assert_equal topics(:fourth).title, topics.first.title
  end

S
Sebastian Martinez 已提交
166 167 168 169 170
  def test_finding_with_reorder
    topics = Topic.order('author_name').order('title').reorder('id').all
    topics_titles = topics.map{ |t| t.title }
    assert_equal ['The First Topic', 'The Second Topic of the day', 'The Third Topic of the day', 'The Fourth Topic of the day'], topics_titles
  end
171

172
  def test_finding_with_order_and_take
173
    entrants = Entrant.order("id ASC").limit(2).to_a
174

P
Pratik Naik 已提交
175 176
    assert_equal 2, entrants.size
    assert_equal entrants(:first).name, entrants.first.name
177 178
  end

179
  def test_finding_with_cross_table_order_and_limit
180 181 182
    tags = Tag.includes(:taggings).
              order("tags.name asc", "taggings.taggable_id asc", "REPLACE('abc', taggings.taggable_type, taggings.taggable_type)").
              limit(1).to_a
183 184 185
    assert_equal 1, tags.length
  end

186
  def test_finding_with_complex_order_and_limit
187
    tags = Tag.includes(:taggings).order("REPLACE('abc', taggings.taggable_type, taggings.taggable_type)").limit(1).to_a
188 189 190 191
    assert_equal 1, tags.length
  end

  def test_finding_with_complex_order
192
    tags = Tag.includes(:taggings).order("REPLACE('abc', taggings.taggable_type, taggings.taggable_type)").to_a
193
    assert_equal 3, tags.length
194 195
  end

196
  def test_finding_with_order_limit_and_offset
197
    entrants = Entrant.order("id ASC").limit(2).offset(1)
198

199
    assert_equal 2, entrants.to_a.size
P
Pratik Naik 已提交
200
    assert_equal entrants(:second).name, entrants.first.name
201

202
    entrants = Entrant.order("id ASC").limit(2).offset(2)
203
    assert_equal 1, entrants.to_a.size
P
Pratik Naik 已提交
204
    assert_equal entrants(:third).name, entrants.first.name
205 206 207
  end

  def test_finding_with_group
208
    developers = Developer.group("salary").select("salary").to_a
209 210 211 212
    assert_equal 4, developers.size
    assert_equal 4, developers.map(&:salary).uniq.size
  end

213 214
  def test_select_with_block
    even_ids = Developer.scoped.select {|d| d.id % 2 == 0 }.map(&:id)
215
    assert_equal [2, 4, 6, 8, 10], even_ids.sort
216 217
  end

218 219 220 221
  def test_joins_with_nil_argument
    assert_nothing_raised { DependentFirm.joins(nil).first }
  end

222
  def test_finding_with_hash_conditions_on_joined_table
223
    firms = DependentFirm.joins(:account).where({:name => 'RailsCore', :accounts => { :credit_limit => 55..60 }}).to_a
224 225 226 227 228
    assert_equal 1, firms.size
    assert_equal companies(:rails_core), firms.first
  end

  def test_find_all_with_join
229
    developers_on_project_one = Developer.joins('LEFT JOIN developers_projects ON developers.id = developers_projects.developer_id').
230
      where('project_id=1').to_a
231 232 233 234 235 236 237 238

    assert_equal 3, developers_on_project_one.length
    developer_names = developers_on_project_one.map { |d| d.name }
    assert developer_names.include?('David')
    assert developer_names.include?('Jamis')
  end

  def test_find_on_hash_conditions
239
    assert_equal Topic.find(:all, :conditions => {:approved => false}), Topic.where({ :approved => false }).to_a
240 241 242
  end

  def test_joins_with_string_array
243
    person_with_reader_and_post = Post.joins([
244 245 246 247 248 249
        "INNER JOIN categorizations ON categorizations.post_id = posts.id",
        "INNER JOIN categories ON categories.id = categorizations.category_id AND categories.type = 'SpecialCategory'"
      ]
    ).to_a
    assert_equal 1, person_with_reader_and_post.size
  end
250

251 252
  def test_scoped_responds_to_delegated_methods
    relation = Topic.scoped
253 254

    ["map", "uniq", "sort", "insert", "delete", "update"].each do |method|
255
      assert_respond_to relation, method, "Topic.scoped should respond to #{method.inspect}"
256 257 258
    end
  end

259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277
  def test_respond_to_delegates_to_relation
    relation = Topic.scoped
    fake_arel = Struct.new(:responds) {
      def respond_to? method, access = false
        responds << [method, access]
      end
    }.new []

    relation.extend(Module.new { attr_accessor :arel })
    relation.arel = fake_arel

    relation.respond_to?(:matching_attributes)
    assert_equal [:matching_attributes, false], fake_arel.responds.first

    fake_arel.responds = []
    relation.respond_to?(:matching_attributes, true)
    assert_equal [:matching_attributes, true], fake_arel.responds.first
  end

278 279 280 281
  def test_respond_to_dynamic_finders
    relation = Topic.scoped

    ["find_by_title", "find_by_title_and_author_name", "find_or_create_by_title", "find_or_initialize_by_title_and_author_name"].each do |method|
282
      assert_respond_to relation, method, "Topic.scoped should respond to #{method.inspect}"
283 284
    end
  end
285

286
  def test_respond_to_class_methods_and_scopes
287 288 289 290
    assert DeveloperOrderedBySalary.scoped.respond_to?(:all_ordered_by_name)
    assert Topic.scoped.respond_to?(:by_lifo)
  end

291
  def test_find_with_readonly_option
292 293
    Developer.scoped.each { |d| assert !d.readonly? }
    Developer.scoped.readonly.each { |d| assert d.readonly? }
294
  end
295 296

  def test_eager_association_loading_of_stis_with_multiple_references
297
    authors = Author.eager_load(:posts => { :special_comments => { :post => [ :special_comments, :very_special_comment ] } }).
298
      order('comments.body, very_special_comments_posts.body').where('posts.id = 4').to_a
299

300 301 302 303 304 305 306
    assert_equal [authors(:david)], authors
    assert_no_queries do
      authors.first.posts.first.special_comments.first.post.special_comments
      authors.first.posts.first.special_comments.first.post.very_special_comment
    end
  end

307
  def test_find_with_preloaded_associations
308
    assert_queries(2) do
309
      posts = Post.preload(:comments).order('posts.id')
310
      assert posts.first.comments.first
311
    end
312

313
    assert_queries(ActiveRecord::IdentityMap.enabled? ? 1 : 2) do
314
      posts = Post.preload(:comments).order('posts.id')
315
      assert posts.first.comments.first
316
    end
317

318
    assert_queries(2) do
319
      posts = Post.preload(:author).order('posts.id')
320
      assert posts.first.author
321
    end
322

323
    assert_queries(ActiveRecord::IdentityMap.enabled? ? 1 : 2) do
324
      posts = Post.preload(:author).order('posts.id')
325 326 327
      assert posts.first.author
    end

328
    assert_queries(ActiveRecord::IdentityMap.enabled? ? 1 : 3) do
329
      posts = Post.preload(:author, :comments).order('posts.id')
330
      assert posts.first.author
331 332 333 334 335 336
      assert posts.first.comments.first
    end
  end

  def test_find_with_included_associations
    assert_queries(2) do
337
      posts = Post.includes(:comments).order('posts.id')
338 339 340
      assert posts.first.comments.first
    end

341
    assert_queries(ActiveRecord::IdentityMap.enabled? ? 1 : 2) do
342
      posts = Post.scoped.includes(:comments).order('posts.id')
343 344 345 346
      assert posts.first.comments.first
    end

    assert_queries(2) do
347
      posts = Post.includes(:author).order('posts.id')
348 349 350
      assert posts.first.author
    end

351
    assert_queries(ActiveRecord::IdentityMap.enabled? ? 1 : 3) do
352
      posts = Post.includes(:author, :comments).order('posts.id')
353
      assert posts.first.author
354
      assert posts.first.comments.first
355 356
    end
  end
357

P
Pratik Naik 已提交
358
  def test_default_scope_with_conditions_string
359
    assert_equal Developer.find_all_by_name('David').map(&:id).sort, DeveloperCalledDavid.scoped.map(&:id).sort
360
    assert_nil DeveloperCalledDavid.create!.name
361 362 363
  end

  def test_default_scope_with_conditions_hash
364
    assert_equal Developer.find_all_by_name('Jamis').map(&:id).sort, DeveloperCalledJamis.scoped.map(&:id).sort
365 366 367
    assert_equal 'Jamis', DeveloperCalledJamis.create!.name
  end

368 369 370 371 372
  def test_default_scoping_finder_methods
    developers = DeveloperCalledDavid.order('id').map(&:id).sort
    assert_equal Developer.find_all_by_name('David').map(&:id).sort, developers
  end

373 374
  def test_loading_with_one_association
    posts = Post.preload(:comments)
375 376 377 378
    post = posts.find { |p| p.id == 1 }
    assert_equal 2, post.comments.size
    assert post.comments.include?(comments(:greetings))

379
    post = Post.where("posts.title = 'Welcome to the weblog'").preload(:comments).first
380 381 382
    assert_equal 2, post.comments.size
    assert post.comments.include?(comments(:greetings))

383
    posts = Post.preload(:last_comment)
384 385 386 387 388
    post = posts.find { |p| p.id == 1 }
    assert_equal Post.find(1).last_comment, post.last_comment
  end

  def test_loading_with_one_association_with_non_preload
389
    posts = Post.eager_load(:last_comment).order('comments.id DESC')
390 391 392
    post = posts.find { |p| p.id == 1 }
    assert_equal Post.find(1).last_comment, post.last_comment
  end
393

394 395 396 397 398 399 400 401 402
  def test_dynamic_find_by_attributes_should_yield_found_object
    david = authors(:david)
    yielded_value = nil
    Author.find_by_name(david.name) do |author|
      yielded_value = author
    end
    assert_equal david, yielded_value
  end

403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420
  def test_dynamic_find_by_attributes
    david = authors(:david)
    author = Author.preload(:taggings).find_by_id(david.id)
    expected_taggings = taggings(:welcome_general, :thinking_general)

    assert_no_queries do
      assert_equal expected_taggings, author.taggings.uniq.sort_by { |t| t.id }
    end

    authors = Author.scoped
    assert_equal david, authors.find_by_id_and_name(david.id, david.name)
    assert_equal david, authors.find_by_id_and_name!(david.id, david.name)
  end

  def test_dynamic_find_by_attributes_bang
    author = Author.scoped.find_by_id!(authors(:david).id)
    assert_equal "David", author.name

P
Pratik Naik 已提交
421
    assert_raises(ActiveRecord::RecordNotFound) { Author.scoped.find_by_id_and_name!(20, 'invalid') }
422 423 424 425 426 427 428 429 430
  end

  def test_dynamic_find_all_by_attributes
    authors = Author.scoped

    davids = authors.find_all_by_name('David')
    assert_kind_of Array, davids
    assert_equal [authors(:david)], davids
  end
431 432 433 434 435 436

  def test_dynamic_find_or_initialize_by_attributes
    authors = Author.scoped

    lifo = authors.find_or_initialize_by_name('Lifo')
    assert_equal "Lifo", lifo.name
437
    assert !lifo.persisted?
438 439 440 441 442 443 444 445 446

    assert_equal authors(:david), authors.find_or_initialize_by_name(:name => 'David')
  end

  def test_dynamic_find_or_create_by_attributes
    authors = Author.scoped

    lifo = authors.find_or_create_by_name('Lifo')
    assert_equal "Lifo", lifo.name
447
    assert lifo.persisted?
448 449 450 451

    assert_equal authors(:david), authors.find_or_create_by_name(:name => 'David')
  end

P
Pratik Naik 已提交
452 453 454 455 456 457
  def test_find_id
    authors = Author.scoped

    david = authors.find(authors(:david).id)
    assert_equal 'David', david.name

P
Pratik Naik 已提交
458
    assert_raises(ActiveRecord::RecordNotFound) { authors.where(:name => 'lifo').find('42') }
P
Pratik Naik 已提交
459
  end
460

P
Pratik Naik 已提交
461 462 463 464 465 466 467 468 469 470
  def test_find_ids
    authors = Author.order('id ASC')

    results = authors.find(authors(:david).id, authors(:mary).id)
    assert_kind_of Array, results
    assert_equal 2, results.size
    assert_equal 'David', results[0].name
    assert_equal 'Mary', results[1].name
    assert_equal results, authors.find([authors(:david).id, authors(:mary).id])

P
Pratik Naik 已提交
471 472
    assert_raises(ActiveRecord::RecordNotFound) { authors.where(:name => 'lifo').find(authors(:david).id, '42') }
    assert_raises(ActiveRecord::RecordNotFound) { authors.find(['42', 43]) }
P
Pratik Naik 已提交
473 474
  end

475 476
  def test_find_in_empty_array
    authors = Author.scoped.where(:id => [])
477
    assert_blank authors.all
478 479
  end

480 481 482 483 484 485
  def test_where_with_ar_object
    author = Author.first
    authors = Author.scoped.where(:id => author)
    assert_equal 1, authors.all.length
  end

486 487 488 489 490 491
  def test_find_with_list_of_ar
    author = Author.first
    authors = Author.find([author])
    assert_equal author, authors.first
  end

492 493 494 495 496 497 498
  class Mary < Author; end

  def test_find_by_classname
    Author.create!(:name => Mary.name)
    assert_equal 1, Author.where(:name => Mary).size
  end

499 500 501 502 503 504
  def test_find_by_id_with_list_of_ar
    author = Author.first
    authors = Author.find_by_id([author])
    assert_equal author, authors
  end

505 506 507 508 509 510
  def test_find_all_using_where_twice_should_or_the_relation
    david = authors(:david)
    relation = Author.unscoped
    relation = relation.where(:name => david.name)
    relation = relation.where(:name => 'Santiago')
    relation = relation.where(:id => david.id)
A
Aaron Patterson 已提交
511
    assert_equal [], relation.all
512 513
  end

A
Aaron Patterson 已提交
514 515 516 517 518 519 520 521
  def test_multi_where_ands_queries
    relation = Author.unscoped
    david = authors(:david)
    sql = relation.where(:name => david.name).where(:name => 'Santiago').to_sql
    assert_match('AND', sql)
  end

  def test_find_all_with_multiple_should_use_and
A
Aaron Patterson 已提交
522 523 524 525 526 527 528 529
    david = authors(:david)
    relation = [
      { :name => david.name },
      { :name => 'Santiago' },
      { :name => 'tenderlove' },
    ].inject(Author.unscoped) do |memo, param|
      memo.where(param)
    end
A
Aaron Patterson 已提交
530
    assert_equal [], relation.all
A
Aaron Patterson 已提交
531
  end
A
Aaron Patterson 已提交
532

533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550
  def test_find_all_using_where_with_relation
    david = authors(:david)
    # switching the lines below would succeed in current rails
    # assert_queries(2) {
    assert_queries(1) {
      relation = Author.where(:id => Author.where(:id => david.id))
      assert_equal [david], relation.all
    }
  end

  def test_find_all_using_where_with_relation_with_joins
    david = authors(:david)
    assert_queries(1) {
      relation = Author.where(:id => Author.joins(:posts).where(:id => david.id))
      assert_equal [david], relation.all
    }
  end

A
Aaron Patterson 已提交
551

552 553 554 555 556 557
  def test_find_all_using_where_with_relation_with_select_to_build_subquery
    david = authors(:david)
    assert_queries(1) {
      relation = Author.where(:name => Author.where(:id => david.id).select(:name))
      assert_equal [david], relation.all
    }
A
Aaron Patterson 已提交
558 559
  end

P
Pratik Naik 已提交
560 561 562 563 564
  def test_exists
    davids = Author.where(:name => 'David')
    assert davids.exists?
    assert davids.exists?(authors(:david).id)
    assert ! davids.exists?(authors(:mary).id)
P
Pratik Naik 已提交
565 566
    assert ! davids.exists?("42")
    assert ! davids.exists?(42)
P
Pratik Naik 已提交
567 568 569 570 571

    fake  = Author.where(:name => 'fake author')
    assert ! fake.exists?
    assert ! fake.exists?(authors(:david).id)
  end
572 573 574

  def test_last
    authors = Author.scoped
575
    assert_equal authors(:bob), authors.last
576 577
  end

P
Pratik Naik 已提交
578 579 580 581 582 583 584 585 586 587 588 589 590
  def test_destroy_all
    davids = Author.where(:name => 'David')

    # Force load
    assert_equal [authors(:david)], davids.to_a
    assert davids.loaded?

    assert_difference('Author.count', -1) { davids.destroy_all }

    assert_equal [], davids.to_a
    assert davids.loaded?
  end

P
Pratik Naik 已提交
591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610
  def test_delete_all
    davids = Author.where(:name => 'David')

    assert_difference('Author.count', -1) { davids.delete_all }
    assert ! davids.loaded?
  end

  def test_delete_all_loaded
    davids = Author.where(:name => 'David')

    # Force load
    assert_equal [authors(:david)], davids.to_a
    assert davids.loaded?

    assert_difference('Author.count', -1) { davids.delete_all }

    assert_equal [], davids.to_a
    assert davids.loaded?
  end

611 612 613 614
  def test_select_argument_error
    assert_raises(ArgumentError) { Developer.select }
  end

615
  def test_relation_merging
616
    devs = Developer.where("salary >= 80000").merge(Developer.limit(2)).merge(Developer.order('id ASC').where("id < 3"))
617 618
    assert_equal [developers(:david), developers(:jamis)], devs.to_a

619
    dev_with_count = Developer.limit(1).merge(Developer.order('id DESC')).merge(Developer.select('developers.*'))
620
    assert_equal [developers(:poor_jamis)], dev_with_count.to_a
621 622 623 624
  end

  def test_relation_merging_with_eager_load
    relations = []
625 626
    relations << Post.order('comments.id DESC').merge(Post.eager_load(:last_comment)).merge(Post.scoped)
    relations << Post.eager_load(:last_comment).merge(Post.order('comments.id DESC')).merge(Post.scoped)
627

628 629 630 631 632 633
    relations.each do |posts|
      post = posts.find { |p| p.id == 1 }
      assert_equal Post.find(1).last_comment, post.last_comment
    end
  end

634
  def test_relation_merging_with_locks
635
    devs = Developer.lock.where("salary >= 80000").order("id DESC").merge(Developer.limit(2))
636
    assert_present devs.locked
637 638
  end

639
  def test_relation_merging_with_preload
M
Marcin Raczkowski 已提交
640
    ActiveRecord::IdentityMap.without do
641 642 643
      [Post.scoped.merge(Post.preload(:author)), Post.preload(:author).merge(Post.scoped)].each do |posts|
        assert_queries(2) { assert posts.first.author }
      end
M
Marcin Raczkowski 已提交
644
    end
645
  end
646

647
  def test_relation_merging_with_joins
648
    comments = Comment.joins(:post).where(:body => 'Thank you for the welcome').merge(Post.where(:body => 'Such a lovely day'))
649 650 651
    assert_equal 1, comments.count
  end

P
Pratik Naik 已提交
652 653 654
  def test_count
    posts = Post.scoped

655 656 657
    assert_equal 11, posts.count
    assert_equal 11, posts.count(:all)
    assert_equal 11, posts.count(:id)
P
Pratik Naik 已提交
658 659

    assert_equal 1, posts.where('comments_count > 1').count
660
    assert_equal 9, posts.where(:comments_count => 0).count
P
Pratik Naik 已提交
661 662 663 664 665 666
  end

  def test_count_with_distinct
    posts = Post.scoped

    assert_equal 3, posts.count(:comments_count, :distinct => true)
667
    assert_equal 11, posts.count(:comments_count, :distinct => false)
P
Pratik Naik 已提交
668 669

    assert_equal 3, posts.select(:comments_count).count(:distinct => true)
670
    assert_equal 11, posts.select(:comments_count).count(:distinct => false)
P
Pratik Naik 已提交
671 672 673 674 675 676
  end

  def test_count_explicit_columns
    Post.update_all(:comments_count => nil)
    posts = Post.scoped

P
Pratik Naik 已提交
677
    assert_equal [0], posts.select('comments_count').where('id is not null').group('id').order('id').count.values.uniq
678
    assert_equal 0, posts.where('id is not null').select('comments_count').count
679

680
    assert_equal 11, posts.select('comments_count').count('id')
P
Pratik Naik 已提交
681 682 683 684
    assert_equal 0, posts.select('comments_count').count
    assert_equal 0, posts.count(:comments_count)
    assert_equal 0, posts.count('comments_count')
  end
685

686 687 688 689 690 691
  def test_multiple_selects
    post = Post.scoped.select('comments_count').select('title').order("id ASC").first
    assert_equal "Welcome to the weblog", post.title
    assert_equal 2, post.comments_count
  end

692 693 694
  def test_size
    posts = Post.scoped

695
    assert_queries(1) { assert_equal 11, posts.size }
696 697 698 699
    assert ! posts.loaded?

    best_posts = posts.where(:comments_count => 0)
    best_posts.to_a # force load
700
    assert_no_queries { assert_equal 9, best_posts.size }
701 702
  end

703
  def test_size_with_limit
704
    posts = Post.limit(10)
705

706
    assert_queries(1) { assert_equal 10, posts.size }
707 708 709 710
    assert ! posts.loaded?

    best_posts = posts.where(:comments_count => 0)
    best_posts.to_a # force load
711
    assert_no_queries { assert_equal 9, best_posts.size }
712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730
  end

  def test_size_with_zero_limit
    posts = Post.limit(0)

    assert_no_queries { assert_equal 0, posts.size }
    assert ! posts.loaded?

    posts.to_a # force load
    assert_no_queries { assert_equal 0, posts.size }
  end

  def test_empty_with_zero_limit
    posts = Post.limit(0)

    assert_no_queries { assert_equal true, posts.empty? }
    assert ! posts.loaded?
  end

731 732 733 734 735 736 737
  def test_count_complex_chained_relations
    posts = Post.select('comments_count').where('id is not null').group("author_id").where("comments_count > 0")

    expected = { 1 => 2 }
    assert_equal expected, posts.count
  end

738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754
  def test_empty
    posts = Post.scoped

    assert_queries(1) { assert_equal false, posts.empty? }
    assert ! posts.loaded?

    no_posts = posts.where(:title => "")
    assert_queries(1) { assert_equal true, no_posts.empty? }
    assert ! no_posts.loaded?

    best_posts = posts.where(:comments_count => 0)
    best_posts.to_a # force load
    assert_no_queries { assert_equal false, best_posts.empty? }
  end

  def test_empty_complex_chained_relations
    posts = Post.select("comments_count").where("id is not null").group("author_id").where("comments_count > 0")
A
Andrew White 已提交
755

756 757 758 759 760 761 762 763
    assert_queries(1) { assert_equal false, posts.empty? }
    assert ! posts.loaded?

    no_posts = posts.where(:title => "")
    assert_queries(1) { assert_equal true, no_posts.empty? }
    assert ! no_posts.loaded?
  end

P
Pratik Naik 已提交
764 765 766
  def test_any
    posts = Post.scoped

767 768 769 770 771 772 773 774
    # This test was failing when run on its own (as opposed to running the entire suite).
    # The second line in the assert_queries block was causing visit_Arel_Attributes_Attribute
    # in Arel::Visitors::ToSql to trigger a SHOW TABLES query. Running that line here causes
    # the SHOW TABLES result to be cached so we don't have to do it again in the block.
    #
    # This is obviously a rubbish fix but it's the best I can come up with for now...
    posts.where(:id => nil).any?

P
Pratik Naik 已提交
775 776 777 778 779 780 781 782 783 784 785 786 787
    assert_queries(3) do
      assert posts.any? # Uses COUNT()
      assert ! posts.where(:id => nil).any?

      assert posts.any? {|p| p.id > 0 }
      assert ! posts.any? {|p| p.id <= 0 }
    end

    assert posts.loaded?
  end

  def test_many
    posts = Post.scoped
788

P
Pratik Naik 已提交
789 790 791 792 793
    assert_queries(2) do
      assert posts.many? # Uses COUNT()
      assert posts.many? {|p| p.id > 0 }
      assert ! posts.many? {|p| p.id < 2 }
    end
794

P
Pratik Naik 已提交
795 796 797
    assert posts.loaded?
  end

798 799 800 801 802 803
  def test_many_with_limits
    posts = Post.scoped

    assert posts.many?
    assert ! posts.limit(1).many?
  end
P
Pratik Naik 已提交
804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819

  def test_build
    posts = Post.scoped

    post = posts.new
    assert_kind_of Post, post
  end

  def test_scoped_build
    posts = Post.where(:title => 'You told a lie')

    post = posts.new
    assert_kind_of Post, post
    assert_equal 'You told a lie', post.title
  end

820 821 822 823 824
  def test_create
    birds = Bird.scoped

    sparrow = birds.create
    assert_kind_of Bird, sparrow
825
    assert !sparrow.persisted?
826 827

    hen = birds.where(:name => 'hen').create
828
    assert hen.persisted?
829 830 831 832 833 834 835 836 837 838
    assert_equal 'hen', hen.name
  end

  def test_create_bang
    birds = Bird.scoped

    assert_raises(ActiveRecord::RecordInvalid) { birds.create! }

    hen = birds.where(:name => 'hen').create!
    assert_kind_of Bird, hen
839
    assert hen.persisted?
840 841
    assert_equal 'hen', hen.name
  end
P
Pratik Naik 已提交
842

843 844 845 846 847 848 849 850
  def test_explicit_create_scope
    hens = Bird.where(:name => 'hen')
    assert_equal 'hen', hens.new.name

    hens = hens.create_with(:name => 'cock')
    assert_equal 'cock', hens.new.name
  end

P
Pratik Naik 已提交
851 852 853 854 855 856 857 858 859 860 861
  def test_except
    relation = Post.where(:author_id => 1).order('id ASC').limit(1)
    assert_equal [posts(:welcome)], relation.all

    author_posts = relation.except(:order, :limit)
    assert_equal Post.where(:author_id => 1).all, author_posts.all

    all_posts = relation.except(:where, :order, :limit)
    assert_equal Post.all, all_posts.all
  end

862 863 864 865
  def test_extensions_with_except
    assert_equal 2, Topic.named_extension.order(:author_name).except(:order).two
  end

E
Emilio Tagua 已提交
866 867 868 869 870 871 872 873 874 875 876
  def test_only
    relation = Post.where(:author_id => 1).order('id ASC').limit(1)
    assert_equal [posts(:welcome)], relation.all

    author_posts = relation.only(:where)
    assert_equal Post.where(:author_id => 1).all, author_posts.all

    all_posts = relation.only(:limit)
    assert_equal Post.limit(1).all.first, all_posts.first
  end

877 878 879 880
  def test_extensions_with_only
    assert_equal 2, Topic.named_extension.order(:author_name).only(:order).two
  end

P
Pratik Naik 已提交
881
  def test_anonymous_extension
882
    relation = Post.where(:author_id => 1).order('id ASC').extending do
P
Pratik Naik 已提交
883 884 885 886 887 888 889 890 891 892 893 894 895 896
      def author
        'lifo'
      end
    end

    assert_equal "lifo", relation.author
    assert_equal "lifo", relation.limit(1).author
  end

  def test_named_extension
    relation = Post.where(:author_id => 1).order('id ASC').extending(Post::NamedExtension)
    assert_equal "lifo", relation.author
    assert_equal "lifo", relation.limit(1).author
  end
897 898 899 900

  def test_order_by_relation_attribute
    assert_equal Post.order(Post.arel_table[:title]).all, Post.order("title").all
  end
901

902
  def test_order_with_find_with_order
903 904
    assert_equal 'zyke', CoolCar.order('name desc').find(:first, :order => 'id').name
    assert_equal 'zyke', FastCar.order('name desc').find(:first, :order => 'id').name
905 906
  end

907
  def test_default_scope_order_with_scope_order
908 909 910 911
    assert_equal 'zyke', CoolCar.order_using_new_style.limit(1).first.name
    assert_equal 'zyke', CoolCar.order_using_old_style.limit(1).first.name
    assert_equal 'zyke', FastCar.order_using_new_style.limit(1).first.name
    assert_equal 'zyke', FastCar.order_using_old_style.limit(1).first.name
912 913 914
  end

  def test_order_using_scoping
915 916
    car1 = CoolCar.order('id DESC').scoping do
      CoolCar.find(:first, :order => 'id asc')
917
    end
918 919 920 921 922 923 924 925 926 927 928 929 930 931
    assert_equal 'zyke', car1.name

    car2 = FastCar.order('id DESC').scoping do
      FastCar.find(:first, :order => 'id asc')
    end
    assert_equal 'zyke', car2.name
  end

  def test_unscoped_block_style
    assert_equal 'honda', CoolCar.unscoped { CoolCar.order_using_new_style.limit(1).first.name}
    assert_equal 'honda', CoolCar.unscoped { CoolCar.order_using_old_style.limit(1).first.name}

    assert_equal 'honda', FastCar.unscoped { FastCar.order_using_new_style.limit(1).first.name}
    assert_equal 'honda', FastCar.unscoped { FastCar.order_using_old_style.limit(1).first.name}
932 933
  end

934 935 936
  def test_intersection_with_array
    relation = Author.where(:name => "David")
    rails_author = relation.first
937

938 939 940
    assert_equal [rails_author], [rails_author] & relation
    assert_equal [rails_author], relation & [rails_author]
  end
941 942 943 944

  def test_removing_limit_with_options
    assert_not_equal 1, Post.limit(1).all(:limit => nil).count
  end
945 946 947 948

  def test_primary_key
    assert_equal "id", Post.scoped.primary_key
  end
949 950 951 952 953 954 955 956 957 958 959 960 961 962 963

  def test_eager_loading_with_conditions_on_joins
    scope = Post.includes(:comments)

    # This references the comments table, and so it should cause the comments to be eager
    # loaded via a JOIN, rather than by subsequent queries.
    scope = scope.joins(
      Post.arel_table.create_join(
        Post.arel_table,
        Post.arel_table.create_on(Comment.arel_table[:id].eq(3))
      )
    )

    assert scope.eager_loading?
  end
964 965

  def test_ordering_with_extra_spaces
966
    assert_equal authors(:david), Author.order('id DESC , name DESC').last
967
  end
968 969 970 971 972 973 974 975

  def test_update_all_with_joins
    comments = Comment.joins(:post).where('posts.id' => posts(:welcome).id)
    count    = comments.count

    assert_equal count, comments.update_all(:post_id => posts(:thinking).id)
    assert_equal posts(:thinking), comments(:greetings).post
  end
P
Pratik Naik 已提交
976
end