company.rb 8.2 KB
Newer Older
1 2 3 4 5
class AbstractCompany < ActiveRecord::Base
  self.abstract_class = true
end

class Company < AbstractCompany
6
  self.sequence_name = :companies_nonstd_seq
7 8

  validates_presence_of :name
9 10

  has_one :dummy_account, :foreign_key => "firm_id", :class_name => "Account"
11 12
  has_many :contracts
  has_many :developers, :through => :contracts
J
Jeremy Kemper 已提交
13

14 15 16 17 18
  scope :of_first_firm, lambda {
    joins(:account => :firm).
    where('firms.id' => 1)
  }

19 20 21
  def arbitrary_method
    "I am Jack's profound disappointment"
  end
22 23 24 25 26 27

  private

  def private_method
    "I am Jack's innermost fears and aspirations"
  end
D
Initial  
David Heinemeier Hansson 已提交
28 29
end

30 31 32
module Namespaced
  class Company < ::Company
  end
33 34 35 36 37 38 39

  class Firm < ::Company
    has_many :clients, :class_name => 'Namespaced::Client'
  end

  class Client < ::Company
  end
40
end
D
Initial  
David Heinemeier Hansson 已提交
41 42

class Firm < Company
43 44
  to_param :name

45
  has_many :clients, -> { order "id" }, :dependent => :destroy, :before_remove => :log_before_remove, :after_remove  => :log_after_remove
46
  has_many :unsorted_clients, :class_name => "Client"
47
  has_many :unsorted_clients_with_symbol, :class_name => :Client
48
  has_many :clients_sorted_desc, -> { order "id DESC" }, :class_name => "Client"
49
  has_many :clients_of_firm, -> { order "id" }, :foreign_key => "client_of", :class_name => "Client", :inverse_of => :firm
50
  has_many :clients_ordered_by_name, -> { order "name" }, :class_name => "Client"
51
  has_many :unvalidated_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :validate => false
52 53 54 55 56 57
  has_many :dependent_clients_of_firm, -> { order "id" }, :foreign_key => "client_of", :class_name => "Client", :dependent => :destroy
  has_many :exclusively_dependent_clients_of_firm, -> { order "id" }, :foreign_key => "client_of", :class_name => "Client", :dependent => :delete_all
  has_many :limited_clients, -> { limit 1 }, :class_name => "Client"
  has_many :clients_with_interpolated_conditions, ->(firm) { where "rating > #{firm.rating}" }, :class_name => "Client"
  has_many :clients_like_ms, -> { where("name = 'Microsoft'").order("id") }, :class_name => "Client"
  has_many :clients_like_ms_with_hash_conditions, -> { where(:name => 'Microsoft').order("id") }, :class_name => "Client"
58
  has_many :plain_clients, :class_name => 'Client'
59 60
  has_many :clients_using_primary_key, :class_name => 'Client',
           :primary_key => 'name', :foreign_key => 'firm_name'
61 62
  has_many :clients_using_primary_key_with_delete_all, :class_name => 'Client',
           :primary_key => 'name', :foreign_key => 'firm_name', :dependent => :delete_all
63 64
  has_many :clients_grouped_by_firm_id, -> { group("firm_id").select("firm_id") }, :class_name => "Client"
  has_many :clients_grouped_by_name, -> { group("name").select("name") }, :class_name => "Client"
D
Initial  
David Heinemeier Hansson 已提交
65

66 67
  has_one :account, :foreign_key => "firm_id", :dependent => :destroy, :validate => true
  has_one :unvalidated_account, :foreign_key => "firm_id", :class_name => 'Account', :validate => false
68 69
  has_one :account_with_select, -> { select("id, firm_id") }, :foreign_key => "firm_id", :class_name=>'Account'
  has_one :readonly_account, -> { readonly }, :foreign_key => "firm_id", :class_name => "Account"
70 71
  # added order by id as in fixtures there are two accounts for Rails Core
  # Oracle tests were failing because of that as the second fixture was selected
72
  has_one :account_using_primary_key, -> { order('id') }, :primary_key => "firm_id", :class_name => "Account"
73
  has_one :account_using_foreign_and_primary_keys, :foreign_key => "firm_name", :primary_key => "name", :class_name => "Account"
74
  has_one :deletable_account, :foreign_key => "firm_id", :class_name => "Account", :dependent => :delete
75

76
  has_one :account_limit_500_with_hash_conditions, -> { where :credit_limit => 500 }, :foreign_key => "firm_id", :class_name => "Account"
77

78 79 80
  has_one :unautosaved_account, :foreign_key => "firm_id", :class_name => 'Account', :autosave => false
  has_many :accounts
  has_many :unautosaved_accounts, :foreign_key => "firm_id", :class_name => 'Account', :autosave => false
81

82
  has_many :association_with_references, -> { references(:foo) }, :class_name => 'Client'
83

84 85 86 87 88 89 90 91 92 93 94 95
  def log
    @log ||= []
  end

  private
    def log_before_remove(record)
      log << "before_remove#{record.id}"
    end

    def log_after_remove(record)
      log << "after_remove#{record.id}"
    end
D
Initial  
David Heinemeier Hansson 已提交
96 97
end

98
class DependentFirm < Company
99 100
  has_one :account, :foreign_key => "firm_id", :dependent => :nullify
  has_many :companies, :foreign_key => 'client_of', :dependent => :nullify
