belongs_to_associations_test.rb 24.6 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
require "cases/helper"
require 'models/developer'
require 'models/project'
require 'models/company'
require 'models/topic'
require 'models/reply'
require 'models/computer'
require 'models/post'
require 'models/author'
require 'models/tag'
require 'models/tagging'
require 'models/comment'
13 14
require 'models/sponsor'
require 'models/member'
15
require 'models/essay'
16
require 'models/toy'
17 18 19

class BelongsToAssociationsTest < ActiveRecord::TestCase
  fixtures :accounts, :companies, :developers, :projects, :topics,
20
           :developers_projects, :computers, :authors, :author_addresses,
21
           :posts, :tags, :taggings, :comments, :sponsors, :members
22 23

  def test_belongs_to
24 25 26
    firm = Client.find(3).firm
    assert_not_nil firm
    assert_equal companies(:first_firm).name, firm.name
27 28
  end

29 30 31 32 33
  def test_belongs_to_with_primary_key
    client = Client.create(:name => "Primary key client", :firm_name => companies(:first_firm).name)
    assert_equal companies(:first_firm).name, client.firm_with_primary_key.name
  end

34 35
  def test_belongs_to_with_primary_key_joins_on_correct_column
    sql = Client.joins(:firm_with_primary_key).to_sql
36
    if current_adapter?(:MysqlAdapter, :Mysql2Adapter)
37 38
      assert_no_match(/`firm_with_primary_keys_companies`\.`id`/, sql)
      assert_match(/`firm_with_primary_keys_companies`\.`name`/, sql)
39 40 41 42
    elsif current_adapter?(:OracleAdapter)
      # on Oracle aliases are truncated to 30 characters and are quoted in uppercase
      assert_no_match(/"firm_with_primary_keys_compani"\."id"/i, sql)
      assert_match(/"firm_with_primary_keys_compani"\."name"/i, sql)
E
Emilio Tagua 已提交
43
    else
44 45
      assert_no_match(/"firm_with_primary_keys_companies"\."id"/, sql)
      assert_match(/"firm_with_primary_keys_companies"\."name"/, sql)
E
Emilio Tagua 已提交
46
    end
47 48
  end

49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
  def test_proxy_assignment
    account = Account.find(1)
    assert_nothing_raised { account.firm = account.firm }
  end

  def test_type_mismatch
    assert_raise(ActiveRecord::AssociationTypeMismatch) { Account.find(1).firm = 1 }
    assert_raise(ActiveRecord::AssociationTypeMismatch) { Account.find(1).firm = Project.find(1) }
  end

  def test_natural_assignment
    apple = Firm.create("name" => "Apple")
    citibank = Account.create("credit_limit" => 10)
    citibank.firm = apple
    assert_equal apple.id, citibank.firm_id
  end

66 67 68 69 70 71 72
  def test_id_assignment
    apple = Firm.create("name" => "Apple")
    citibank = Account.create("credit_limit" => 10)
    citibank.firm_id = apple
    assert_nil citibank.firm_id
  end

73 74 75 76 77 78 79
  def test_natural_assignment_with_primary_key
    apple = Firm.create("name" => "Apple")
    citibank = Client.create("name" => "Primary key client")
    citibank.firm_with_primary_key = apple
    assert_equal apple.name, citibank.firm_name
  end

80
  def test_eager_loading_with_primary_key
A
Aaron Patterson 已提交
81 82
    Firm.create("name" => "Apple")
    Client.create("name" => "Citibank", :firm_name => "Apple")
83
    citibank_result = Client.all.merge!(:where => {:name => "Citibank"}, :includes => :firm_with_primary_key).first
84
    assert citibank_result.association_cache.key?(:firm_with_primary_key)
85 86
  end

87 88 89
  def test_eager_loading_with_primary_key_as_symbol
    Firm.create("name" => "Apple")
    Client.create("name" => "Citibank", :firm_name => "Apple")
90
    citibank_result = Client.all.merge!(:where => {:name => "Citibank"}, :includes => :firm_with_primary_key_symbols).first
91
    assert citibank_result.association_cache.key?(:firm_with_primary_key_symbols)
92 93
  end

