has_many_associations_test.rb

# frozen_string_literal: true

require "cases/helper"
require "models/developer"
require "models/computer"
require "models/project"
require "models/company"
require "models/contract"
require "models/topic"
require "models/reply"
require "models/category"
require "models/image"
require "models/post"
require "models/author"
require "models/essay"
require "models/comment"
require "models/person"
require "models/reader"
require "models/tagging"
require "models/tag"
require "models/invoice"
require "models/line_item"
require "models/car"
require "models/bulb"
require "models/engine"
require "models/categorization"
require "models/minivan"
require "models/speedometer"
require "models/reference"
require "models/job"
require "models/college"
require "models/student"
require "models/pirate"
require "models/ship"
require "models/ship_part"
require "models/treasure"
require "models/parrot"
require "models/tyre"
require "models/subscriber"
require "models/subscription"
require "models/zine"
require "models/interest"
require "models/human"
require "models/sharded"
require "models/cpk"
require "models/comment_overlapping_counter_cache"

class HasManyAssociationsTestForReorderWithJoinDependency < ActiveRecord::TestCase
  fixtures :authors, :author_addresses, :posts, :comments

  def test_should_generate_valid_sql
    author = authors(:david)
    # this can fail on adapters which require ORDER BY expressions to be included in the SELECT expression
    # if the reorder clauses are not correctly handled
    assert author.posts_with_comments_sorted_by_comment_id.where("comments.id > 0").reorder("posts.comments_count": :desc, "posts.tags_count": :desc).last
  end
end

class HasManyAssociationsTestPrimaryKeys < ActiveRecord::TestCase
  fixtures :authors, :author_addresses, :essays, :subscribers, :subscriptions, :people

  def test_custom_primary_key_on_new_record_should_fetch_with_query
    subscriber = Subscriber.new(nick: "webster132")
    assert_not_predicate subscriber.subscriptions, :loaded?

    assert_queries_count 1 do
      assert_equal 2, subscriber.subscriptions.size
    end

    assert_equal Subscription.where(subscriber_id: "webster132"), subscriber.subscriptions
  end

  def test_association_primary_key_on_new_record_should_fetch_with_query
    author = Author.new(name: "David")
    assert_not_predicate author.essays, :loaded?

    assert_queries_count 1 do
      assert_equal 1, author.essays.size
    end

    assert_equal Essay.where(writer_id: "David"), author.essays
  end

  def test_has_many_custom_primary_key
    david = authors(:david)
    assert_equal Essay.where(writer_id: "David"), david.essays
  end

  def test_ids_on_unloaded_association_with_custom_primary_key
    david = people(:david)
    assert_equal Essay.where(writer_id: "David").pluck(:id), david.essay_ids
  end

  def test_ids_on_loaded_association_with_custom_primary_key
    david = people(:david)
    david.essays.load
    assert_equal Essay.where(writer_id: "David").pluck(:id), david.essay_ids
  end

  def test_has_many_assignment_with_custom_primary_key
    david = people(:david)

    assert_equal ["A Modest Proposal"], david.essays.map(&:name)
    david.essays = [Essay.create!(name: "Remote Work")]
    assert_equal ["Remote Work"], david.essays.map(&:name)
  end

  def test_blank_custom_primary_key_on_new_record_should_not_run_queries
    author = Author.new
    assert_not_predicate author.essays, :loaded?

    assert_queries_count 0 do
      assert_equal 0, author.essays.size
    end
  end
end