101
  has_one :company, :foreign_key => 'client_of', :dependent => :nullify
102 103
end

104 105 106 107 108 109 110 111
class RestrictedWithExceptionFirm < Company
  has_one :account, -> { order("id") }, :foreign_key => "firm_id", :dependent => :restrict_with_exception
  has_many :companies, -> { order("id") }, :foreign_key => 'client_of', :dependent => :restrict_with_exception
end

class RestrictedWithErrorFirm < Company
  has_one :account, -> { order("id") }, :foreign_key => "firm_id", :dependent => :restrict_with_error
  has_many :companies, -> { order("id") }, :foreign_key => 'client_of', :dependent => :restrict_with_error
112 113
end

D
Initial  
David Heinemeier Hansson 已提交
114 115 116
class Client < Company
  belongs_to :firm, :foreign_key => "client_of"
  belongs_to :firm_with_basic_id, :class_name => "Firm", :foreign_key => "firm_id"
117
  belongs_to :firm_with_select, -> { select("id") }, :class_name => "Firm", :foreign_key => "firm_id"
D
Initial  
David Heinemeier Hansson 已提交
118
  belongs_to :firm_with_other_name, :class_name => "Firm", :foreign_key => "client_of"
119
  belongs_to :firm_with_condition, -> { where "1 = ?", 1 }, :class_name => "Firm", :foreign_key => "client_of"
120
  belongs_to :firm_with_primary_key, :class_name => "Firm", :primary_key => "name", :foreign_key => "firm_name"
121
  belongs_to :firm_with_primary_key_symbols, :class_name => "Firm", :primary_key => :name, :foreign_key => :firm_name
122 123
  belongs_to :readonly_firm, -> { readonly }, :class_name => "Firm", :foreign_key => "firm_id"
  belongs_to :bob_firm, -> { where :name => "Bob" }, :class_name => "Firm", :foreign_key => "client_of"
124
  has_many :accounts, :through => :firm, :source => :accounts
125
  belongs_to :account
126

127 128 129 130
  validate do
    firm
  end

J
Jon Leighton 已提交
131 132 133 134 135 136 137 138 139 140 141 142
  class RaisedOnSave < RuntimeError; end
  attr_accessor :raise_on_save
  before_save do
    raise RaisedOnSave if raise_on_save
  end

  class RaisedOnDestroy < RuntimeError; end
  attr_accessor :raise_on_destroy
  before_destroy do
    raise RaisedOnDestroy if raise_on_destroy
  end

143 144 145 146 147 148 149 150 151 152 153 154 155
  # Record destruction so we can test whether firm.clients.clear has
  # is calling client.destroy, deleting from the database, or setting
  # foreign keys to NULL.
  def self.destroyed_client_ids
    @destroyed_client_ids ||= Hash.new { |h,k| h[k] = [] }
  end

  before_destroy do |client|
    if client.firm
      Client.destroyed_client_ids[client.firm.id] << client.id
    end
    true
  end
J
Jeremy Kemper 已提交
156

157 158
  before_destroy :overwrite_to_raise

159 160 161 162
  # Used to test that read and question methods are not generated for these attributes
  def rating?
    query_attribute :rating
  end
163

164 165 166
  def overwrite_to_raise
  end

167 168 169 170 171 172 173
  class << self
    private

    def private_method
      "darkness"
    end
  end
D
Initial  
David Heinemeier Hansson 已提交
174 175
end

176 177
class ExclusivelyDependentFirm < Company
  has_one :account, :foreign_key => "firm_id", :dependent => :delete
178 179
  has_many :dependent_sanitized_conditional_clients_of_firm, -> { order("id").where("name = 'BigShot Inc.'") }, :foreign_key => "client_of", :class_name => "Client", :dependent => :delete_all
  has_many :dependent_conditional_clients_of_firm, -> { order("id").where("name = ?", 'BigShot Inc.') }, :foreign_key => "client_of", :class_name => "Client", :dependent => :delete_all
180
end
D
Initial  
David Heinemeier Hansson 已提交
181 182 183 184 185 186 187 188

class SpecialClient < Client
end

class VerySpecialClient < SpecialClient
end

class Account < ActiveRecord::Base
189
  belongs_to :firm, :class_name => 'Company'
190
  belongs_to :unautosaved_firm, :foreign_key => "firm_id", :class_name => "Firm", :autosave => false
J
Jeremy Kemper 已提交
191

192 193
  alias_attribute :available_credit, :credit_limit

194 195 196 197
  def self.destroyed_account_ids
    @destroyed_account_ids ||= Hash.new { |h,k| h[k] = [] }
  end

198 199 200 201 202
  # Test private kernel method through collection proxy using has_many.
  def self.open
    where('firm_name = ?', '37signals')
  end

203 204 205 206 207 208
  before_destroy do |account|
    if account.firm
      Account.destroyed_account_ids[account.firm.id] << account.id
    end
    true
  end
J
Jeremy Kemper 已提交
209

210 211
  validate :check_empty_credit_limit

D
Initial  
David Heinemeier Hansson 已提交
212
  protected
213 214 215 216

  def check_empty_credit_limit
    errors.add_on_empty "credit_limit"
  end
217 218 219 220 221 222

  private

  def private_method
    "Sir, yes sir!"
  end
J
Jeremy Kemper 已提交
223
end