94 95 96 97 98 99 100 101 102
  def test_creating_the_belonging_object
    citibank = Account.create("credit_limit" => 10)
    apple    = citibank.create_firm("name" => "Apple")
    assert_equal apple, citibank.firm
    citibank.save
    citibank.reload
    assert_equal apple, citibank.firm
  end

103 104 105 106 107 108 109 110 111
  def test_creating_the_belonging_object_with_primary_key
    client = Client.create(:name => "Primary key client")
    apple  = client.create_firm_with_primary_key("name" => "Apple")
    assert_equal apple, client.firm_with_primary_key
    client.save
    client.reload
    assert_equal apple, client.firm_with_primary_key
  end

112 113 114 115 116 117 118
  def test_building_the_belonging_object
    citibank = Account.create("credit_limit" => 10)
    apple    = citibank.build_firm("name" => "Apple")
    citibank.save
    assert_equal apple.id, citibank.firm_id
  end

119 120 121
  def test_building_the_belonging_object_with_implicit_sti_base_class
    account = Account.new
    company = account.build_firm
122
    assert_kind_of Company, company, "Expected #{company.class} to be a Company"
123 124 125 126 127
  end

  def test_building_the_belonging_object_with_explicit_sti_base_class
    account = Account.new
    company = account.build_firm(:type => "Company")
128
    assert_kind_of Company, company, "Expected #{company.class} to be a Company"
129 130 131 132 133
  end

  def test_building_the_belonging_object_with_sti_subclass
    account = Account.new
    company = account.build_firm(:type => "Firm")
134
    assert_kind_of Firm, company, "Expected #{company.class} to be a Firm"
135 136 137 138 139 140 141 142 143 144 145 146
  end

  def test_building_the_belonging_object_with_an_invalid_type
    account = Account.new
    assert_raise(ActiveRecord::SubclassNotFound) { account.build_firm(:type => "InvalidType") }
  end

  def test_building_the_belonging_object_with_an_unrelated_type
    account = Account.new
    assert_raise(ActiveRecord::SubclassNotFound) { account.build_firm(:type => "Account") }
  end

147 148 149 150 151 152 153
  def test_building_the_belonging_object_with_primary_key
    client = Client.create(:name => "Primary key client")
    apple  = client.build_firm_with_primary_key("name" => "Apple")
    client.save
    assert_equal apple.name, client.firm_name
  end

154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170
  def test_create!
    client  = Client.create!(:name => "Jimmy")
    account = client.create_account!(:credit_limit => 10)
    assert_equal account, client.account
    assert account.persisted?
    client.save
    client.reload
    assert_equal account, client.account
  end

  def test_failing_create!
    client  = Client.create!(:name => "Jimmy")
    assert_raise(ActiveRecord::RecordInvalid) { client.create_account! }
    assert_not_nil client.account
    assert client.account.new_record?
  end

171 172 173 174 175 176 177 178
  def test_natural_assignment_to_nil
    client = Client.find(3)
    client.firm = nil
    client.save
    assert_nil client.firm(true)
    assert_nil client.client_of
  end

179 180 181 182 183 184 185 186
  def test_natural_assignment_to_nil_with_primary_key
    client = Client.create(:name => "Primary key client", :firm_name => companies(:first_firm).name)
    client.firm_with_primary_key = nil
    client.save
    assert_nil client.firm_with_primary_key(true)
    assert_nil client.client_of
  end

187 188 189 190 191 192 193 194 195 196
  def test_with_different_class_name
    assert_equal Company.find(1).name, Company.find(3).firm_with_other_name.name
    assert_not_nil Company.find(3).firm_with_other_name, "Microsoft should have a firm"
  end

  def test_with_condition
    assert_equal Company.find(1).name, Company.find(3).firm_with_condition.name
    assert_not_nil Company.find(3).firm_with_condition, "Microsoft should have a firm"
  end

197 198 199 200 201 202 203 204 205
  def test_polymorphic_association_class
    sponsor = Sponsor.new
    assert_nil sponsor.association(:sponsorable).send(:klass)

    sponsor.sponsorable_type = '' # the column doesn't have to be declared NOT NULL
    assert_nil sponsor.association(:sponsorable).send(:klass)

    sponsor.sponsorable = Member.new :name => "Bert"
    assert_equal Member, sponsor.association(:sponsorable).send(:klass)
