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

class Company < AbstractCompany
D
Initial  
David Heinemeier Hansson 已提交
6
  attr_protected :rating
7
  set_sequence_name :companies_nonstd_seq
8 9

  validates_presence_of :name
10 11

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

15 16 17
  def arbitrary_method
    "I am Jack's profound disappointment"
  end
18 19 20 21 22 23

  private

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

26 27 28
module Namespaced
  class Company < ::Company
  end
29 30 31 32 33 34 35

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

  class Client < ::Company
  end
36
end
D
Initial  
David Heinemeier Hansson 已提交
37 38

class Firm < Company
39
  has_many :clients, :order => "id", :dependent => :destroy, :counter_sql =>
40
      "SELECT COUNT(*) FROM companies WHERE firm_id = 1 " +
41 42 43
      "AND (#{QUOTED_TYPE} = 'Client' OR #{QUOTED_TYPE} = 'SpecialClient' OR #{QUOTED_TYPE} = 'VerySpecialClient' )",
      :before_remove => :log_before_remove,
      :after_remove  => :log_after_remove
44
  has_many :unsorted_clients, :class_name => "Client"
D
Initial  
David Heinemeier Hansson 已提交
45 46
  has_many :clients_sorted_desc, :class_name => "Client", :order => "id DESC"
  has_many :clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id"
47
  has_many :unvalidated_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :validate => false
48 49
  has_many :dependent_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id", :dependent => :destroy
  has_many :exclusively_dependent_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id", :dependent => :delete_all
50
  has_many :limited_clients, :class_name => "Client", :limit => 1
D
Initial  
David Heinemeier Hansson 已提交
51
  has_many :clients_like_ms, :conditions => "name = 'Microsoft'", :class_name => "Client", :order => "id"
52
  has_many :clients_with_interpolated_conditions, :class_name => "Client", :conditions => 'rating > #{rating}'
J
Jeremy Kemper 已提交
53
  has_many :clients_like_ms_with_hash_conditions, :conditions => { :name => 'Microsoft' }, :class_name => "Client", :order => "id"
D
Initial  
David Heinemeier Hansson 已提交
54
  has_many :clients_using_sql, :class_name => "Client", :finder_sql => 'SELECT * FROM companies WHERE client_of = #{id}'
55 56 57 58
  has_many :clients_using_multiline_sql, :class_name => "Client", :finder_sql => '
  SELECT
  companies.*
  FROM companies WHERE companies.client_of = #{id}'
59
  has_many :clients_using_counter_sql, :class_name => "Client",
60
           :finder_sql  => 'SELECT * FROM companies WHERE client_of = #{id}',
61 62
           :counter_sql => 'SELECT COUNT(*) FROM companies WHERE client_of = #{id}'
  has_many :clients_using_zero_counter_sql, :class_name => "Client",
63
           :finder_sql  => 'SELECT * FROM companies WHERE client_of = #{id}',
64
           :counter_sql => 'SELECT 0 FROM companies WHERE client_of = #{id}'
65 66 67
  has_many :no_clients_using_counter_sql, :class_name => "Client",
           :finder_sql  => 'SELECT * FROM companies WHERE client_of = 1000',
           :counter_sql => 'SELECT COUNT(*) FROM companies WHERE client_of = 1000'
68
  has_many :clients_using_finder_sql, :class_name => "Client", :finder_sql => 'SELECT * FROM companies WHERE 1=1'
69
  has_many :plain_clients, :class_name => 'Client'
70
  has_many :readonly_clients, :class_name => 'Client', :readonly => true
71 72
  has_many :clients_using_primary_key, :class_name => 'Client',
           :primary_key => 'name', :foreign_key => 'firm_name'
73 74
  has_many :clients_using_primary_key_with_delete_all, :class_name => 'Client',
           :primary_key => 'name', :foreign_key => 'firm_name', :dependent => :delete_all
75 76
  has_many :clients_grouped_by_firm_id, :class_name => "Client", :group => "firm_id", :select => "firm_id"
  has_many :clients_grouped_by_name, :class_name => "Client", :group => "name", :select => "name"
D
Initial  
David Heinemeier Hansson 已提交
77

78 79
  has_one :account, :foreign_key => "firm_id", :dependent => :destroy, :validate => true
  has_one :unvalidated_account, :foreign_key => "firm_id", :class_name => 'Account', :validate => false
