diff --git a/app/models/deliverable.rb b/app/models/deliverable.rb index 56dde0b..8e169ff 100644 --- a/app/models/deliverable.rb +++ b/app/models/deliverable.rb @@ -6,6 +6,7 @@ class Deliverable < ActiveRecord::Base belongs_to :manager, :class_name => 'User', :foreign_key => 'manager_id' has_many :labor_budgets has_many :overhead_budgets + has_many :issues accepts_nested_attributes_for :labor_budgets accepts_nested_attributes_for :overhead_budgets diff --git a/db/migrate/006_add_deliverable_id_to_issues.rb b/db/migrate/006_add_deliverable_id_to_issues.rb new file mode 100644 index 0000000..92557d9 --- /dev/null +++ b/db/migrate/006_add_deliverable_id_to_issues.rb @@ -0,0 +1,12 @@ +class AddDeliverableIdToIssues < ActiveRecord::Migration + def self.up + # Skip adding the column if it exists from the Budget plugin + unless Issue.column_names.include?('deliverable_id') + add_column :issues, :deliverable_id, :integer + end + end + + def self.down + remove_column :issues, :deliverable_id + end +end diff --git a/test/unit/deliverable_test.rb b/test/unit/deliverable_test.rb index 67da0f8..b03611d 100644 --- a/test/unit/deliverable_test.rb +++ b/test/unit/deliverable_test.rb @@ -5,6 +5,7 @@ class DeliverableTest < ActiveSupport::TestCase should_belong_to :manager should_have_many :labor_budgets should_have_many :overhead_budgets + should_have_many :issues should_validate_presence_of :title should_validate_presence_of :type