206
    assert_equal "members", sponsor.association(:sponsorable).aliased_table_name
207 208
  end

209 210 211 212 213 214 215 216 217
  def test_with_polymorphic_and_condition
    sponsor = Sponsor.create
    member = Member.create :name => "Bert"
    sponsor.sponsorable = member

    assert_equal member, sponsor.sponsorable
    assert_nil sponsor.sponsorable_with_conditions
  end

218
  def test_with_select
J
Jon Leighton 已提交
219
    assert_equal 1, Company.find(2).firm_with_select.attributes.size
220
    assert_equal 1, Company.all.merge!(:includes => :firm_with_select ).find(2).firm_with_select.attributes.size
221 222
  end

223 224 225 226 227 228 229 230 231 232 233 234
  def test_belongs_to_counter
    debate = Topic.create("title" => "debate")
    assert_equal 0, debate.send(:read_attribute, "replies_count"), "No replies yet"

    trash = debate.replies.create("title" => "blah!", "content" => "world around!")
    assert_equal 1, Topic.find(debate.id).send(:read_attribute, "replies_count"), "First reply created"

    trash.destroy
    assert_equal 0, Topic.find(debate.id).send(:read_attribute, "replies_count"), "First reply deleted"
  end

  def test_belongs_to_counter_with_assigning_nil
235 236
    post = Post.find(1)
    comment = Comment.find(1)
237

238 239
    assert_equal post.id, comment.post_id
    assert_equal 2, Post.find(post.id).comments.size
240

241
    comment.post = nil
242

243
    assert_equal 1, Post.find(post.id).comments.size
244 245
  end

246 247 248 249 250 251 252 253 254
  def test_belongs_to_with_primary_key_counter
    debate  = Topic.create("title" => "debate")
    debate2 = Topic.create("title" => "debate2")
    reply   = Reply.create("title" => "blah!", "content" => "world around!", "parent_title" => "debate")

    assert_equal 1, debate.reload.replies_count
    assert_equal 0, debate2.reload.replies_count

    reply.topic_with_primary_key = debate2
255

256 257
    assert_equal 0, debate.reload.replies_count
    assert_equal 1, debate2.reload.replies_count
258 259 260

    reply.topic_with_primary_key = nil

261 262
    assert_equal 0, debate.reload.replies_count
    assert_equal 0, debate2.reload.replies_count
263 264
  end

265
  def test_belongs_to_counter_with_reassigning
266 267 268 269
    topic1 = Topic.create("title" => "t1")
    topic2 = Topic.create("title" => "t2")
    reply1 = Reply.new("title" => "r1", "content" => "r1")
    reply1.topic = topic1
270

271 272 273
    assert reply1.save
    assert_equal 1, Topic.find(topic1.id).replies.size
    assert_equal 0, Topic.find(topic2.id).replies.size
274

275
    reply1.topic = Topic.find(topic2.id)
276

277
    assert_no_queries do
278
      reply1.topic = topic2
279 280
    end

281 282 283
    assert reply1.save
    assert_equal 0, Topic.find(topic1.id).replies.size
    assert_equal 1, Topic.find(topic2.id).replies.size
284

285
    reply1.topic = nil
286

287 288
    assert_equal 0, Topic.find(topic1.id).replies.size
    assert_equal 0, Topic.find(topic2.id).replies.size
289

290
    reply1.topic = topic1
291

292 293
    assert_equal 1, Topic.find(topic1.id).replies.size
    assert_equal 0, Topic.find(topic2.id).replies.size
294

295
    reply1.destroy
296

297 298
    assert_equal 0, Topic.find(topic1.id).replies.size
    assert_equal 0, Topic.find(topic2.id).replies.size
299 300
  end

301
  def test_belongs_to_reassign_with_namespaced_models_and_counters
302 303 304 305
    topic1 = Web::Topic.create("title" => "t1")
    topic2 = Web::Topic.create("title" => "t2")
    reply1 = Web::Reply.new("title" => "r1", "content" => "r1")
    reply1.topic = topic1
306

307 308 309
    assert reply1.save
    assert_equal 1, Web::Topic.find(topic1.id).replies.size
    assert_equal 0, Web::Topic.find(topic2.id).replies.size
310

311
    reply1.topic = Web::Topic.find(topic2.id)