80
  has_one :account_with_select, :foreign_key => "firm_id", :select => "id, firm_id", :class_name=>'Account'
81
  has_one :readonly_account, :foreign_key => "firm_id", :class_name => "Account", :readonly => true
82 83 84
  # 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
  has_one :account_using_primary_key, :primary_key => "firm_id", :class_name => "Account", :order => "id"
85
  has_one :account_using_foreign_and_primary_keys, :foreign_key => "firm_name", :primary_key => "name", :class_name => "Account"
86
  has_one :deletable_account, :foreign_key => "firm_id", :class_name => "Account", :dependent => :delete
87

88 89
  has_one :account_limit_500_with_hash_conditions, :foreign_key => "firm_id", :class_name => "Account", :conditions => { :credit_limit => 500 }

90 91 92
  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
93 94 95 96 97 98 99 100 101 102 103 104 105

  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 已提交
106 107
end

108
class DependentFirm < Company
109 110
  has_one :account, :foreign_key => "firm_id", :dependent => :nullify
  has_many :companies, :foreign_key => 'client_of', :dependent => :nullify
111 112
end

113 114 115 116 117
class RestrictedFirm < Company
  has_one :account, :foreign_key => "firm_id", :dependent => :restrict, :order => "id"
  has_many :companies, :foreign_key => 'client_of', :order => "id", :dependent => :restrict
end

D
Initial  
David Heinemeier Hansson 已提交
118 119 120
class Client < Company
  belongs_to :firm, :foreign_key => "client_of"
  belongs_to :firm_with_basic_id, :class_name => "Firm", :foreign_key => "firm_id"
121
  belongs_to :firm_with_select, :class_name => "Firm", :foreign_key => "firm_id", :select => "id"
D
Initial  
David Heinemeier Hansson 已提交
122
  belongs_to :firm_with_other_name, :class_name => "Firm", :foreign_key => "client_of"
123
  belongs_to :firm_with_condition, :class_name => "Firm", :foreign_key => "client_of", :conditions => ["1 = ?", 1]
124
  belongs_to :firm_with_primary_key, :class_name => "Firm", :primary_key => "name", :foreign_key => "firm_name"
125
  belongs_to :firm_with_primary_key_symbols, :class_name => "Firm", :primary_key => :name, :foreign_key => :firm_name
126
  belongs_to :readonly_firm, :class_name => "Firm", :foreign_key => "firm_id", :readonly => true
127 128 129 130 131 132 133 134 135 136 137 138 139 140

  # 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 已提交
141

142 143
  before_destroy :overwrite_to_raise

144 145 146 147
  # Used to test that read and question methods are not generated for these attributes
  def ruby_type
    read_attribute :ruby_type
  end
J
Jeremy Kemper 已提交
148

149 150 151
  def rating?
    query_attribute :rating
  end
152

153 154 155
  def overwrite_to_raise
  end

156 157 158 159 160 161 162
  class << self
    private

    def private_method
      "darkness"
    end
  end
D
Initial  
David Heinemeier Hansson 已提交
163 164
end

165 166 167 168 169 170
class ExclusivelyDependentFirm < Company
  has_one :account, :foreign_key => "firm_id", :dependent => :delete
  has_many :dependent_sanitized_conditional_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id", :dependent => :delete_all, :conditions => "name = 'BigShot Inc.'"
  has_many :dependent_conditional_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id", :dependent => :delete_all, :conditions => ["name = ?", 'BigShot Inc.']
  has_many :dependent_hash_conditional_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id", :dependent => :delete_all, :conditions => {:name => 'BigShot Inc.'}
end
D
Initial  
David Heinemeier Hansson 已提交
171 172 173 174 175 176 177 178

class SpecialClient < Client
end

class VerySpecialClient < SpecialClient
end

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

182 183 184 185 186 187 188 189 190 191
  def self.destroyed_account_ids
    @destroyed_account_ids ||= Hash.new { |h,k| h[k] = [] }
  end

  before_destroy do |account|
    if account.firm
      Account.destroyed_account_ids[account.firm.id] << account.id
    end
    true
  end
J
Jeremy Kemper 已提交
192

193 194
  validate :check_empty_credit_limit

D
Initial  
David Heinemeier Hansson 已提交
195
  protected
196 197 198 199

  def check_empty_credit_limit
    errors.add_on_empty "credit_limit"
  end
200 201 202 203 204 205

  private

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