# frozen_string_literal: true

require "bundler/inline"

gemfile(true) do
  source "https://rubygems.org"

  git_source(:github) { |repo| "https://github.com/#{repo}.git" }

  gem "rails", "6.0.2.2"
  gem "sqlite3"
end

require "active_record"
require "minitest/autorun"
require "logger"

# This connection will do for database-independent bug reports.
ActiveRecord::Base.establish_connection(adapter: "sqlite3", database: ":memory:")
ActiveRecord::Base.logger = Logger.new(STDOUT)

ActiveRecord::Schema.define do
  create_table :entities, force: true do |t|
  end

  create_table :memberships, force: true do |t|
    t.integer :parent_id
    t.integer :child_id
  end

  create_table :documents, force: true do |t|
    t.integer :entity_id
  end
end

class Entity < ActiveRecord::Base
  has_many :memberships, foreign_key: :parent_id
  has_many :children,    through: :memberships
  has_many :documents

  has_one  :parent_membership, class_name: 'Membership', foreign_key: :child_id
  has_one  :parent,            through: :parent_membership
  has_many :shared_documents,  through: :parent, source: :documents
end

class Membership < ActiveRecord::Base
  belongs_to :parent, class_name: 'Entity'
  belongs_to :child,  class_name: 'Entity'
end

class Document < ActiveRecord::Base
  belongs_to :customer
end

class BugTest < Minitest::Test
  def test_association_stuff
    parent = Entity.create!
    parent.documents.create!

    3.times do
      parent.children << Entity.create!
    end

    # Each parent and child entities have only one document
    assert_equal 1, Entity.preload(:shared_documents)[1].shared_documents.size

    # Loading the parent alongside the documents will return 3 documents
    assert_equal 1, Entity.preload(:parent, :shared_documents)[1].shared_documents.size
  end
end