312

313 314 315
    assert reply1.save
    assert_equal 0, Web::Topic.find(topic1.id).replies.size
    assert_equal 1, Web::Topic.find(topic2.id).replies.size
316 317
  end

318 319 320 321 322 323 324 325 326
  def test_belongs_to_counter_after_save
    topic = Topic.create!(:title => "monday night")
    topic.replies.create!(:title => "re: monday night", :content => "football")
    assert_equal 1, Topic.find(topic.id)[:replies_count]

    topic.save!
    assert_equal 1, Topic.find(topic.id)[:replies_count]
  end

327 328 329
  def test_belongs_to_counter_after_update
    topic = Topic.create!(title: "37s")
    topic.replies.create!(title: "re: 37s", content: "rails")
330 331
    assert_equal 1, Topic.find(topic.id)[:replies_count]

332
    topic.update(title: "37signals")
333 334 335
    assert_equal 1, Topic.find(topic.id)[:replies_count]
  end

336
  def test_belongs_to_counter_when_update_columns
G
ganesh 已提交
337 338 339 340
    topic = Topic.create!(:title => "37s")
    topic.replies.create!(:title => "re: 37s", :content => "rails")
    assert_equal 1, Topic.find(topic.id)[:replies_count]

V
Vipul A M 已提交
341
    topic.update_columns(content: "rails is wonderful")
G
ganesh 已提交
342 343 344
    assert_equal 1, Topic.find(topic.id)[:replies_count]
  end

345 346 347 348
  def test_assignment_before_child_saved
    final_cut = Client.new("name" => "Final Cut")
    firm = Firm.find(1)
    final_cut.firm = firm
349
    assert !final_cut.persisted?
350
    assert final_cut.save
351 352
    assert final_cut.persisted?
    assert firm.persisted?
353 354 355 356
    assert_equal firm, final_cut.firm
    assert_equal firm, final_cut.firm(true)
  end

357 358 359 360
  def test_assignment_before_child_saved_with_primary_key
    final_cut = Client.new("name" => "Final Cut")
    firm = Firm.find(1)
    final_cut.firm_with_primary_key = firm
361
    assert !final_cut.persisted?
362
    assert final_cut.save
363 364
    assert final_cut.persisted?
    assert firm.persisted?
365 366 367 368
    assert_equal firm, final_cut.firm_with_primary_key
    assert_equal firm, final_cut.firm_with_primary_key(true)
  end

369
  def test_new_record_with_foreign_key_but_no_object
370
    client = Client.new("firm_id" => 1)
371
    # sometimes tests on Oracle fail if ORDER BY is not provided therefore add always :order with :first
372
    assert_equal Firm.all.merge!(:order => "id").first, client.firm_with_basic_id
373 374
  end

375 376 377 378
  def test_setting_foreign_key_after_nil_target_loaded
    client = Client.new
    client.firm_with_basic_id
    client.firm_id = 1
379

380 381 382 383 384 385 386 387 388 389
    assert_equal companies(:first_firm), client.firm_with_basic_id
  end

  def test_polymorphic_setting_foreign_key_after_nil_target_loaded
    sponsor = Sponsor.new
    sponsor.sponsorable
    sponsor.sponsorable_id = 1
    sponsor.sponsorable_type = "Member"

    assert_equal members(:groucho), sponsor.sponsorable
390 391
  end

392 393 394 395 396 397
  def test_dont_find_target_when_foreign_key_is_null
    tagging = taggings(:thinking_general)
    queries = assert_sql { tagging.super_tag }
    assert_equal 0, queries.length
  end

398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416
  def test_field_name_same_as_foreign_key
    computer = Computer.find(1)
    assert_not_nil computer.developer, ":foreign key == attribute didn't lock up" # '
  end

  def test_counter_cache
    topic = Topic.create :title => "Zoom-zoom-zoom"
    assert_equal 0, topic[:replies_count]

    reply = Reply.create(:title => "re: zoom", :content => "speedy quick!")
    reply.topic = topic

    assert_equal 1, topic.reload[:replies_count]
    assert_equal 1, topic.replies.size

    topic[:replies_count] = 15
    assert_equal 15, topic.replies.size
  end

