diff --git a/app/models/contract.rb b/app/models/contract.rb index 3c3f440..958bf07 100644 --- a/app/models/contract.rb +++ b/app/models/contract.rb @@ -1,3 +1,8 @@ class Contract < ActiveRecord::Base unloadable + + # Associations + belongs_to :project + belongs_to :account_executive, :class_name => 'User', :foreign_key => 'account_executive_id' + end diff --git a/test/unit/contract_test.rb b/test/unit/contract_test.rb index 9095528..c128b6e 100644 --- a/test/unit/contract_test.rb +++ b/test/unit/contract_test.rb @@ -1,4 +1,6 @@ require File.dirname(__FILE__) + '/../test_helper' class ContractTest < ActiveSupport::TestCase + should_belong_to :account_executive + should_belong_to :project end