class HasManyAssociationsTest < ActiveRecord::TestCase
  fixtures :accounts, :categories, :companies, :developers, :projects,
           :developers_projects, :topics, :authors, :author_addresses, :comments,
           :posts, :readers, :taggings, :cars, :tags,
           :categorizations, :zines, :interests, :humans,
           :sharded_blog_posts, :sharded_comments, :cpk_books, :cpk_authors

  def setup
    Client.destroyed_client_ids.clear
  end

  def test_sti_subselect_count
    tag = Tag.first
    len = Post.tagged_with(tag.id).limit(10).size
    assert_operator len, :>, 0
  end

  def test_anonymous_has_many
    developer = Class.new(ActiveRecord::Base) {
      self.table_name = "developers"
      dev = self

      developer_project = Class.new(ActiveRecord::Base) {
        self.table_name = "developers_projects"
        belongs_to :developer, anonymous_class: dev
      }
      has_many :developer_projects, anonymous_class: developer_project, foreign_key: "developer_id"
    }
    dev = developer.first
    named = Developer.find(dev.id)
    assert_operator dev.developer_projects.count, :>, 0
    assert_equal named.projects.map(&:id).sort,
                 dev.developer_projects.map(&:project_id).sort
  end

  def test_default_scope_on_relations_is_not_cached
    counter = 0
    posts = Class.new(ActiveRecord::Base) {
      self.table_name = "posts"
      self.inheritance_column = "not_there"
      post = self

      comments = Class.new(ActiveRecord::Base) {
        self.table_name = "comments"
        self.inheritance_column = "not_there"
        belongs_to :post, anonymous_class: post
        default_scope -> {
          counter += 1
          where("id = :inc", inc: counter)
        }
      }
      has_many :comments, anonymous_class: comments, foreign_key: "post_id"
    }
    assert_equal 0, counter
    post = posts.first
    assert_equal 0, counter
    queries = capture_sql_and_binds { post.comments.to_a }
    post.comments.reset
    assert_not_equal queries, capture_sql_and_binds { post.comments.to_a }
  end

  def test_has_many_build_with_options
    college = College.create(name: "UFMT")
    Student.create(active: true, college_id: college.id, name: "Sarah")

    assert_equal college.students, Student.where(active: true, college_id: college.id)
  end

  def test_add_record_to_collection_should_change_its_updated_at
    ship = Ship.create(name: "dauntless")
    part = ShipPart.create(name: "cockpit")
    updated_at = part.updated_at

    travel(1.second) do
      ship.parts << part
    end

    assert_equal part.ship, ship
    assert_not_equal part.updated_at, updated_at
  end

  def test_clear_collection_should_not_change_updated_at
    # GH#17161: .clear calls delete_all (and returns the association),
    # which is intended to not touch associated objects's updated_at field
    ship = Ship.create(name: "dauntless")
    part = ShipPart.create(name: "cockpit", ship_id: ship.id)

    ship.parts.clear
    part.reload

    assert_nil part.ship
    assert_not_predicate part, :updated_at_changed?
  end

  def test_create_from_association_should_respect_default_scope
    car = Car.create(name: "honda")
    assert_equal "honda", car.name

    bulb = Bulb.create
    assert_equal "defaulty", bulb.name

    bulb = car.bulbs.build
    assert_equal "defaulty", bulb.name

    bulb = car.bulbs.create
    assert_equal "defaulty", bulb.name

    bulb = car.bulbs.create!
    assert_equal "defaulty", bulb.name
  end

  def test_build_and_create_from_association_should_respect_passed_attributes_over_default_scope
    car = Car.create(name: "honda")

    bulb = car.bulbs.where(name: "exotic").build
    assert_equal "exotic", bulb.name
    assert_nil bulb.count_after_create

    bulb = car.bulbs.where(name: "exotic").create
    assert_equal "exotic", bulb.name
    assert_equal 1, bulb.count_after_create

    bulb = car.bulbs.where(name: "exotic").create!
    assert_equal "exotic", bulb.name
    assert_equal 2, bulb.count_after_create

    bulb = car.bulbs.build(name: "exotic")
    assert_equal "exotic", bulb.name

    bulb = car.bulbs.create(name: "exotic")
    assert_equal "exotic", bulb.name

    bulb = car.bulbs.create!(name: "exotic")
    assert_equal "exotic", bulb.name

    bulb = car.awesome_bulbs.build(frickinawesome: false)
    assert_equal false, bulb.frickinawesome

    bulb = car.awesome_bulbs.create(frickinawesome: false)
    assert_equal false, bulb.frickinawesome

    bulb = car.awesome_bulbs.create!(frickinawesome: false)
    assert_equal false, bulb.frickinawesome
  end

  def test_build_and_create_from_association_should_respect_unscope_over_default_scope
    car = Car.create(name: "honda")

    bulb = car.bulbs.unscope(where: :name).build
    assert_nil bulb.name

    bulb = car.bulbs.unscope(where: :name).create
    assert_nil bulb.name

    bulb = car.bulbs.unscope(where: :name).create!
    assert_nil bulb.name

    bulb = car.awesome_bulbs.unscope(where: :frickinawesome).build
    assert_equal false, bulb.frickinawesome

    bulb = car.awesome_bulbs.unscope(where: :frickinawesome).create
    assert_equal false, bulb.frickinawesome

    bulb = car.awesome_bulbs.unscope(where: :frickinawesome).create!
    assert_equal false, bulb.frickinawesome
  end

  def test_build_from_association_should_respect_scope
    author = Author.new

    post = author.thinking_posts.build
    assert_equal "So I was thinking", post.title
  end

  def test_create_from_association_with_nil_values_should_work
    car = Car.create(name: "honda")

    bulb = car.bulbs.new(nil)
    assert_equal "defaulty", bulb.name

    bulb = car.bulbs.build(nil)
    assert_equal "defaulty", bulb.name

    bulb = car.bulbs.create(nil)
    assert_equal "defaulty", bulb.name
  end

  def test_build_from_association_sets_inverse_instance
    car = Car.new(name: "honda")

    bulb = car.bulbs.build
    assert_equal car, bulb.car
  end

  def test_do_not_call_callbacks_for_delete_all
    car = Car.create(name: "honda")
    car.funky_bulbs.create!
    assert_equal 1, car.funky_bulbs.count
    assert_equal 1, car.reload.funky_bulbs.delete_all
    assert_equal 0, car.funky_bulbs.count, "bulbs should have been deleted using :delete_all strategy"
  end

  def test_delete_all_on_association_is_the_same_as_not_loaded
    author = authors :david
    author.thinking_posts.create!(body: "test")
    author.reload
    expected_sql = capture_sql { author.thinking_posts.delete_all }

    author.thinking_posts.create!(body: "test")
    author.reload
    author.thinking_posts.inspect
    loaded_sql = capture_sql { author.thinking_posts.delete_all }
    assert_equal(expected_sql, loaded_sql)
  end

  def test_delete_all_on_association_with_nil_dependency_is_the_same_as_not_loaded
    author = authors :david
    author.posts.create!(title: "test", body: "body")
    author.reload
    expected_sql = capture_sql { author.posts.delete_all }

    author.posts.create!(title: "test", body: "body")
    author.reload
    author.posts.to_a
    loaded_sql = capture_sql { author.posts.delete_all }
    assert_equal(expected_sql, loaded_sql)
  end

  def test_delete_all_on_association_clears_scope
    author = Author.create!(name: "Gannon")
    posts = author.posts
    posts.create!(title: "test", body: "body")
    posts.delete_all
    assert_nil posts.first
  end

  def test_building_the_associated_object_with_implicit_sti_base_class
    firm = DependentFirm.new
    company = firm.companies.build
    assert_kind_of Company, company, "Expected #{company.class} to be a Company"
  end

  def test_building_the_associated_object_with_explicit_sti_base_class
    firm = DependentFirm.new
    company = firm.companies.build(type: "Company")
    assert_kind_of Company, company, "Expected #{company.class} to be a Company"
  end

  def test_building_the_associated_object_with_sti_subclass
    firm = DependentFirm.new
    company = firm.companies.build(type: "Client")
    assert_kind_of Client, company, "Expected #{company.class} to be a Client"
  end

  def test_building_the_associated_object_with_an_invalid_type
    firm = DependentFirm.new
    assert_raise(ActiveRecord::SubclassNotFound) { firm.companies.build(type: "Invalid") }
  end

  def test_building_the_associated_object_with_an_unrelated_type
    firm = DependentFirm.new
    assert_raise(ActiveRecord::SubclassNotFound) { firm.companies.build(type: "Account") }
  end

  test "build the association with an array" do
    speedometer = Speedometer.new(speedometer_id: "a")
    data = [{ name: "first" }, { name: "second" }]
    speedometer.minivans.where(color: "blue").build(data)

    assert_equal 2, speedometer.minivans.size
    assert speedometer.save

    speedometer.reload

    assert_equal ["first", "second"], speedometer.minivans.map(&:name).sort
    assert_equal ["blue", "blue"], speedometer.minivans.map(&:color)
  end

  test "new the association with an array" do
    speedometer = Speedometer.new(speedometer_id: "a")
    data = [{ name: "first" }, { name: "second" }]
    speedometer.minivans.where(color: "blue").new(data)

    assert_equal 2, speedometer.minivans.size
    assert speedometer.save

    speedometer.reload

    assert_equal ["first", "second"], speedometer.minivans.map(&:name).sort
    assert_equal ["blue", "blue"], speedometer.minivans.map(&:color)
  end

  test "create the association with an array" do
    speedometer = Speedometer.create!(speedometer_id: "a")
    data = [{ name: "first" }, { name: "second" }]
    speedometer.minivans.where(color: "blue").create(data)

    assert_equal 2, speedometer.minivans.size

    speedometer.reload

    assert_equal ["first", "second"], speedometer.minivans.map(&:name).sort
    assert_equal ["blue", "blue"], speedometer.minivans.map(&:color)
  end

  test "create! the association with an array" do
    speedometer = Speedometer.create!(speedometer_id: "a")
    data = [{ name: "first" }, { name: "second" }]
    speedometer.minivans.where(color: "blue").create!(data)

    assert_equal 2, speedometer.minivans.size

    speedometer.reload

    assert_equal ["first", "second"], speedometer.minivans.map(&:name).sort
    assert_equal ["blue", "blue"], speedometer.minivans.map(&:color)
  end

  def test_association_keys_bypass_attribute_protection
    car = Car.create(name: "honda")

    bulb = car.bulbs.new
    assert_equal car.id, bulb.car_id

    bulb = car.bulbs.new car_id: car.id + 1
    assert_equal car.id, bulb.car_id

    bulb = car.bulbs.build
    assert_equal car.id, bulb.car_id

    bulb = car.bulbs.build car_id: car.id + 1
    assert_equal car.id, bulb.car_id

    bulb = car.bulbs.create
    assert_equal car.id, bulb.car_id

    bulb = car.bulbs.create car_id: car.id + 1
    assert_equal car.id, bulb.car_id
  end

  def test_association_protect_foreign_key
    invoice = Invoice.create

    line_item = invoice.line_items.new
    assert_equal invoice.id, line_item.invoice_id

    line_item = invoice.line_items.new invoice_id: invoice.id + 1
    assert_equal invoice.id, line_item.invoice_id

    line_item = invoice.line_items.build
    assert_equal invoice.id, line_item.invoice_id

    line_item = invoice.line_items.build invoice_id: invoice.id + 1
    assert_equal invoice.id, line_item.invoice_id

    line_item = invoice.line_items.create
    assert_equal invoice.id, line_item.invoice_id

    line_item = invoice.line_items.create invoice_id: invoice.id + 1
    assert_equal invoice.id, line_item.invoice_id
  end

  class SpecialAuthor < ActiveRecord::Base
    self.table_name = "authors"
    has_many :books, class_name: "SpecialBook", foreign_key: :author_id
  end

  class SpecialBook < ActiveRecord::Base
    self.table_name = "books"

    belongs_to :author
    enum :last_read, { unread: 0, reading: 2, read: 3, forgotten: nil }
  end

  def test_association_enum_works_properly
    author = SpecialAuthor.create!(name: "Test")
    book = SpecialBook.create!(last_read: "reading")
    author.books << book

    assert_equal "reading", book.last_read
    assert_not_equal 0, SpecialAuthor.joins(:books).where(books: { last_read: "reading" }).count
  end

  # When creating objects on the association, we must not do it within a scope (even though it
  # would be convenient), because this would cause that scope to be applied to any callbacks etc.
  def test_build_and_create_should_not_happen_within_scope
    car = cars(:honda)
    scope = car.foo_bulbs.where_values_hash

    bulb = car.foo_bulbs.build
    assert_not_equal scope, bulb.scope_after_initialize.where_values_hash

    bulb = car.foo_bulbs.create
    assert_not_equal scope, bulb.scope_after_initialize.where_values_hash

    bulb = car.foo_bulbs.create!
    assert_not_equal scope, bulb.scope_after_initialize.where_values_hash
  end

  def test_no_sql_should_be_fired_if_association_already_loaded
    Car.create(name: "honda")
    bulbs = Car.first.bulbs
    bulbs.to_a # to load all instances of bulbs

    assert_no_queries do
      bulbs.first()
    end

    assert_no_queries do
      bulbs.second()
    end

    assert_no_queries do
      bulbs.third()
    end

    assert_no_queries do
      bulbs.fourth()
    end

    assert_no_queries do
      bulbs.fifth()
    end

    assert_no_queries do
      bulbs.forty_two()
    end

    assert_no_queries do
      bulbs.third_to_last()
    end

    assert_no_queries do
      bulbs.second_to_last()
    end

    assert_no_queries do
      bulbs.last()
    end
  end

  def test_finder_method_with_dirty_target
    company = companies(:first_firm)
    new_clients = []

    assert_queries_count(0) do
      new_clients << company.clients_of_firm.build(name: "Another Client")
      new_clients << company.clients_of_firm.build(name: "Another Client II")
      new_clients << company.clients_of_firm.build(name: "Another Client III")
    end

    assert_not_predicate company.clients_of_firm, :loaded?
    assert_queries_count(1) do
      assert_same new_clients[0], company.clients_of_firm.third
      assert_same new_clients[1], company.clients_of_firm.fourth
      assert_same new_clients[2], company.clients_of_firm.fifth
      assert_same new_clients[0], company.clients_of_firm.third_to_last
      assert_same new_clients[1], company.clients_of_firm.second_to_last
      assert_same new_clients[2], company.clients_of_firm.last
    end
  end

  def test_finder_bang_method_with_dirty_target
    company = companies(:first_firm)
    new_clients = []

    assert_queries_count(0) do
      new_clients << company.clients_of_firm.build(name: "Another Client")
      new_clients << company.clients_of_firm.build(name: "Another Client II")
      new_clients << company.clients_of_firm.build(name: "Another Client III")
    end

    assert_not_predicate company.clients_of_firm, :loaded?
    assert_queries_count(1) do
      assert_same new_clients[0], company.clients_of_firm.third!
      assert_same new_clients[1], company.clients_of_firm.fourth!
      assert_same new_clients[2], company.clients_of_firm.fifth!
      assert_same new_clients[0], company.clients_of_firm.third_to_last!
      assert_same new_clients[1], company.clients_of_firm.second_to_last!
      assert_same new_clients[2], company.clients_of_firm.last!
    end
  end

  def test_create_resets_cached_counters
    Reader.delete_all

    person = Person.create!(first_name: "tenderlove")

    post   = Post.first

    assert_equal [], person.readers
    assert_nil person.readers.find_by_post_id(post.id)

    person.readers.create(post_id: post.id)

    assert_equal 1, person.readers.count
    assert_equal 1, person.readers.length
    assert_equal post, person.readers.first.post
    assert_equal person, person.readers.first.person
  end

  def test_update_all_respects_association_scope
    person = Person.new
    person.first_name = "Naruto"
    person.references << Reference.new
    person.save!
    assert_equal 1, person.references.update_all(favorite: true)
  end

  def test_exists_respects_association_scope
    person = Person.new
    person.first_name = "Sasuke"
    person.references << Reference.new
    person.save!
    assert_predicate person.references, :exists?
  end

  def test_counting_with_counter_sql
    assert_equal 3, Firm.first.clients.count
  end

  def test_counting
    assert_equal 3, Firm.first.plain_clients.count
  end

  def test_counting_with_single_hash
    assert_equal 1, Firm.first.plain_clients.where(name: "Microsoft").count
  end

  def test_counting_with_column_name_and_hash
    assert_equal 3, Firm.first.plain_clients.count(:name)
  end

  def test_counting_with_association_limit
    firm = companies(:first_firm)
    assert_equal firm.limited_clients.length, firm.limited_clients.size
    assert_equal firm.limited_clients.length, firm.limited_clients.count
  end

  def test_finding
    assert_equal 3, Firm.first.clients.length
  end

  def test_finding_array_compatibility
    assert_equal 3, Firm.order(:id).find { |f| f.id > 0 }.clients.length
  end

  def test_find_many_with_merged_options
    assert_equal 1, companies(:first_firm).limited_clients.size
    assert_equal 1, companies(:first_firm).limited_clients.to_a.size
    assert_equal 3, companies(:first_firm).limited_clients.limit(nil).to_a.size
  end

  def test_find_should_append_to_association_order
    ordered_clients = companies(:first_firm).clients_sorted_desc.order("companies.id")
    assert_equal ["id DESC", "companies.id"], ordered_clients.order_values
  end

  def test_dynamic_find_should_respect_association_order
    assert_equal companies(:another_first_firm_client), companies(:first_firm).clients_sorted_desc.where("type = 'Client'").first
    assert_equal companies(:another_first_firm_client), companies(:first_firm).clients_sorted_desc.find_by_type("Client")
  end

  def test_taking
    posts(:other_by_bob).destroy
    assert_equal posts(:misc_by_bob), authors(:bob).posts.take
    assert_equal posts(:misc_by_bob), authors(:bob).posts.take!
    authors(:bob).posts.to_a
    assert_equal posts(:misc_by_bob), authors(:bob).posts.take
    assert_equal posts(:misc_by_bob), authors(:bob).posts.take!
  end

  def test_taking_not_found
    authors(:bob).posts.delete_all
    assert_raise(ActiveRecord::RecordNotFound) { authors(:bob).posts.take! }
    authors(:bob).posts.to_a
    assert_raise(ActiveRecord::RecordNotFound) { authors(:bob).posts.take! }
  end

  def test_taking_with_a_number
    klass = Class.new(Author) do
      has_many :posts, -> { order(:id) }

      def self.name
        "Author"
      end
    end

    # taking from unloaded Relation
    bob = klass.find(authors(:bob).id)
    new_post = bob.posts.build
    assert_not_predicate bob.posts, :loaded?
    assert_equal [posts(:misc_by_bob)], bob.posts.take(1)
    assert_equal [posts(:misc_by_bob), posts(:other_by_bob)], bob.posts.take(2)
    assert_equal [posts(:misc_by_bob), posts(:other_by_bob), new_post], bob.posts.take(3)

    # taking from loaded Relation
    bob.posts.load
    assert_predicate bob.posts, :loaded?
    assert_equal [posts(:misc_by_bob)], bob.posts.take(1)
    assert_equal [posts(:misc_by_bob), posts(:other_by_bob)], bob.posts.take(2)
    assert_equal [posts(:misc_by_bob), posts(:other_by_bob), new_post], bob.posts.take(3)
  end

  def test_taking_with_inverse_of
    interests(:woodsmanship).destroy
    interests(:survival).destroy

    zine = zines(:going_out)
    interest = zine.interests.take
    assert_equal interests(:hunting), interest
    assert_same zine, interest.zine
  end

  def test_cant_save_has_many_readonly_association
    authors(:david).readonly_comments.each { |c| assert_raise(ActiveRecord::ReadOnlyRecord) { c.save! } }
    authors(:david).readonly_comments.each { |c| assert_predicate c, :readonly? }
  end

  def test_finding_default_orders
    assert_equal "Summit", Firm.first.clients.first.name
  end

  def test_finding_with_different_class_name_and_order
    assert_equal "Apex", Firm.first.clients_sorted_desc.first.name
  end

  def test_finding_with_foreign_key
    assert_equal "Microsoft", Firm.first.clients_of_firm.first.name
  end

  def test_finding_with_condition
    assert_equal "Microsoft", Firm.first.clients_like_ms.first.name
  end

  def test_finding_with_condition_hash
    assert_equal "Microsoft", Firm.first.clients_like_ms_with_hash_conditions.first.name
  end

  def test_finding_using_primary_key
    assert_equal "Summit", Firm.first.clients_using_primary_key.first.name
  end

  def test_update_all_on_association_accessed_before_save
    firm = Firm.new(name: "Firm")
    firm.clients << Client.first
    firm.save!
    assert_equal firm.clients.count, firm.clients.update_all(description: "Great!")
  end

  def test_update_all_on_association_accessed_before_save_with_explicit_foreign_key
    firm = Firm.new(name: "Firm", id: 100)
    firm.clients << Client.first
    firm.save!
    assert_equal firm.clients.count, firm.clients.update_all(description: "Great!")
  end

  def test_belongs_to_with_new_object
    c = Client.new
    assert_nil c.firm, "belongs_to failed on new object"
  end

  def test_find_ids
    firm = Firm.first

    assert_raise(ActiveRecord::RecordNotFound) { firm.clients.find }

    client = firm.clients.find(2)
    assert_kind_of Client, client

    client_ary = firm.clients.find([2])
    assert_kind_of Array, client_ary
    assert_equal client, client_ary.first

    client_ary = firm.clients.find(2, 3)
    assert_kind_of Array, client_ary
    assert_equal 2, client_ary.size
    assert_equal client, client_ary.first

    assert_raise(ActiveRecord::RecordNotFound) { firm.clients.find(2, 99) }
  end

  def test_find_one_message_on_primary_key
    firm = Firm.first

    e = assert_raises(ActiveRecord::RecordNotFound) do
      firm.clients.find(0)
    end
    assert_equal 0, e.id
    assert_equal "id", e.primary_key
    assert_equal "Client", e.model
    assert_match (/\ACouldn't find Client with 'id'=0/), e.message
  end

  def test_find_ids_and_inverse_of
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    firm = companies(:first_firm)
    client = firm.clients_of_firm.find(3)
    assert_kind_of Client, client

    client_ary = firm.clients_of_firm.find([3])
    assert_kind_of Array, client_ary
    assert_equal client, client_ary.first
  end

  def test_find_all
    firm = Firm.first
    assert_equal 3, firm.clients.where("#{QUOTED_TYPE} = 'Client'").to_a.length
    assert_equal 1, firm.clients.where("name = 'Summit'").to_a.length
  end

  def test_find_each
    firm = companies(:first_firm)

    assert_not_predicate firm.clients, :loaded?

    assert_queries_count(4) do
      firm.clients.find_each(batch_size: 1) { |c| assert_equal firm.id, c.firm_id }
    end

    assert_not_predicate firm.clients, :loaded?
  end

  def test_find_each_with_conditions
    firm = companies(:first_firm)

    assert_queries_count(2) do
      firm.clients.where(name: "Microsoft").find_each(batch_size: 1) do |c|
        assert_equal firm.id, c.firm_id
        assert_equal "Microsoft", c.name
      end
    end

    assert_not_predicate firm.clients, :loaded?
  end

  def test_find_in_batches
    firm = companies(:first_firm)

    assert_not_predicate firm.clients, :loaded?

    assert_queries_count(2) do
      firm.clients.find_in_batches(batch_size: 2) do |clients|
        clients.each { |c| assert_equal firm.id, c.firm_id }
      end
    end

    assert_not_predicate firm.clients, :loaded?
  end

  def test_find_all_sanitized
    firm = Firm.first
    summit = firm.clients.where("name = 'Summit'").to_a
    assert_equal summit, firm.clients.where("name = ?", "Summit").to_a
    assert_equal summit, firm.clients.where("name = :name", name: "Summit").to_a
  end

  def test_find_first
    firm = Firm.first
    client2 = Client.find(2)
    assert_equal firm.clients.first, firm.clients.order("id").first
    assert_equal client2, firm.clients.where("#{QUOTED_TYPE} = 'Client'").order("id").first
  end

  def test_find_first_sanitized
    firm = Firm.first
    client2 = Client.find(2)
    assert_equal client2, firm.clients.where("#{QUOTED_TYPE} = ?", "Client").first
    assert_equal client2, firm.clients.where("#{QUOTED_TYPE} = :type", type: "Client").first
  end

  def test_find_first_after_reset_scope
    firm = Firm.first
    collection = firm.clients

    original_object = collection.first
    assert_same original_object, collection.first, "Expected second call to #first to cache the same object"

    # It should return a different object, since the association has been reloaded
    assert_not_same original_object, firm.clients.first, "Expected #first to return a new object"
  end

  def test_find_first_after_reset
    firm = Firm.first
    collection = firm.clients

    original_object = collection.first
    assert_same original_object, collection.first, "Expected second call to #first to cache the same object"
    collection.reset

    # It should return a different object, since the association has been reloaded
    assert_not_same original_object, collection.first, "Expected #first after #reset to return a new object"
  end

  def test_find_first_after_reload
    firm = Firm.first
    collection = firm.clients

    original_object = collection.first
    assert_same original_object, collection.first, "Expected second call to #first to cache the same object"
    collection.reload

    # It should return a different object, since the association has been reloaded
    assert_not_same original_object, collection.first, "Expected #first after #reload to return a new object"
  end

  def test_reload_with_query_cache
    connection = ActiveRecord::Base.connection
    connection.enable_query_cache!
    connection.clear_query_cache

    # Populate the cache with a query
    firm = Firm.first
    # Populate the cache with a second query
    firm.clients.load

    assert_equal 2, connection.query_cache.size

    # Clear the cache and fetch the clients again, populating the cache with a query
    assert_queries_count(1) { firm.clients.reload }
    # This query is cached, so it shouldn't make a real SQL query
    assert_queries_count(0) { firm.clients.load }

    assert_equal 1, connection.query_cache.size
  ensure
    ActiveRecord::Base.connection.disable_query_cache!
  end

  def test_reloading_unloaded_associations_with_query_cache
    connection = ActiveRecord::Base.connection
    connection.enable_query_cache!
    connection.clear_query_cache

    firm = Firm.create!(name: "firm name")
    client = firm.clients.create!(name: "client name")
    firm.clients.to_a # add request to cache

    connection.uncached do
      client.update!(name: "new client name")
    end

    firm = Firm.find(firm.id)

    assert_equal [client.name], firm.clients.reload.map(&:name)
  ensure
    ActiveRecord::Base.connection.disable_query_cache!
  end

  def test_find_all_with_include_and_conditions
    assert_nothing_raised do
      Developer.all.merge!(joins: :audit_logs, where: { "audit_logs.message" => nil, :name => "Smith" }).to_a
    end
  end

  def test_find_in_collection
    assert_equal Client.find(2).name, companies(:first_firm).clients.find(2).name
    assert_raise(ActiveRecord::RecordNotFound) { companies(:first_firm).clients.find(6) }
  end

  def test_find_grouped
    all_clients_of_firm1 = Client.all.merge!(where: "firm_id = 1").to_a
    grouped_clients_of_firm1 = Client.all.merge!(where: "firm_id = 1", group: "firm_id", select: "firm_id, count(id) as clients_count").to_a
    assert_equal 3, all_clients_of_firm1.size
    assert_equal 1, grouped_clients_of_firm1.size
  end

  def test_find_scoped_grouped
    assert_equal 1, companies(:first_firm).clients_grouped_by_firm_id.size
    assert_equal 1, companies(:first_firm).clients_grouped_by_firm_id.length
    assert_equal 3, companies(:first_firm).clients_grouped_by_name.size
    assert_equal 3, companies(:first_firm).clients_grouped_by_name.length
  end

  def test_find_scoped_grouped_having
    assert_equal 2, authors(:david).popular_grouped_posts.length
    assert_equal 0, authors(:mary).popular_grouped_posts.length
  end

  def test_default_select
    assert_equal Comment.column_names.sort, posts(:welcome).comments.first.attributes.keys.sort
  end

  def test_select_query_method
    assert_equal ["id", "body"], posts(:welcome).comments.select(:id, :body).first.attributes.keys
  end

  def test_select_with_block
    assert_equal [1], posts(:welcome).comments.select { |c| c.id == 1 }.map(&:id)
  end

  def test_select_with_block_and_dirty_target
    assert_equal 2, posts(:welcome).comments.select { true }.size
    posts(:welcome).comments.build
    assert_equal 3, posts(:welcome).comments.select { true }.size
  end

  def test_select_without_foreign_key
    assert_equal companies(:first_firm).accounts.first.credit_limit, companies(:first_firm).accounts.select(:credit_limit).first.credit_limit
  end

  def test_adding
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    natural = Client.new("name" => "Natural Company")
    companies(:first_firm).clients_of_firm << natural
    assert_equal 3, companies(:first_firm).clients_of_firm.size # checking via the collection
    assert_equal 3, companies(:first_firm).clients_of_firm.reload.size # checking using the db
    assert_equal natural, companies(:first_firm).clients_of_firm.last
  end

  def test_adding_using_create
    first_firm = companies(:first_firm)
    assert_equal 3, first_firm.plain_clients.size
    first_firm.plain_clients.create(name: "Natural Company")
    assert_equal 4, first_firm.plain_clients.length
    assert_equal 4, first_firm.plain_clients.size
  end

  def test_create_with_bang_on_has_many_when_parent_is_new_raises
    firm = Firm.new
    error = assert_raise(ActiveRecord::RecordNotSaved) do
      firm.plain_clients.create! name: "Whoever"
    end

    assert_equal "You cannot call create unless the parent is saved", error.message
    assert_equal firm, error.record
  end

  def test_regular_create_on_has_many_when_parent_is_new_raises
    firm = Firm.new
    error = assert_raise(ActiveRecord::RecordNotSaved) do
      firm.plain_clients.create name: "Whoever"
    end

    assert_equal "You cannot call create unless the parent is saved", error.message
    assert_equal firm, error.record
  end

  def test_create_with_bang_on_has_many_raises_when_record_not_saved
    assert_raise(ActiveRecord::RecordInvalid) do
      firm = Firm.first
      firm.plain_clients.create!
    end
  end

  def test_create_with_bang_on_habtm_when_parent_is_new_raises
    developer = Developer.new("name" => "Aredridel")
    error = assert_raise(ActiveRecord::RecordNotSaved) do
      developer.projects.create!
    end

    assert_equal "You cannot call create unless the parent is saved", error.message
    assert_equal developer, error.record
  end

  def test_adding_a_mismatch_class
    assert_raise(ActiveRecord::AssociationTypeMismatch) { companies(:first_firm).clients_of_firm << nil }
    assert_raise(ActiveRecord::AssociationTypeMismatch) { companies(:first_firm).clients_of_firm << 1 }
    assert_raise(ActiveRecord::AssociationTypeMismatch) { companies(:first_firm).clients_of_firm << Topic.find(1) }
  end

  def test_adding_a_collection
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    result = companies(:first_firm).clients_of_firm.concat([Client.new("name" => "Natural Company"), Client.new("name" => "Apple")])
    assert_equal 4, companies(:first_firm).clients_of_firm.size
    assert_equal 4, companies(:first_firm).clients_of_firm.reload.size
    assert_equal companies(:first_firm).clients_of_firm, result
  end

  def test_transactions_when_adding_to_persisted
    good = Client.new(name: "Good")
    bad  = Client.new(name: "Bad", raise_on_save: true)

    begin
      companies(:first_firm).clients_of_firm.concat(good, bad)
    rescue Client::RaisedOnSave
    end

    assert_not_includes companies(:first_firm).clients_of_firm.reload, good
  end

  def test_transactions_when_adding_to_new_record
    firm = Firm.new
    assert_queries_count(0) do
      firm.clients_of_firm.concat(Client.new("name" => "Natural Company"))
    end
  end

  def test_inverse_on_before_validate
    firm = companies(:first_firm)
    assert_queries_count(3) do
      firm.clients_of_firm << Client.new("name" => "Natural Company")
    end
  end

  def test_new_aliased_to_build
    company = companies(:first_firm)

    new_client = assert_queries_count(0) { company.clients_of_firm.new("name" => "Another Client") }
    assert_not_predicate company.clients_of_firm, :loaded?

    assert_equal "Another Client", new_client.name
    assert_not_predicate new_client, :persisted?
    assert_equal new_client, company.clients_of_firm.last
  end

  def test_build
    company = companies(:first_firm)

    new_client = assert_queries_count(0) { company.clients_of_firm.build("name" => "Another Client") }
    assert_not_predicate company.clients_of_firm, :loaded?

    assert_equal "Another Client", new_client.name
    assert_not_predicate new_client, :persisted?
    assert_equal new_client, company.clients_of_firm.last
  end

  def test_collection_size_after_building
    company = companies(:first_firm)  # company already has one client
    company.clients_of_firm.build("name" => "Another Client")
    company.clients_of_firm.build("name" => "Yet Another Client")
    assert_equal 4, company.clients_of_firm.size
    assert_equal 4, company.clients_of_firm.uniq.size
  end

  def test_collection_not_empty_after_building
    company = companies(:first_firm)
    assert_empty company.contracts
    company.contracts.build
    assert_not_empty company.contracts
  end

  def test_collection_size_with_dirty_target
    post = posts(:thinking)
    assert_equal [], post.reader_ids
    assert_equal 0, post.readers.size
    post.readers.reset
    post.readers.build
    assert_equal [nil], post.reader_ids
    assert_equal 1, post.readers.size
  end

  def test_collection_empty_with_dirty_target
    post = posts(:thinking)
    assert_equal [], post.reader_ids
    assert_empty post.readers
    post.readers.reset
    post.readers.build
    assert_equal [nil], post.reader_ids
    assert_not_empty post.readers
  end

  def test_collection_size_twice_for_regressions
    post = posts(:thinking)
    assert_equal 0, post.readers.size
    # This test needs a post that has no readers, we assert it to ensure it holds,
    # but need to reload the post because the very call to #size hides the bug.
    post.reload
    post.readers.build
    size1 = post.readers.size
    size2 = post.readers.size
    assert_equal size1, size2
  end

  def test_build_many
    company = companies(:first_firm)

    new_clients = assert_queries_count(0) { company.clients_of_firm.build([{ "name" => "Another Client" }, { "name" => "Another Client II" }]) }
    assert_equal 2, new_clients.size
  end

  def test_build_followed_by_save_does_not_load_target
    companies(:first_firm).clients_of_firm.build("name" => "Another Client")
    assert companies(:first_firm).save
    assert_not_predicate companies(:first_firm).clients_of_firm, :loaded?
  end

  def test_build_without_loading_association
    first_topic = topics(:first)

    assert_equal 1, first_topic.replies.length

    assert_queries_count(0) do
      first_topic.replies.build(title: "Not saved", content: "Superstars")
      assert_equal 2, first_topic.replies.size
    end

    assert_equal 2, first_topic.replies.to_ary.size
  end

  def test_build_via_block
    company = companies(:first_firm)

    new_client = assert_queries_count(0) { company.clients_of_firm.build { |client| client.name = "Another Client" } }
    assert_not_predicate company.clients_of_firm, :loaded?

    assert_equal "Another Client", new_client.name
    assert_not_predicate new_client, :persisted?
    assert_equal new_client, company.clients_of_firm.last
  end

  def test_build_many_via_block
    company = companies(:first_firm)

    new_clients = assert_queries_count(0) do
      company.clients_of_firm.build([{ "name" => "Another Client" }, { "name" => "Another Client II" }]) do |client|
        client.name = "changed"
      end
    end

    assert_equal 2, new_clients.size
    assert_equal "changed", new_clients.first.name
    assert_equal "changed", new_clients.last.name
  end

  def test_create_without_loading_association
    first_firm = companies(:first_firm)

    assert_equal 2, first_firm.clients_of_firm.size
    first_firm.clients_of_firm.reset

    assert_queries_count(3) do
      first_firm.clients_of_firm.create(name: "Superstars")
    end

    assert_equal 3, first_firm.clients_of_firm.size
  end

  def test_create
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    new_client = companies(:first_firm).clients_of_firm.create("name" => "Another Client")
    assert_predicate new_client, :persisted?
    assert_equal new_client, companies(:first_firm).clients_of_firm.last
    assert_equal new_client, companies(:first_firm).clients_of_firm.reload.last
  end

  def test_create_many
    companies(:first_firm).clients_of_firm.create([{ "name" => "Another Client" }, { "name" => "Another Client II" }])
    assert_equal 4, companies(:first_firm).clients_of_firm.reload.size
  end

  def test_create_followed_by_save_does_not_load_target
    companies(:first_firm).clients_of_firm.create("name" => "Another Client")
    assert companies(:first_firm).save
    assert_not_predicate companies(:first_firm).clients_of_firm, :loaded?
  end

  def test_deleting
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    companies(:first_firm).clients_of_firm.delete(companies(:first_firm).clients_of_firm.first)
    assert_equal 1, companies(:first_firm).clients_of_firm.size
    assert_equal 1, companies(:first_firm).clients_of_firm.reload.size
  end

  def test_deleting_before_save
    new_firm = Firm.new("name" => "A New Firm, Inc.")
    new_client = new_firm.clients_of_firm.build("name" => "Another Client")
    assert_equal 1, new_firm.clients_of_firm.size
    new_firm.clients_of_firm.delete(new_client)
    assert_equal 0, new_firm.clients_of_firm.size
  end

  def test_deleting_models_with_composite_keys
    great_author = cpk_authors(:cpk_great_author)
    books = great_author.books

    assert_equal 2, books.size

    great_author.books.delete(books.first)
    great_author.reload

    assert_equal 1, great_author.books.size
  end

  def test_sharded_deleting_models
    blog_post = sharded_blog_posts(:great_post_blog_one)
    comments = blog_post.delete_comments

    assert_equal 3, comments.size

    comments_to_delete = [comments.first, comments.second]

    sql = capture_sql do
      blog_post.delete_comments.delete(comments_to_delete)
    end

    c = Sharded::Comment.connection

    blog_id = Regexp.escape(c.quote_table_name("sharded_comments.blog_id"))
    id = Regexp.escape(c.quote_table_name("sharded_comments.id"))

    query_constraints = /#{blog_id} = .* AND #{id} = .*/
    expectation = /DELETE.*WHERE.* \(#{query_constraints} OR #{query_constraints}\)/

    assert_match(expectation, sql.second)

    blog_post.reload

    assert_equal 1, blog_post.comments.size
  end

  def test_has_many_without_counter_cache_option
    # Ship has a conventionally named `treasures_count` column, but the counter_cache
    # option is not given on the association.
    ship = Ship.create!(name: "Countless", treasures_count: 10)

    assert_not_predicate Ship.reflect_on_association(:treasures), :has_cached_counter?

    # Count should come from sql count() of treasures rather than treasures_count attribute
    assert_equal 0, ship.treasures.size

    assert_no_difference lambda { ship.reload.treasures_count }, "treasures_count should not be changed" do
      ship.treasures.create(name: "Gold")
    end

    assert_no_difference lambda { ship.reload.treasures_count }, "treasures_count should not be changed" do
      ship.treasures.destroy_all
    end
  end

  def test_deleting_updates_counter_cache
    topic = Topic.order("id ASC").first
    assert_equal topic.replies.to_a.size, topic.replies_count

    topic.replies.delete(topic.replies.first)
    topic.reload
    assert_equal topic.replies.to_a.size, topic.replies_count
  end

  def test_counter_cache_updates_in_memory_after_concat
    topic = Topic.create title: "Zoom-zoom-zoom"

    topic.replies << Reply.create(title: "re: zoom", content: "speedy quick!")
    assert_equal 1, topic.replies_count
    assert_equal 1, topic.replies.size
    assert_equal 1, topic.reload.replies.size
  end

  def test_counter_cache_updates_in_memory_after_create
    topic = Topic.create title: "Zoom-zoom-zoom"

    topic.replies.create!(title: "re: zoom", content: "speedy quick!")
    assert_equal 1, topic.replies_count
    assert_equal 1, topic.replies.size
    assert_equal 1, topic.reload.replies.size
  end

  def test_counter_cache_updates_in_memory_after_create_with_array
    topic = Topic.create title: "Zoom-zoom-zoom"

    topic.replies.create!([
      { title: "re: zoom", content: "speedy quick!" },
      { title: "re: zoom 2", content: "OMG lol!" },
    ])
    assert_equal 2, topic.replies_count
    assert_equal 2, topic.replies.size
    assert_equal 2, topic.reload.replies.size
  end

  def test_counter_cache_updates_in_memory_after_update_with_inverse_of_disabled
    topic = Topic.create!(title: "Zoom-zoom-zoom")

    assert_equal 0, topic.replies_count

    reply1 = Reply.create!(title: "re: zoom", content: "speedy quick!")
    reply2 = Reply.create!(title: "re: zoom 2", content: "OMG lol!")

    assert_queries_count(6) do
      topic.replies << [reply1, reply2]
    end

    assert_equal 2, topic.replies_count
    assert_equal 2, topic.reload.replies_count
  end

  def test_counter_cache_updates_in_memory_after_create_with_overlapping_counter_cache_columns
    user = UserCommentsCount.create!
    post = PostCommentsCount.create!

    assert_difference "user.comments_count", +1 do
      assert_no_difference "post.comments_count" do
        post.comments << CommentOverlappingCounterCache.create!(user_comments_count: user)
      end
    end

    assert_difference "user.comments_count", +1 do
      assert_no_difference "post.comments_count" do
        user.comments << CommentOverlappingCounterCache.create!(post_comments_count: post)
      end
    end
  end

  def test_counter_cache_updates_in_memory_after_update_with_inverse_of_enabled
    category = Category.create!(name: "Counter Cache")

    assert_nil category.categorizations_count

    categorization1 = Categorization.create!
    categorization2 = Categorization.create!

    assert_queries_count(6) do
      category.categorizations << [categorization1, categorization2]
    end

    assert_equal 2, category.categorizations_count
    assert_equal 2, category.reload.categorizations_count
  end

  def test_pushing_association_updates_counter_cache
    topic = Topic.order("id ASC").first
    reply = Reply.create!

    assert_difference "topic.reload.replies_count", 1 do
      topic.replies << reply
    end
  end

  def test_deleting_updates_counter_cache_without_dependent_option
    post = posts(:welcome)

    assert_difference "post.reload.tags_count", -1 do
      post.taggings.delete(post.taggings.first)
    end
  end

  def test_deleting_updates_counter_cache_with_dependent_delete_all
    post = posts(:welcome)
    post.update_columns(taggings_with_delete_all_count: post.tags_count)

    assert_difference "post.reload.taggings_with_delete_all_count", -1 do
      post.taggings_with_delete_all.delete(post.taggings_with_delete_all.first)
    end
  end

  def test_deleting_updates_counter_cache_with_dependent_destroy
    post = posts(:welcome)
    post.update_columns(taggings_with_destroy_count: post.tags_count)

    assert_difference "post.reload.taggings_with_destroy_count", -1 do
      post.taggings_with_destroy.delete(post.taggings_with_destroy.first)
    end
  end

  def test_calling_empty_with_counter_cache
    post = posts(:welcome)
    assert_no_queries do
      assert_not_empty post.comments
    end
  end

  def test_custom_named_counter_cache
    topic = topics(:first)

    assert_difference "topic.reload.replies_count", -1 do
      topic.approved_replies.clear
    end
  end

  def test_calling_update_on_id_changes_the_counter_cache
    topic = Topic.order("id ASC").first
    original_count = topic.replies.to_a.size
    assert_equal original_count, topic.replies_count

    first_reply = topic.replies.first
    first_reply.update(parent_id: nil)
    assert_equal original_count - 1, topic.reload.replies_count

    first_reply.update(parent_id: topic.id)
    assert_equal original_count, topic.reload.replies_count
  end

  def test_calling_update_changing_ids_changes_the_counter_cache
    topic1 = Topic.find(1)
    topic2 = Topic.find(3)
    original_count1 = topic1.replies.to_a.size
    original_count2 = topic2.replies.to_a.size

    reply1 = topic1.replies.first
    reply2 = topic2.replies.first

    reply1.update(parent_id: topic2.id)
    assert_equal original_count1 - 1, topic1.reload.replies_count
    assert_equal original_count2 + 1, topic2.reload.replies_count

    reply2.update(parent_id: topic1.id)
    assert_equal original_count1, topic1.reload.replies_count
    assert_equal original_count2, topic2.reload.replies_count
  end

  def test_calling_update_changing_ids_of_inversed_association_changes_the_counter_cache
    assert_predicate Post.reflect_on_association(:comments), :has_inverse?

    post1 = Post.first
    post2 = Post.second

    original_count1 = post1.comments.count
    original_count2 = post2.comments.count

    post1.comments.first.update(post_id: post2.id)
    assert_equal original_count1 - 1, post1.reload.comments_count
    assert_equal original_count2 + 1, post2.reload.comments_count

    post2.comments.first.update(post_id: post1.id)
    assert_equal original_count1, post1.reload.comments_count
    assert_equal original_count2, post2.reload.comments_count
  end

  def test_deleting_a_collection
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    companies(:first_firm).clients_of_firm.create("name" => "Another Client")
    assert_equal 3, companies(:first_firm).clients_of_firm.size
    companies(:first_firm).clients_of_firm.delete([companies(:first_firm).clients_of_firm[0], companies(:first_firm).clients_of_firm[1], companies(:first_firm).clients_of_firm[2]])
    assert_equal 0, companies(:first_firm).clients_of_firm.size
    assert_equal 0, companies(:first_firm).clients_of_firm.reload.size
  end

  def test_delete_all
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    companies(:first_firm).dependent_clients_of_firm.create("name" => "Another Client")
    clients = companies(:first_firm).dependent_clients_of_firm.to_a
    assert_equal 3, clients.count

    assert_difference "Client.count", -(clients.count) do
      assert_equal clients.count, companies(:first_firm).dependent_clients_of_firm.delete_all
    end
  end

  def test_delete_all_with_not_yet_loaded_association_collection
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    companies(:first_firm).clients_of_firm.create("name" => "Another Client")
    assert_equal 3, companies(:first_firm).clients_of_firm.size
    companies(:first_firm).clients_of_firm.reset
    companies(:first_firm).clients_of_firm.delete_all
    assert_equal 0, companies(:first_firm).clients_of_firm.size
    assert_equal 0, companies(:first_firm).clients_of_firm.reload.size
  end

  def test_transaction_when_deleting_persisted
    good = Client.new(name: "Good")
    bad  = Client.new(name: "Bad", raise_on_destroy: true)

    companies(:first_firm).clients_of_firm = [good, bad]

    begin
      companies(:first_firm).clients_of_firm.destroy(good, bad)
    rescue Client::RaisedOnDestroy
    end

    assert_equal [good, bad], companies(:first_firm).clients_of_firm.reload
  end

  def test_transaction_when_deleting_new_record
    firm = Firm.new
    assert_queries_count(0) do
      client = Client.new("name" => "New Client")
      firm.clients_of_firm << client
      firm.clients_of_firm.destroy(client)
    end
  end

  def test_clearing_an_association_collection
    firm = companies(:first_firm)
    client_id = firm.clients_of_firm.first.id
    assert_equal 2, firm.clients_of_firm.size

    firm.clients_of_firm.clear

    assert_equal 0, firm.clients_of_firm.size
    assert_equal 0, firm.clients_of_firm.reload.size
    assert_equal [], Client.destroyed_client_ids[firm.id]

    # Should not be destroyed since the association is not dependent.
    assert_nothing_raised do
      assert_nil Client.find(client_id).firm
    end
  end

  def test_clearing_updates_counter_cache
    topic = Topic.first

    assert_difference "topic.reload.replies_count", -1 do
      topic.replies.clear
    end
  end

  def test_clearing_updates_counter_cache_when_inverse_counter_cache_is_a_symbol_with_dependent_destroy
    car = Car.first
    car.engines.create!

    assert_difference "car.reload.engines_count", -1 do
      car.engines.clear
    end
  end

  def test_clearing_a_dependent_association_collection
    firm = companies(:first_firm)
    client_id = firm.dependent_clients_of_firm.first.id
    assert_equal 2, firm.dependent_clients_of_firm.size
    assert_equal 1, Client.find_by_id(client_id).client_of

    # :delete_all is called on each client since the dependent options is :destroy
    firm.dependent_clients_of_firm.clear

    assert_equal 0, firm.dependent_clients_of_firm.size
    assert_equal 0, firm.dependent_clients_of_firm.reload.size
    assert_equal [], Client.destroyed_client_ids[firm.id]

    # Should be destroyed since the association is dependent.
    assert_nil Client.find_by_id(client_id)
  end

  def test_delete_all_with_option_delete_all
    firm = companies(:first_firm)
    client_id = firm.dependent_clients_of_firm.first.id
    count = firm.dependent_clients_of_firm.count
    assert_equal count, firm.dependent_clients_of_firm.delete_all(:delete_all)
    assert_nil Client.find_by_id(client_id)
  end

  def test_delete_all_with_option_nullify
    firm = companies(:first_firm)
    client_id = firm.dependent_clients_of_firm.first.id
    count = firm.dependent_clients_of_firm.count
    assert_equal firm, Client.find(client_id).firm
    assert_equal count, firm.dependent_clients_of_firm.delete_all(:nullify)
    assert_nil Client.find(client_id).firm
  end

  def test_delete_all_accepts_limited_parameters
    firm = companies(:first_firm)
    assert_raise(ArgumentError) do
      firm.dependent_clients_of_firm.delete_all(:destroy)
    end
  end

  def test_clearing_an_exclusively_dependent_association_collection
    firm = companies(:first_firm)
    client_id = firm.exclusively_dependent_clients_of_firm.first.id
    assert_equal 2, firm.exclusively_dependent_clients_of_firm.size

    assert_equal [], Client.destroyed_client_ids[firm.id]

    # :exclusively_dependent means each client is deleted directly from
    # the database without looping through them calling destroy.
    firm.exclusively_dependent_clients_of_firm.clear

    assert_equal 0, firm.exclusively_dependent_clients_of_firm.size
    assert_equal 0, firm.exclusively_dependent_clients_of_firm.reload.size
    # no destroy-filters should have been called
    assert_equal [], Client.destroyed_client_ids[firm.id]

    # Should be destroyed since the association is exclusively dependent.
    assert_nil Client.find_by_id(client_id)
  end

  def test_dependent_association_respects_optional_conditions_on_delete
    firm = companies(:odegy)
    Client.create(client_of: firm.id, name: "BigShot Inc.")
    Client.create(client_of: firm.id, name: "SmallTime Inc.")
    # only one of two clients is included in the association due to the :conditions key
    assert_equal 2, Client.where(client_of: firm.id).size
    assert_equal 1, firm.dependent_conditional_clients_of_firm.size
    firm.destroy
    # only the correctly associated client should have been deleted
    assert_equal 1, Client.where(client_of: firm.id).size
  end

  def test_dependent_association_respects_optional_sanitized_conditions_on_delete
    firm = companies(:odegy)
    Client.create(client_of: firm.id, name: "BigShot Inc.")
    Client.create(client_of: firm.id, name: "SmallTime Inc.")
    # only one of two clients is included in the association due to the :conditions key
    assert_equal 2, Client.where(client_of: firm.id).size
    assert_equal 1, firm.dependent_sanitized_conditional_clients_of_firm.size
    firm.destroy
    # only the correctly associated client should have been deleted
    assert_equal 1, Client.where(client_of: firm.id).size
  end

  def test_dependent_association_respects_optional_hash_conditions_on_delete
    firm = companies(:odegy)
    Client.create(client_of: firm.id, name: "BigShot Inc.")
    Client.create(client_of: firm.id, name: "SmallTime Inc.")
    # only one of two clients is included in the association due to the :conditions key
    assert_equal 2, Client.where(client_of: firm.id).size
    assert_equal 1, firm.dependent_hash_conditional_clients_of_firm.size
    firm.destroy
    # only the correctly associated client should have been deleted
    assert_equal 1, Client.where(client_of: firm.id).size
  end

  def test_delete_all_association_with_primary_key_deletes_correct_records
    firm = Firm.first
    # break the vanilla firm_id foreign key
    assert_equal 3, firm.clients.count
    firm.clients.first.update_columns(firm_id: nil)
    assert_equal 2, firm.clients.reload.count
    assert_equal 2, firm.clients_using_primary_key_with_delete_all.count
    old_record = firm.clients_using_primary_key_with_delete_all.first
    firm = Firm.first
    firm.destroy
    assert_nil Client.find_by_id(old_record.id)
  end

  def test_creation_respects_hash_condition
    ms_client = companies(:first_firm).clients_like_ms_with_hash_conditions.build

    assert ms_client.save
    assert_equal "Microsoft", ms_client.name

    another_ms_client = companies(:first_firm).clients_like_ms_with_hash_conditions.create

    assert_predicate another_ms_client, :persisted?
    assert_equal "Microsoft", another_ms_client.name
  end

  def test_clearing_without_initial_access
    firm = companies(:first_firm)

    firm.clients_of_firm.clear

    assert_equal 0, firm.clients_of_firm.size
    assert_equal 0, firm.clients_of_firm.reload.size
  end

  def test_deleting_a_item_which_is_not_in_the_collection
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    summit = Client.find_by_name("Summit")
    companies(:first_firm).clients_of_firm.delete(summit)
    assert_equal 2, companies(:first_firm).clients_of_firm.size
    assert_equal 2, companies(:first_firm).clients_of_firm.reload.size
    assert_equal 2, summit.client_of
  end

  def test_deleting_by_integer_id
    david = Developer.find(1)

    assert_difference "david.projects.count", -1 do
      assert_equal 1, david.projects.delete(1).size
    end

    assert_equal 1, david.projects.size
  end

  def test_deleting_by_string_id
    david = Developer.find(1)

    assert_difference "david.projects.count", -1 do
      assert_equal 1, david.projects.delete("1").size
    end

    assert_equal 1, david.projects.size
  end

  def test_deleting_self_type_mismatch
    david = Developer.find(1)
    david.projects.reload
    assert_raise(ActiveRecord::AssociationTypeMismatch) { david.projects.delete(Project.find(1).developers) }
  end

  def test_destroying
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    assert_difference "Client.count", -1 do
      companies(:first_firm).clients_of_firm.destroy(companies(:first_firm).clients_of_firm.first)
    end

    assert_equal 1, companies(:first_firm).reload.clients_of_firm.size
    assert_equal 1, companies(:first_firm).clients_of_firm.reload.size
  end

  def test_destroying_by_integer_id
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    assert_difference "Client.count", -1 do
      companies(:first_firm).clients_of_firm.destroy(companies(:first_firm).clients_of_firm.first.id)
    end

    assert_equal 1, companies(:first_firm).reload.clients_of_firm.size
    assert_equal 1, companies(:first_firm).clients_of_firm.reload.size
  end

  def test_destroying_by_string_id
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    assert_difference "Client.count", -1 do
      companies(:first_firm).clients_of_firm.destroy(companies(:first_firm).clients_of_firm.first.id.to_s)
    end

    assert_equal 1, companies(:first_firm).reload.clients_of_firm.size
    assert_equal 1, companies(:first_firm).clients_of_firm.reload.size
  end

  def test_destroying_a_collection
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    companies(:first_firm).clients_of_firm.create("name" => "Another Client")
    assert_equal 3, companies(:first_firm).clients_of_firm.size

    assert_difference "Client.count", -2 do
      companies(:first_firm).clients_of_firm.destroy([companies(:first_firm).clients_of_firm[0], companies(:first_firm).clients_of_firm[1]])
    end

    assert_equal 1, companies(:first_firm).reload.clients_of_firm.size
    assert_equal 1, companies(:first_firm).clients_of_firm.reload.size
  end

  def test_destroy_all
    force_signal37_to_load_all_clients_of_firm

    assert_predicate companies(:first_firm).clients_of_firm, :loaded?

    clients = companies(:first_firm).clients_of_firm.to_a
    assert_not clients.empty?, "37signals has clients after load"
    destroyed = companies(:first_firm).clients_of_firm.destroy_all
    assert_equal clients.sort_by(&:id), destroyed.sort_by(&:id)
    assert destroyed.all?(&:frozen?), "destroyed clients should be frozen"
    assert_predicate companies(:first_firm).clients_of_firm, :empty?, "37signals has no clients after destroy all"
    assert_predicate companies(:first_firm).clients_of_firm.reload, :empty?, "37signals has no clients after destroy all and refresh"
  end

  def test_destroy_all_on_association_clears_scope
    author = Author.create!(name: "Gannon")
    posts = author.posts
    posts.create!(title: "test", body: "body")
    posts.destroy_all
    assert_nil posts.first
  end

  def test_destroy_all_on_desynced_counter_cache_association
    category = categories(:general)
    assert_operator category.categorizations.count, :>, 0

    category.categorizations.destroy_all
    assert_equal 0, category.categorizations.count
  end

  def test_destroy_on_association_clears_scope
    author = Author.create!(name: "Gannon")
    posts = author.posts
    post = posts.create!(title: "test", body: "body")
    posts.destroy(post)
    assert_nil posts.first
  end

  def test_delete_on_association_clears_scope
    author = Author.create!(name: "Gannon")
    posts = author.posts
    post = posts.create!(title: "test", body: "body")
    posts.delete(post)
    assert_nil posts.first
  end

  def test_dependence
    firm = companies(:first_firm)
    assert_equal 3, firm.clients.size
    firm.destroy
    assert_empty Client.all.merge!(where: "firm_id=#{firm.id}").to_a
  end

  def test_dependence_for_associations_with_hash_condition
    david = authors(:david)
    assert_difference("Post.count", -1) { assert david.destroy }
  end

  def test_destroy_dependent_when_deleted_from_association
    firm = Firm.first
    assert_equal 3, firm.clients.size

    client = firm.clients.first
    firm.clients.delete(client)

    assert_raise(ActiveRecord::RecordNotFound) { Client.find(client.id) }
    assert_raise(ActiveRecord::RecordNotFound) { firm.clients.find(client.id) }
    assert_equal 2, firm.clients.size
  end

  def test_three_levels_of_dependence
    topic = Topic.create "title" => "neat and simple"
    reply = topic.replies.create "title" => "neat and simple", "content" => "still digging it"
    reply.replies.create "title" => "neat and simple", "content" => "ain't complaining"

    assert_nothing_raised { topic.destroy }
  end

  def test_dependence_with_transaction_support_on_failure
    firm = companies(:first_firm)
    clients = firm.clients
    assert_equal 3, clients.length
    clients.last.instance_eval { def overwrite_to_raise() raise "Trigger rollback" end }

    firm.destroy rescue "do nothing"

    assert_equal 3, Client.all.merge!(where: "firm_id=#{firm.id}").to_a.size
  end

  def test_dependence_on_account
    num_accounts = Account.count
    companies(:first_firm).destroy
    assert_equal num_accounts - 1, Account.count
  end

  def test_depends_and_nullify
    num_accounts = Account.count

    core = companies(:rails_core)
    assert_equal accounts(:rails_core_account), core.account
    assert_equal companies(:leetsoft, :jadedpixel).sort_by(&:id), core.companies.sort_by(&:id)
    core.destroy
    assert_nil accounts(:rails_core_account).reload.firm_id
    assert_nil companies(:leetsoft).reload.client_of
    assert_nil companies(:jadedpixel).reload.client_of

    assert_equal num_accounts, Account.count
  end

  def test_depends_and_nullify_on_polymorphic_assoc
    author = PersonWithPolymorphicDependentNullifyComments.create!(first_name: "Laertis")
    comment = posts(:welcome).comments.first
    comment.author = author
    comment.save!

    assert_equal comment.author_id, author.id
    assert_equal comment.author_type, author.class.name

    author.destroy
    comment.reload

    assert_nil comment.author_id
    assert_nil comment.author_type
  end

  def test_restrict_with_exception
    firm = RestrictedWithExceptionFirm.create!(name: "restrict")
    firm.companies.create(name: "child")

    assert_not_empty firm.companies
    assert_raise(ActiveRecord::DeleteRestrictionError) { firm.destroy }
    assert RestrictedWithExceptionFirm.exists?(name: "restrict")
    assert firm.companies.exists?(name: "child")
  end

  def test_restrict_with_error
    firm = RestrictedWithErrorFirm.create!(name: "restrict")
    firm.companies.create(name: "child")

    assert_not_empty firm.companies

    firm.destroy

    assert_not_empty firm.errors

    assert_equal "Cannot delete record because dependent companies exist", firm.errors[:base].first
    assert RestrictedWithErrorFirm.exists?(name: "restrict")
    assert firm.companies.exists?(name: "child")
  end

  def test_restrict_with_error_with_locale
    I18n.backend = I18n::Backend::Simple.new
    I18n.backend.store_translations "en", activerecord: { attributes: { restricted_with_error_firm: { companies: "client companies" } } }
    firm = RestrictedWithErrorFirm.create!(name: "restrict")
    firm.companies.create(name: "child")

    assert_not_empty firm.companies

    firm.destroy

    assert_not_empty firm.errors

    assert_equal "Cannot delete record because dependent client companies exist", firm.errors[:base].first
    assert RestrictedWithErrorFirm.exists?(name: "restrict")
    assert firm.companies.exists?(name: "child")
  ensure
    I18n.backend.reload!
  end

  def test_included_in_collection
    assert_equal true, companies(:first_firm).clients.include?(Client.find(2))
  end

  def test_included_in_collection_for_new_records
    client = Client.create(name: "Persisted")
    assert_nil client.client_of
    assert_equal false, Firm.new.clients_of_firm.include?(client),
     "includes a client that does not belong to any firm"
  end

  def test_adding_array_and_collection
    assert_nothing_raised { Firm.first.clients + Firm.all.last.clients }
  end

  def test_replace_with_less
    firm = Firm.first
    firm.clients = [companies(:first_client)]
    assert firm.save, "Could not save firm"
    firm.reload
    assert_equal 1, firm.clients.length
  end

  def test_replace_with_less_and_dependent_nullify
    num_companies = Company.count
    companies(:rails_core).companies = []
    assert_equal num_companies, Company.count
  end

  def test_replace_with_new
    firm = Firm.first
    firm.clients = [companies(:second_client), Client.new("name" => "New Client")]
    firm.save
    firm.reload
    assert_equal 2, firm.clients.length
    assert_equal false, firm.clients.include?(:first_client)
  end

  def test_replace_failure
    firm = companies(:first_firm)
    account = Account.new
    orig_accounts = firm.accounts.to_a

    assert_not_predicate account, :valid?
    assert_not_empty orig_accounts
    error = assert_raise ActiveRecord::RecordNotSaved do
      firm.accounts = [account]
    end

    assert_equal orig_accounts, firm.accounts
    assert_equal "Failed to replace accounts because one or more of the " \
                 "new records could not be saved.", error.message
  end

  def test_replace_with_same_content
    firm = Firm.first
    firm.clients = []
    firm.save

    assert_no_queries do
      firm.clients = []
    end

    assert_equal [], firm.public_send("clients=", [])
  end

  def test_transactions_when_replacing_on_persisted
    good = Client.new(name: "Good")
    bad  = Client.new(name: "Bad", raise_on_save: true)

    companies(:first_firm).clients_of_firm = [good]

    begin
      companies(:first_firm).clients_of_firm = [bad]
    rescue Client::RaisedOnSave
    end

    assert_equal [good], companies(:first_firm).clients_of_firm.reload
  end

  def test_transactions_when_replacing_on_new_record
    firm = Firm.new
    assert_queries_count(0) do
      firm.clients_of_firm = [Client.new("name" => "New Client")]
    end
  end

  def test_get_ids
    assert_equal [companies(:first_client).id, companies(:second_client).id, companies(:another_first_firm_client).id], companies(:first_firm).client_ids
  end

  def test_get_ids_for_loaded_associations
    company = companies(:first_firm)
    company.clients.reload
    assert_no_queries do
      company.client_ids
      company.client_ids
    end
  end

  def test_get_ids_for_unloaded_associations_does_not_load_them
    company = companies(:first_firm)
    assert_not_predicate company.clients, :loaded?
    assert_equal [companies(:first_client).id, companies(:second_client).id, companies(:another_first_firm_client).id], company.client_ids
    assert_not_predicate company.clients, :loaded?
  end

  def test_counter_cache_on_unloaded_association
    car = Car.create(name: "My AppliCar")
    assert_equal 0, car.engines.size
  end

  def test_ids_reader_cache_not_used_for_size_when_association_is_dirty
    firm = Firm.create!(name: "Startup")
    assert_equal 0, firm.client_ids.size
    firm.clients.build
    assert_equal 1, firm.clients.size
  end

  def test_ids_reader_cache_should_be_cleared_when_collection_is_deleted
    firm = companies(:first_firm)
    assert_equal [2, 3, 11], firm.client_ids
    client = firm.clients.first
    firm.clients.delete(client)
    assert_equal [3, 11], firm.client_ids
  end

  def test_get_ids_ignores_include_option
    assert_equal [readers(:michael_welcome).id], posts(:welcome).readers_with_person_ids
  end

  def test_get_ids_for_ordered_association
    assert_equal [companies(:another_first_firm_client).id, companies(:second_client).id, companies(:first_client).id], companies(:first_firm).clients_ordered_by_name_ids
  end

  def test_get_ids_for_association_on_new_record_does_not_try_to_find_records
    company = Company.new
    assert_queries_count(0) do
      company.contract_ids
    end

    assert_equal [], company.contract_ids
  end

  def test_set_ids_for_association_on_new_record_applies_association_correctly
    contract_a = Contract.create!
    contract_b = Contract.create!
    Contract.create! # another contract
    company = Company.new(name: "Some Company")

    company.contract_ids = [contract_a.id, contract_b.id]
    assert_equal [contract_a.id, contract_b.id], company.contract_ids
    assert_equal [contract_a, contract_b], company.contracts

    company.save!
    assert_equal company, contract_a.reload.company
    assert_equal company, contract_b.reload.company
  end

  def test_assign_ids_ignoring_blanks
    firm = Firm.create!(name: "Apple")
    firm.client_ids = [companies(:first_client).id, nil, companies(:second_client).id, ""]
    firm.save!

    assert_equal 2, firm.clients.reload.size
    assert_equal true, firm.clients.include?(companies(:second_client))
  end

  def test_get_ids_for_through
    assert_equal [comments(:eager_other_comment1).id], authors(:mary).comment_ids
  end

  def test_modifying_a_through_a_has_many_should_raise
    [
      lambda { authors(:mary).comment_ids = [comments(:greetings).id, comments(:more_greetings).id] },
      lambda { authors(:mary).comments = [comments(:greetings), comments(:more_greetings)] },
      lambda { authors(:mary).comments << Comment.create!(body: "Yay", post_id: 424242) },
      lambda { authors(:mary).comments.delete(authors(:mary).comments.first) },
    ].each { |block| assert_raise(ActiveRecord::HasManyThroughCantAssociateThroughHasOneOrManyReflection, &block) }
  end

  def test_associations_order_should_be_priority_over_throughs_order
    original = authors(:david)
    expected = [13, 12, 10, 9, 8, 7, 6, 5, 3, 2, 1]
    assert_equal expected, original.comments_desc.map(&:id)
    preloaded = Author.includes(:comments_desc).find(original.id)
    assert_equal expected, preloaded.comments_desc.map(&:id)
    assert_equal original.posts_sorted_by_id.first.comments.map(&:id), preloaded.posts_sorted_by_id.first.comments.map(&:id)
  end

  def test_dynamic_find_should_respect_association_order_for_through
    assert_equal Comment.find(10), authors(:david).comments_desc.where("comments.type = 'SpecialComment'").first
    assert_equal Comment.find(10), authors(:david).comments_desc.find_by_type("SpecialComment")
  end

  def test_has_many_through_respects_hash_conditions
    assert_equal authors(:david).hello_posts.sort_by(&:id), authors(:david).hello_posts_with_hash_conditions.sort_by(&:id)
    assert_equal authors(:david).hello_post_comments.sort_by(&:id), authors(:david).hello_post_comments_with_hash_conditions.sort_by(&:id)
  end

  def test_include_uses_array_include_after_loaded
    firm = companies(:first_firm)
    firm.clients.load_target

    client = firm.clients.first

    assert_no_queries do
      assert_predicate firm.clients, :loaded?
      assert_equal true, firm.clients.include?(client)
    end
  end

  def test_include_checks_if_record_exists_if_target_not_loaded
    firm = companies(:first_firm)
    client = firm.clients.first

    firm.reload
    assert_not_predicate firm.clients, :loaded?
    assert_queries_count(1) do
      assert_equal true, firm.clients.include?(client)
    end
    assert_not_predicate firm.clients, :loaded?
  end

  def test_include_returns_false_for_non_matching_record_to_verify_scoping
    firm = companies(:first_firm)
    client = Client.create!(name: "Not Associated")

    assert_not_predicate firm.clients, :loaded?
    assert_equal false, firm.clients.include?(client)
  end

  def test_calling_first_nth_or_last_on_association_should_not_load_association
    firm = companies(:first_firm)
    firm.clients.first
    firm.clients.second
    firm.clients.last
    assert_not_predicate firm.clients, :loaded?
  end

  def test_calling_first_or_last_on_loaded_association_should_not_fetch_with_query
    firm = companies(:first_firm)
    firm.clients.load_target
    assert_predicate firm.clients, :loaded?

    assert_no_queries do
      firm.clients.first
      assert_equal 2, firm.clients.first(2).size
      firm.clients.last
      assert_equal 2, firm.clients.last(2).size
    end
  end

  def test_calling_first_nth_or_last_on_existing_record_with_build_should_load_association
    firm = companies(:first_firm)
    firm.clients.build(name: "Foo")
    assert_not_predicate firm.clients, :loaded?

    assert_queries_count 1 do
      firm.clients.first
      firm.clients.second
      firm.clients.last
    end

    assert_predicate firm.clients, :loaded?

    author = Author.create!(name: "Carl")
    third  = topics(:third)
    fourth = topics(:fourth).becomes(Topic)

    new_topic = author.topics_without_type.build

    assert_not_predicate author.topics_without_type, :loaded?

    assert_queries_count(1) do
      if current_adapter?(:Mysql2Adapter, :TrilogyAdapter, :SQLite3Adapter)
        assert_equal fourth, author.topics_without_type.first
        assert_equal third, author.topics_without_type.second
      end
      assert_equal new_topic, author.topics_without_type.last
    end

    assert_predicate author.topics_without_type, :loaded?
  end

  def test_calling_first_nth_or_last_on_existing_record_with_create_should_not_load_association
    firm = companies(:first_firm)
    firm.clients.create(name: "Foo")
    assert_not_predicate firm.clients, :loaded?

    assert_queries_count 3 do
      firm.clients.first
      firm.clients.second
      firm.clients.last
    end

    assert_not_predicate firm.clients, :loaded?
  end

  def test_calling_first_nth_or_last_on_new_record_should_not_run_queries
    firm = Firm.new

    assert_no_queries do
      firm.clients.first
      firm.clients.second
      firm.clients.last
    end
  end

  def test_calling_first_or_last_with_integer_on_association_should_not_load_association
    firm = companies(:first_firm)
    firm.clients.create(name: "Foo")
    assert_not_predicate firm.clients, :loaded?

    assert_queries_count 2 do
      firm.clients.first(2)
      firm.clients.last(2)
    end

    assert_not_predicate firm.clients, :loaded?
  end

  def test_calling_many_should_count_instead_of_loading_association
    firm = companies(:first_firm)
    assert_queries_count(1) do
      firm.clients.many?  # use count query
    end
    assert_not_predicate firm.clients, :loaded?
  end

  def test_calling_many_on_loaded_association_should_not_use_query
    firm = companies(:first_firm)
    firm.clients.load  # force load
    assert_no_queries { assert_predicate firm.clients, :many? }
  end

  def test_subsequent_calls_to_many_should_use_query
    firm = companies(:first_firm)
    assert_queries_count(2) do
      firm.clients.many?
      firm.clients.many?
    end
  end

  def test_calling_many_should_defer_to_collection_if_using_a_block
    firm = companies(:first_firm)
    assert_queries_count(1) do
      assert_not_called(firm.clients, :size) do
        firm.clients.many? { true }
      end
    end
    assert_predicate firm.clients, :loaded?
  end

  def test_calling_many_should_return_false_if_none_or_one
    firm = companies(:another_firm)
    assert_not_predicate firm.clients_like_ms, :many?
    assert_equal 0, firm.clients_like_ms.size

    firm = companies(:first_firm)
    assert_not_predicate firm.limited_clients, :many?
    assert_equal 1, firm.limited_clients.size
  end

  def test_calling_many_should_return_true_if_more_than_one
    firm = companies(:first_firm)
    assert_predicate firm.clients, :many?
    assert_equal 3, firm.clients.size
  end

  def test_calling_none_should_count_instead_of_loading_association
    firm = companies(:first_firm)
    assert_queries_count(1) do
      firm.clients.none?  # use count query
    end
    assert_not_predicate firm.clients, :loaded?
  end

  def test_calling_none_on_loaded_association_should_not_use_query
    firm = companies(:first_firm)
    firm.clients.load  # force load
    assert_no_queries { assert_not firm.clients.none? }
  end

  def test_calling_none_should_defer_to_collection_if_using_a_block
    firm = companies(:first_firm)
    assert_queries_count(1) do
      assert_not_called(firm.clients, :size) do
        firm.clients.none? { true }
      end
    end
    assert_predicate firm.clients, :loaded?
  end

  def test_calling_none_should_return_true_if_none
    firm = companies(:another_firm)
    assert_predicate firm.clients_like_ms, :none?
    assert_equal 0, firm.clients_like_ms.size
  end

  def test_calling_none_should_return_false_if_any
    firm = companies(:first_firm)
    assert_not_predicate firm.limited_clients, :none?
    assert_equal 1, firm.limited_clients.size
  end

  def test_calling_one_should_count_instead_of_loading_association
    firm = companies(:first_firm)
    assert_queries_count(1) do
      firm.clients.one?  # use count query
    end
    assert_not_predicate firm.clients, :loaded?
  end

  def test_calling_one_on_loaded_association_should_not_use_query
    firm = companies(:first_firm)
    firm.clients.load  # force load
    assert_no_queries { assert_not firm.clients.one? }
  end

  def test_subsequent_calls_to_one_should_use_query
    firm = companies(:first_firm)
    assert_queries_count(2) do
      firm.clients.one?
      firm.clients.one?
    end
  end

  def test_calling_one_should_defer_to_collection_if_using_a_block
    firm = companies(:first_firm)
    assert_queries_count(1) do
      assert_not_called(firm.clients, :size) do
        firm.clients.one? { true }
      end
    end
    assert_predicate firm.clients, :loaded?
  end

  def test_calling_one_should_return_false_if_zero
    firm = companies(:another_firm)
    assert_not_predicate firm.clients_like_ms, :one?
    assert_equal 0, firm.clients_like_ms.size
  end

  def test_calling_one_should_return_true_if_one
    firm = companies(:first_firm)
    assert_predicate firm.limited_clients, :one?
    assert_equal 1, firm.limited_clients.size
  end

  def test_calling_one_should_return_false_if_more_than_one
    firm = companies(:first_firm)
    assert_not_predicate firm.clients, :one?
    assert_equal 3, firm.clients.size
  end

  def test_joins_with_namespaced_model_should_use_correct_type
    old = ActiveRecord::Base.store_full_sti_class
    ActiveRecord::Base.store_full_sti_class = true

    firm = Namespaced::Firm.create(name: "Some Company")
    firm.clients.create(name: "Some Client")

    stats = Namespaced::Firm.all.merge!(
      select: "#{Namespaced::Firm.table_name}.id, COUNT(#{Namespaced::Client.table_name}.id) AS num_clients",
      joins: :clients,
      group: "#{Namespaced::Firm.table_name}.id"
    ).find firm.id
    assert_equal 1, stats.num_clients.to_i
  ensure
    ActiveRecord::Base.store_full_sti_class = old
  end

  def test_association_proxy_transaction_method_starts_transaction_in_association_class
    assert_called(Comment, :transaction) do
      Post.first.comments.transaction do
        # nothing
      end
    end
  end

  def test_sending_new_to_association_proxy_should_have_same_effect_as_calling_new
    client_association = companies(:first_firm).clients
    assert_equal client_association.new.attributes, client_association.send(:new).attributes
  end

  def test_creating_using_primary_key
    firm = Firm.first
    client = firm.clients_using_primary_key.create!(name: "test")
    assert_equal firm.name, client.firm_name
  end

  def test_defining_has_many_association_with_delete_all_dependency_lazily_evaluates_target_class
    assert_not_called_on_instance_of(
      ActiveRecord::Reflection::AssociationReflection,
      :class_name,
    ) do
      class_eval(<<-EOF, __FILE__, __LINE__ + 1)
        class DeleteAllModel < ActiveRecord::Base
          has_many :nonentities, :dependent => :delete_all
        end
      EOF
    end
  end

  def test_defining_has_many_association_with_nullify_dependency_lazily_evaluates_target_class
    assert_not_called_on_instance_of(
      ActiveRecord::Reflection::AssociationReflection,
      :class_name,
    ) do
      class_eval(<<-EOF, __FILE__, __LINE__ + 1)
        class NullifyModel < ActiveRecord::Base
          has_many :nonentities, :dependent => :nullify
        end
      EOF
    end
  end

  def test_attributes_are_being_set_when_initialized_from_has_many_association_with_where_clause
    new_comment = posts(:welcome).comments.where(body: "Some content").build
    assert_equal "Some content", new_comment.body
  end

  def test_attributes_are_being_set_when_initialized_from_has_many_association_with_multiple_where_clauses
    new_comment = posts(:welcome).comments.where(body: "Some content").where(type: "SpecialComment").build
    assert_equal "Some content", new_comment.body
    assert_equal "SpecialComment", new_comment.type
    assert_equal new_comment.post_id, posts(:welcome).id
  end

  def test_include_method_in_has_many_association_should_return_true_for_instance_added_with_build
    post = Post.new
    comment = post.comments.build
    assert_equal true, post.comments.include?(comment)
  end

  def test_load_target_respects_protected_attributes
    topic = Topic.create!
    reply = topic.replies.create(title: "reply 1")
    reply.approved = false
    reply.save!

    # Save with a different object instance, so the instance that's still held
    # in topic.relies doesn't know about the changed attribute.
    reply2 = Reply.find(reply.id)
    reply2.approved = true
    reply2.save!

    # Force loading the collection from the db. This will merge the existing
    # object (reply) with what gets loaded from the db (which includes the
    # changed approved attribute). approved is a protected attribute, so if mass
    # assignment is used, it won't get updated and will still be false.
    first = topic.replies.to_a.first
    assert_equal reply.id, first.id
    assert_equal true, first.approved?
  end

  def test_to_a_should_dup_target
    ary    = topics(:first).replies.to_a
    target = topics(:first).replies.target

    assert_not_same target, ary
  end

  def test_merging_with_custom_attribute_writer
    bulb = Bulb.new(color: "red")
    assert_equal "RED!", bulb.color

    car = Car.create!
    car.bulbs << bulb

    assert_equal "RED!", car.bulbs.to_a.first.color
  end

  def test_abstract_class_with_polymorphic_has_many
    post = SubStiPost.create! title: "fooo", body: "baa"
    tagging = Tagging.create! taggable: post
    assert_equal [tagging], post.taggings
  end

  def test_with_polymorphic_has_many_with_custom_columns_name
    post = Post.create! title: "foo", body: "bar"
    image = Image.create!

    post.images << image

    assert_equal [image], post.images
    assert_equal post, image.imageable
  end

  def test_joining_through_a_polymorphic_association_with_a_where_clause
    writer   = humans(:gordon)
    category = categories(:general)
    TypedEssay.create! category: category, writer: writer

    assert_equal 1, Category.joins(:human_writers_of_typed_essays).count
  end

  def test_build_with_polymorphic_has_many_does_not_allow_to_override_type_and_id
    welcome = posts(:welcome)
    tagging = welcome.taggings.build(taggable_id: 99, taggable_type: "ShouldNotChange")

    assert_equal welcome.id, tagging.taggable_id
    assert_equal "Post", tagging.taggable_type
  end

  def test_build_from_polymorphic_association_sets_inverse_instance
    post = Post.new
    tagging = post.taggings.build

    assert_equal post, tagging.taggable
  end

  def test_dont_call_save_callbacks_twice_on_has_many
    firm = companies(:first_firm)
    contract = firm.contracts.create!

    assert_equal 1, contract.hi_count
    assert_equal 1, contract.bye_count
  end

  def test_association_attributes_are_available_to_after_initialize
    car = Car.create(name: "honda")
    bulb = car.bulbs.build

    assert_equal car.id, bulb.attributes_after_initialize["car_id"]
  end

  def test_attributes_are_set_when_initialized_from_has_many_null_relationship
    car  = Car.new name: "honda"
    bulb = car.bulbs.where(name: "headlight").first_or_initialize
    assert_equal "headlight", bulb.name
  end

  def test_attributes_are_set_when_initialized_from_polymorphic_has_many_null_relationship
    post    = Post.new title: "title", body: "bar"
    tag     = Tag.create!(name: "foo")

    tagging = post.taggings.where(tag: tag).first_or_initialize

    assert_equal tag.id, tagging.tag_id
    assert_equal "Post", tagging.taggable_type
  end

  def test_replace
    car = Car.create(name: "honda")
    bulb1 = car.bulbs.create
    bulb2 = Bulb.create

    assert_equal [bulb1], car.bulbs
    car.bulbs.replace([bulb2])
    assert_equal [bulb2], car.bulbs
    assert_equal [bulb2], car.reload.bulbs
  end

  def test_replace_returns_target
    car = Car.create(name: "honda")
    bulb1 = car.bulbs.create
    bulb2 = car.bulbs.create
    bulb3 = Bulb.create

    assert_equal [bulb1, bulb2], car.bulbs
    result = car.bulbs.replace([bulb3, bulb1])
    assert_equal [bulb1, bulb3], car.bulbs
    assert_equal [bulb1, bulb3], result
  end

  def test_collection_association_with_private_kernel_method
    firm = companies(:first_firm)
    assert_equal [accounts(:signals37)], firm.accounts.open
    assert_equal [accounts(:signals37)], firm.accounts.available
  end

  def test_association_with_or_doesnt_set_inverse_instance_key
    firm = companies(:first_firm)
    accounts = firm.accounts.or(Account.where(firm_id: nil)).order(:id)
    assert_equal [firm.id, nil], accounts.map(&:firm_id)
  end

  def test_association_with_rewhere_doesnt_set_inverse_instance_key
    firm = companies(:first_firm)
    accounts = firm.accounts.rewhere(firm_id: [firm.id, nil]).order(:id)
    assert_equal [firm.id, nil], accounts.map(&:firm_id)
  end

  test "first_or_initialize adds the record to the association" do
    firm = Firm.create! name: "omg"
    client = firm.clients_of_firm.where(name: "lol").first_or_initialize do
      assert_equal 5, Client.count
    end
    assert_equal "lol", client.name
    assert_equal [client], firm.clients_of_firm
  end

  test "first_or_create adds the record to the association" do
    firm = Firm.create! name: "omg"
    firm.clients_of_firm.load_target
    client = firm.clients_of_firm.where(name: "lol").first_or_create do
      assert_equal 5, Client.count
    end
    assert_equal "lol", client.name
    assert_equal [client], firm.clients_of_firm
    assert_equal [client], firm.reload.clients_of_firm
  end

  test "first_or_create! adds the record to the association" do
    firm = Firm.create! name: "omg"
    firm.clients_of_firm.load_target
    client = firm.clients_of_firm.where(name: "lol").first_or_create! do
      assert_equal 5, Client.count
    end
    assert_equal "lol", client.name
    assert_equal [client], firm.clients_of_firm
    assert_equal [client], firm.reload.clients_of_firm
  end

  test "delete_all, when not loaded, doesn't load the records" do
    post = posts(:welcome)

    assert post.taggings_with_delete_all.count > 0
    assert_not_predicate post.taggings_with_delete_all, :loaded?

    # 2 queries: one DELETE and another to update the counter cache
    assert_queries_count(2) do
      post.taggings_with_delete_all.delete_all
    end
  end

  test "has many associations on new records use null relations" do
    post = Post.new

    assert_no_queries do
      assert_equal [], post.comments
      assert_equal [], post.comments.where(body: "omg")
      assert_equal [], post.comments.pluck(:body)
      assert_equal 0,  post.comments.sum(:id)
      assert_equal 0,  post.comments.count
    end
  end

  test "collection proxy respects default scope" do
    author = authors(:mary)
    assert_not_predicate author.first_posts, :exists?
  end

  test "association with extend option" do
    post = posts(:welcome)
    assert_equal "lifo", post.comments_with_extend.author
    assert_equal "hello :)", post.comments_with_extend.greeting
  end

  test "association with extend option with multiple extensions" do
    post = posts(:welcome)
    assert_equal "lifo", post.comments_with_extend_2.author
    assert_equal "hullo :)", post.comments_with_extend_2.greeting
  end

  test "extend option affects per association" do
    post = posts(:welcome)
    assert_equal "lifo", post.comments_with_extend.author
    assert_equal "lifo", post.comments_with_extend_2.author
    assert_equal "hello :)", post.comments_with_extend.greeting
    assert_equal "hullo :)", post.comments_with_extend_2.greeting
  end

  test "delete record with complex joins" do
    david = authors(:david)

    post = david.posts.first
    post.type = "PostWithSpecialCategorization"
    post.save

    categorization = post.categorizations.first
    categorization.special = true
    categorization.save

    assert_not_equal [], david.posts_with_special_categorizations
    david.posts_with_special_categorizations = []
    assert_equal [], david.posts_with_special_categorizations
  end

  test "does not duplicate associations when used with natural primary keys" do
    speedometer = Speedometer.create!(id: "4")
    speedometer.minivans.create!(minivan_id: "a-van-red", name: "a van", color: "red")

    assert_equal 1, speedometer.minivans.to_a.size, "Only one association should be present:\n#{speedometer.minivans.to_a}"
    assert_equal 1, speedometer.reload.minivans.to_a.size
  end

  test "can unscope the default scope of the associated model" do
    car = Car.create!
    bulb1 = Bulb.create! name: "defaulty", car: car
    bulb2 = Bulb.create! name: "other",    car: car

    assert_equal [bulb1], car.bulbs
    assert_equal [bulb1, bulb2], car.all_bulbs.sort_by(&:id)
    assert_equal [bulb1, bulb2], Car.includes(:all_bulbs).find(car.id).all_bulbs.sort_by(&:id)
    assert_equal [bulb1, bulb2], Car.eager_load(:all_bulbs).find(car.id).all_bulbs.sort_by(&:id)
  end

  test "can unscope and where the default scope of the associated model" do
    car = Car.create!
    bulb1 = Bulb.create! name: "defaulty", car: car
    bulb2 = Bulb.create! name: "other",    car: car

    assert_equal [bulb1], car.bulbs
    assert_equal [bulb2], car.other_bulbs
  end

  test "can rewhere the default scope of the associated model" do
    car = Car.create!
    bulb1 = Bulb.create! name: "defaulty", car: car
    bulb2 = Bulb.create! name: "old",      car: car

    assert_equal [bulb1], car.bulbs
    assert_equal [bulb2], car.old_bulbs
  end

  test "unscopes the default scope of associated model when used with include" do
    car = Car.create!
    bulb1 = Bulb.create! name: "defaulty", car: car
    bulb2 = Bulb.create! name: "other",    car: car

    assert_equal [bulb1, bulb2], Car.includes(:all_bulbs2).find(car.id).all_bulbs2.sort_by(&:id)
    assert_equal [bulb1, bulb2], Car.eager_load(:all_bulbs2).find(car.id).all_bulbs2.sort_by(&:id)
  end

  test "raises RecordNotDestroyed when replaced child can't be destroyed" do
    car = Car.create!
    original_child = FailedBulb.create!(car: car)

    error = assert_raise(ActiveRecord::RecordNotDestroyed) do
      car.failed_bulbs = [FailedBulb.create!]
    end

    assert_equal [original_child], car.reload.failed_bulbs
    assert_equal "Failed to destroy FailedBulb with #{FailedBulb.primary_key}=#{original_child.id}", error.message
  end

  test "updates counter cache when default scope is given" do
    topic = DefaultRejectedTopic.create approved: true

    assert_difference "topic.reload.replies_count", 1 do
      topic.approved_replies.create!
    end
  end

  test "dangerous association name raises ArgumentError" do
    [:errors, "errors", :save, "save"].each do |name|
      assert_raises(ArgumentError, "Association #{name} should not be allowed") do
        Class.new(ActiveRecord::Base) do
          has_many name
        end
      end
    end
  end

  test "passes custom context validation to validate children" do
    pirate = FamousPirate.new
    pirate.famous_ships << ship = FamousShip.new

    assert_predicate pirate, :valid?
    assert_not pirate.valid?(:conference)
    assert_equal "can't be blank", ship.errors[:name].first
  end

  test "association with instance dependent scope" do
    bob = authors(:bob)
    Post.create!(title: "signed post by bob", body: "stuff", author: authors(:bob))
    Post.create!(title: "anonymous post", body: "more stuff", author: authors(:bob))
    assert_equal ["misc post by bob", "other post by bob",
                  "signed post by bob"], bob.posts_with_signature.map(&:title).sort

    assert_equal [], authors(:david).posts_with_signature.map(&:title)
  end

  test "associations autosaves when object is already persisted" do
    bulb = Bulb.create!
    tyre = Tyre.create!

    car = Car.create!(name: "honda") do |c|
      c.bulbs << bulb
      c.tyres << tyre
    end

    assert_equal [nil, "honda"], car.saved_change_to_name

    assert_equal 1, car.bulbs.count
    assert_equal 1, car.tyres.count
  end

  test "associations replace in memory when records have the same id" do
    bulb = Bulb.create!
    car = Car.create!(name: "honda", bulbs: [bulb])

    assert_equal [nil, "honda"], car.saved_change_to_name

    new_bulb = Bulb.find(bulb.id)
    new_bulb.name = "foo"
    car.bulbs = [new_bulb]

    assert_equal "foo", car.bulbs.first.name
  end

  test "in memory replacement executes no queries" do
    bulb = Bulb.create!
    car = Car.create!(name: "honda", bulbs: [bulb])

    assert_equal [nil, "honda"], car.saved_change_to_name

    new_bulb = Bulb.find(bulb.id)

    assert_no_queries do
      car.bulbs = [new_bulb]
    end
  end

  test "in memory replacements do not execute callbacks" do
    raise_after_add = false
    klass = Class.new(ActiveRecord::Base) do
      self.table_name = :cars
      has_many :bulbs, after_add: proc { raise if raise_after_add }

      def self.name
        "Car"
      end
    end
    bulb = Bulb.create!
    car = klass.create!(bulbs: [bulb])

    new_bulb = Bulb.find(bulb.id)
    raise_after_add = true

    assert_nothing_raised do
      car.bulbs = [new_bulb]
    end
  end

  test "in memory replacements sets inverse instance" do
    bulb = Bulb.create!
    car = Car.create!(name: "honda", bulbs: [bulb])

    assert_equal [nil, "honda"], car.saved_change_to_name

    new_bulb = Bulb.find(bulb.id)
    car.bulbs = [new_bulb]

    assert_same car, new_bulb.car
  end

  test "reattach to new objects replaces inverse association and foreign key" do
    bulb = Bulb.create!(car: Car.create!)
    assert bulb.car_id
    car = Car.new
    car.bulbs << bulb
    assert_equal car, bulb.car
    assert_nil bulb.car_id
  end

  test "in memory replacement maintains order" do
    first_bulb = Bulb.create!
    second_bulb = Bulb.create!
    car = Car.create!(name: "honda", bulbs: [first_bulb, second_bulb])

    assert_equal [nil, "honda"], car.saved_change_to_name

    same_bulb = Bulb.find(first_bulb.id)
    car.bulbs = [second_bulb, same_bulb]

    assert_equal [first_bulb, second_bulb], car.bulbs
  end

  test "association size calculation works with default scoped selects when not previously fetched" do
    firm = Firm.create!(name: "Firm")
    5.times { firm.developers_with_select << Developer.create!(name: "Developer") }

    same_firm = Firm.find(firm.id)
    assert_equal 5, same_firm.developers_with_select.size
  end

  test "prevent double insertion of new object when the parent association loaded in the after save callback" do
    reset_callbacks(Bulb, :save) do
      Bulb.after_save { |record| record.car.bulbs.load }

      car = Car.create!
      car.bulbs << Bulb.new

      assert_equal 1, car.bulbs.size
    end
  end

  test "prevent double firing the before save callback of new object when the parent association saved in the callback" do
    reset_callbacks(Bulb, :save) do
      count = 0
      Bulb.before_save { |record| record.car.save && count += 1 }

      car = Car.create!
      car.bulbs.create!

      assert_equal 1, count
    end
  end

  test "calling size on an association that has not been loaded performs a query" do
    car = Car.create!
    Bulb.create(car_id: car.id)

    car_two = Car.create!

    assert_queries_count(1) do
      assert_equal 1, car.bulbs.size
    end

    assert_queries_count(1) do
      assert_equal 0, car_two.bulbs.size
    end
  end

  test "calling size on an association that has been loaded does not perform query" do
    car = Car.create!
    Bulb.create(car_id: car.id)
    car.bulb_ids

    car_two = Car.create!
    car_two.bulb_ids

    assert_no_queries do
      assert_equal 1, car.bulbs.size
    end

    assert_no_queries do
      assert_equal 0, car_two.bulbs.size
    end
  end

  test "calling empty on an association that has not been loaded performs a query" do
    car = Car.create!
    Bulb.create(car_id: car.id)

    car_two = Car.create!

    assert_queries_count(1) do
      assert_not_empty car.bulbs
    end

    assert_queries_count(1) do
      assert_empty car_two.bulbs
    end
  end

  test "calling empty on an association that has been loaded does not performs query" do
    car = Car.create!
    Bulb.create(car_id: car.id)
    car.bulb_ids

    car_two = Car.create!
    car_two.bulb_ids

    assert_no_queries do
      assert_not_empty car.bulbs
    end

    assert_no_queries do
      assert_empty car_two.bulbs
    end
  end

  class AuthorWithErrorDestroyingAssociation < ActiveRecord::Base
    self.table_name = "authors"
    has_many :posts_with_error_destroying,
      class_name: "PostWithErrorDestroying",
      foreign_key: :author_id,
      dependent: :destroy
  end

  class PostWithErrorDestroying < ActiveRecord::Base
    self.table_name = "posts"
    self.inheritance_column = nil
    before_destroy -> { throw :abort }
  end

  def test_destroy_does_not_raise_when_association_errors_on_destroy
    assert_no_difference "AuthorWithErrorDestroyingAssociation.count" do
      author = AuthorWithErrorDestroyingAssociation.first

      assert_not author.destroy
    end
  end

  def test_destroy_with_bang_bubbles_errors_from_associations
    error = assert_raises ActiveRecord::RecordNotDestroyed do
      AuthorWithErrorDestroyingAssociation.first.destroy!
    end

    assert_instance_of PostWithErrorDestroying, error.record
  end

  def test_ids_reader_memoization
    car = Car.create!(name: "Tofaş")
    bulb = Bulb.create!(car: car)

    assert_equal [bulb.id], car.bulb_ids
    assert_no_queries { car.bulb_ids }

    bulb2 = car.bulbs.create!

    assert_equal [bulb.id, bulb2.id], car.bulb_ids.sort
    assert_no_queries { car.bulb_ids }
  end

  def test_loading_association_in_validate_callback_doesnt_affect_persistence
    reset_callbacks(Bulb, :validation) do
      Bulb.after_validation { |record| record.car.bulbs.load }

      car = Car.create!(name: "Car")
      bulb = car.bulbs.create!

      assert_equal [bulb], car.bulbs
    end
  end

  def test_create_children_could_be_rolled_back_by_after_save
    firm = Firm.create!(name: "A New Firm, Inc")
    assert_no_difference "Client.count" do
      client = firm.clients.create(name: "New Client") do |cli|
        cli.rollback_on_save = true
        assert_not cli.rollback_on_create_called
      end
      assert client.rollback_on_create_called
    end
  end

  def test_has_many_with_out_of_range_value
    reference = Reference.create!(id: 2147483648) # out of range in the integer
    assert_equal [], reference.ideal_jobs
  end

  def test_has_many_preloading_with_duplicate_records
    posts = Post.joins(:comments).preload(:comments).order(:id).to_a
    assert_equal [1, 2], posts.first.comments.map(&:id).sort
  end

  def test_has_many_association_with_same_foreign_key_name
    assert_nothing_raised do
      firm = Firm.find(15)
      assert_not_nil(firm.comments.first)
    end
  end

  def test_key_ensuring_owner_was_is_not_valid_without_dependent_option
    error = assert_raises(ArgumentError) do
      Class.new(ActiveRecord::Base) do
        has_many :books, ensuring_owner_was: :destroyed?
      end
    end

    assert_match(/Unknown key: :ensuring_owner_was/, error.message)
  end

  def test_key_ensuring_owner_was_is_valid_when_dependent_option_is_destroy_async
    Class.new(ActiveRecord::Base) do
      self.destroy_association_async_job = Class.new

      has_many :books, dependent: :destroy_async, ensuring_owner_was: :destroyed?
    end
  end

  test "composite primary key malformed association class" do
    error = assert_raises(ActiveRecord::CompositePrimaryKeyMismatchError) do
      order = Cpk::BrokenOrder.new(id: [1, 2], books: [Cpk::Book.new(title: "Some book")])
      order.save!
    end

    assert_equal(<<~MESSAGE.squish, error.message)
      Association Cpk::BrokenOrder#books primary key ["shop_id", "status"]
      doesn't match with foreign key broken_order_id. Please specify query_constraints, or primary_key and foreign_key values.
    MESSAGE
  end

  test "composite primary key malformed association owner class" do
    error = assert_raises(ActiveRecord::CompositePrimaryKeyMismatchError) do
      order = Cpk::BrokenOrderWithNonCpkBooks.new(id: [1, 2], books: [Cpk::NonCpkBook.new(title: "Some book")])
      order.save!
    end

    assert_equal(<<~MESSAGE.squish, error.message)
    Association Cpk::BrokenOrderWithNonCpkBooks#books primary key [\"shop_id\", \"status\"]
    doesn't match with foreign key broken_order_with_non_cpk_books_id. Please specify query_constraints, or primary_key and foreign_key values.
    MESSAGE
  end

  private
    def force_signal37_to_load_all_clients_of_firm
      companies(:first_firm).clients_of_firm.load_target
    end
end