417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436
  def test_counter_cache_double_destroy
    topic = Topic.create :title => "Zoom-zoom-zoom"

    5.times do
      topic.replies.create(:title => "re: zoom", :content => "speedy quick!")
    end

    assert_equal 5, topic.reload[:replies_count]
    assert_equal 5, topic.replies.size

    reply = topic.replies.first

    reply.destroy
    assert_equal 4, topic.reload[:replies_count]

    reply.destroy
    assert_equal 4, topic.reload[:replies_count]
    assert_equal 4, topic.replies.size
  end

437 438 439 440 441 442 443 444 445 446 447 448 449 450 451
  def test_custom_counter_cache
    reply = Reply.create(:title => "re: zoom", :content => "speedy quick!")
    assert_equal 0, reply[:replies_count]

    silly = SillyReply.create(:title => "gaga", :content => "boo-boo")
    silly.reply = reply

    assert_equal 1, reply.reload[:replies_count]
    assert_equal 1, reply.replies.size

    reply[:replies_count] = 17
    assert_equal 17, reply.replies.size
  end

  def test_association_assignment_sticks
452
    post = Post.first
453

454
    author1, author2 = Author.all.merge!(:limit => 2).to_a
455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475
    assert_not_nil author1
    assert_not_nil author2

    # make sure the association is loaded
    post.author

    # set the association by id, directly
    post.author_id = author2.id

    # save and reload
    post.save!
    post.reload

    # the author id of the post should be the id we set
    assert_equal post.author_id, author2.id
  end

  def test_cant_save_readonly_association
    assert_raise(ActiveRecord::ReadOnlyRecord) { companies(:first_client).readonly_firm.save! }
    assert companies(:first_client).readonly_firm.readonly?
  end
476

477 478 479 480 481 482
  def test_polymorphic_assignment_foreign_type_field_updating
    # should update when assigning a saved record
    sponsor = Sponsor.new
    member = Member.create
    sponsor.sponsorable = member
    assert_equal "Member", sponsor.sponsorable_type
483

484 485 486 487 488 489
    # should update when assigning a new record
    sponsor = Sponsor.new
    member = Member.new
    sponsor.sponsorable = member
    assert_equal "Member", sponsor.sponsorable_type
  end
490 491 492 493 494 495 496 497 498 499 500 501 502 503

  def test_polymorphic_assignment_with_primary_key_foreign_type_field_updating
    # should update when assigning a saved record
    essay = Essay.new
    writer = Author.create(:name => "David")
    essay.writer = writer
    assert_equal "Author", essay.writer_type

    # should update when assigning a new record
    essay = Essay.new
    writer = Author.new
    essay.writer = writer
    assert_equal "Author", essay.writer_type
  end
504

505 506 507 508
  def test_polymorphic_assignment_updates_foreign_id_field_for_new_and_saved_records
    sponsor = Sponsor.new
    saved_member = Member.create
    new_member = Member.new
509

510 511
    sponsor.sponsorable = saved_member
    assert_equal saved_member.id, sponsor.sponsorable_id
512

513
    sponsor.sponsorable = new_member
514
    assert_nil sponsor.sponsorable_id
515
  end
516

517 518 519 520 521 522 523 524 525 526 527 528
  def test_assignment_updates_foreign_id_field_for_new_and_saved_records
    client = Client.new
    saved_firm = Firm.create :name => "Saved"
    new_firm = Firm.new

    client.firm = saved_firm
    assert_equal saved_firm.id, client.client_of

    client.firm = new_firm
    assert_nil client.client_of
  end

529 530 531 532 533 534 535 536 537
  def test_polymorphic_assignment_with_primary_key_updates_foreign_id_field_for_new_and_saved_records
    essay = Essay.new
    saved_writer = Author.create(:name => "David")
    new_writer = Author.new

    essay.writer = saved_writer
    assert_equal saved_writer.name, essay.writer_id

    essay.writer = new_writer
538
    assert_nil essay.writer_id
539 540
  end

541
  def test_belongs_to_proxy_should_not_respond_to_private_methods
542 543
    assert_raise(NoMethodError) { companies(:first_firm).private_method }
    assert_raise(NoMethodError) { companies(:second_client).firm.private_method }
544 545 546 547 548 549
  end

  def test_belongs_to_proxy_should_respond_to_private_methods_via_send
    companies(:first_firm).send(:private_method)
    companies(:second_client).firm.send(:private_method)
  end
550 551 552 553 554 555

  def test_save_of_record_with_loaded_belongs_to
    @account = companies(:first_firm).account

    assert_nothing_raised do
      Account.find(@account.id).save!
556
      Account.all.merge!(:includes => :firm).find(@account.id).save!
557 558 559 560 561 562
    end

    @account.firm.delete

    assert_nothing_raised do
      Account.find(@account.id).save!
563
      Account.all.merge!(:includes => :firm).find(@account.id).save!
564 565
    end
  end
566 567 568 569 570 571 572 573 574 575

  def test_dependent_delete_and_destroy_with_belongs_to
    author_address = author_addresses(:david_address)
    author_address_extra = author_addresses(:david_address_extra)
    assert_equal [], AuthorAddress.destroyed_author_address_ids

    assert_difference "AuthorAddress.count", -2 do
      authors(:david).destroy
    end

J
Jon Leighton 已提交
576
    assert_equal [], AuthorAddress.where(id: [author_address.id, author_address_extra.id])
577 578 579
    assert_equal [author_address.id], AuthorAddress.destroyed_author_address_ids
  end

580
  def test_invalid_belongs_to_dependent_option_nullify_raises_exception
581
    assert_raise ArgumentError do
582
      Class.new(Author).belongs_to :special_author_address, :dependent => :nullify
583 584
    end
  end
E
Emilio Tagua 已提交
585

586 587
  def test_invalid_belongs_to_dependent_option_restrict_raises_exception
    assert_raise ArgumentError do
588
      Class.new(Author).belongs_to :special_author_address, :dependent => :restrict
589 590
    end
  end
591 592 593 594 595

  def test_attributes_are_being_set_when_initialized_from_belongs_to_association_with_where_clause
    new_firm = accounts(:signals37).build_firm(:name => 'Apple')
    assert_equal new_firm.name, "Apple"
  end
596

597 598 599 600 601
  def test_attributes_are_set_without_error_when_initialized_from_belongs_to_association_with_array_in_where_clause
    new_account = Account.where(:credit_limit => [ 50, 60 ]).new
    assert_nil new_account.credit_limit
  end

602
  def test_reassigning_the_parent_id_updates_the_object
603
    client = companies(:second_client)
604

605 606 607 608
    client.firm
    client.firm_with_condition
    firm_proxy                = client.send(:association_instance_get, :firm)
    firm_with_condition_proxy = client.send(:association_instance_get, :firm_with_condition)
609

610 611 612 613 614 615 616 617 618 619 620
    assert !firm_proxy.stale_target?
    assert !firm_with_condition_proxy.stale_target?
    assert_equal companies(:first_firm), client.firm
    assert_equal companies(:first_firm), client.firm_with_condition

    client.client_of = companies(:another_firm).id

    assert firm_proxy.stale_target?
    assert firm_with_condition_proxy.stale_target?
    assert_equal companies(:another_firm), client.firm
    assert_equal companies(:another_firm), client.firm_with_condition
621 622 623
  end

  def test_polymorphic_reassignment_of_associated_id_updates_the_object
624 625 626 627 628 629 630
    sponsor = sponsors(:moustache_club_sponsor_for_groucho)

    sponsor.sponsorable
    proxy = sponsor.send(:association_instance_get, :sponsorable)

    assert !proxy.stale_target?
    assert_equal members(:groucho), sponsor.sponsorable
631

632
    sponsor.sponsorable_id = members(:some_other_guy).id
633

634 635
    assert proxy.stale_target?
    assert_equal members(:some_other_guy), sponsor.sponsorable
636 637 638
  end

  def test_polymorphic_reassignment_of_associated_type_updates_the_object
639
    sponsor = sponsors(:moustache_club_sponsor_for_groucho)
640

641 642
    sponsor.sponsorable
    proxy = sponsor.send(:association_instance_get, :sponsorable)
643

644 645 646 647
    assert !proxy.stale_target?
    assert_equal members(:groucho), sponsor.sponsorable

    sponsor.sponsorable_type = 'Firm'
648

649 650 651
    assert proxy.stale_target?
    assert_equal companies(:first_firm), sponsor.sponsorable
  end
652 653 654

  def test_reloading_association_with_key_change
    client = companies(:second_client)
655
    firm = client.association(:firm)
656 657

    client.firm = companies(:another_firm)
658 659
    firm.reload
    assert_equal companies(:another_firm), firm.target
660 661

    client.client_of = companies(:first_firm).id
662 663
    firm.reload
    assert_equal companies(:first_firm), firm.target
664
  end
665 666

  def test_polymorphic_counter_cache
667 668 669
    tagging = taggings(:welcome_general)
    post    = posts(:welcome)
    comment = comments(:greetings)
670

671
    assert_difference lambda { post.reload.taggings_count }, -1 do
672 673 674 675 676
      assert_difference 'comment.reload.taggings_count', +1 do
        tagging.taggable = comment
      end
    end
  end
677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695

  def test_polymorphic_with_custom_foreign_type
    sponsor = sponsors(:moustache_club_sponsor_for_groucho)
    groucho = members(:groucho)
    other   = members(:some_other_guy)

    assert_equal groucho, sponsor.sponsorable
    assert_equal groucho, sponsor.thing

    sponsor.thing = other

    assert_equal other, sponsor.sponsorable
    assert_equal other, sponsor.thing

    sponsor.sponsorable = groucho

    assert_equal groucho, sponsor.sponsorable
    assert_equal groucho, sponsor.thing
  end
696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716

  def test_build_with_conditions
    client = companies(:second_client)
    firm   = client.build_bob_firm

    assert_equal "Bob", firm.name
  end

  def test_create_with_conditions
    client = companies(:second_client)
    firm   = client.create_bob_firm

    assert_equal "Bob", firm.name
  end

  def test_create_bang_with_conditions
    client = companies(:second_client)
    firm   = client.create_bob_firm!

    assert_equal "Bob", firm.name
  end
717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737

  def test_build_with_block
    client = Client.create(:name => 'Client Company')

    firm = client.build_firm{ |f| f.name = 'Agency Company' }
    assert_equal 'Agency Company', firm.name
  end

  def test_create_with_block
    client = Client.create(:name => 'Client Company')

    firm = client.create_firm{ |f| f.name = 'Agency Company' }
    assert_equal 'Agency Company', firm.name
  end

  def test_create_bang_with_block
    client = Client.create(:name => 'Client Company')

    firm = client.create_firm!{ |f| f.name = 'Agency Company' }
    assert_equal 'Agency Company', firm.name
  end
738 739 740 741 742 743 744 745 746 747 748 749 750 751

  def test_should_set_foreign_key_on_create_association
    client = Client.create! :name => "fuu"

    firm = client.create_firm :name => "baa"
    assert_equal firm.id, client.client_of
  end

  def test_should_set_foreign_key_on_create_association!
    client = Client.create! :name => "fuu"

    firm = client.create_firm! :name => "baa"
    assert_equal firm.id, client.client_of
  end
752 753 754 755 756 757 758 759 760

  def test_self_referential_belongs_to_with_counter_cache_assigning_nil
    comment = Comment.create! :post => posts(:thinking), :body => "fuu"
    comment.parent = nil
    comment.save!

    assert_equal nil, comment.reload.parent
    assert_equal 0, comments(:greetings).reload.children_count
  end
761 762 763 764 765 766 767

  def test_polymorphic_with_custom_primary_key
    toy = Toy.create!
    sponsor = Sponsor.create!(:sponsorable => toy)

    assert_equal toy, sponsor.reload.sponsorable
  end
J
Jon Leighton 已提交
768 769 770 771 772 773 774 775 776 777 778 779

  test "stale tracking doesn't care about the type" do
    apple = Firm.create("name" => "Apple")
    citibank = Account.create("credit_limit" => 10)

    citibank.firm_id = apple.id
    citibank.firm # load it

    citibank.firm_id = apple.id.to_s

    assert !citibank.association(:firm).stale_target?
  end
780

781 782 783 784 785 786 787 788 789 790
  def test_reflect_the_most_recent_change
    author1, author2 = Author.limit(2)
    post = Post.new(:title => "foo", :body=> "bar")

    post.author    = author1
    post.author_id = author2.id

    assert post.save
    assert_equal post.author_id, author2.id
  end
791
end