diff --git a/test/integration/deliverables_edit_test.rb b/test/integration/deliverables_edit_test.rb index 29667f0..ae44254 100644 --- a/test/integration/deliverables_edit_test.rb +++ b/test/integration/deliverables_edit_test.rb @@ -238,4 +238,55 @@ class DeliverablesEditTest < ActionController::IntegrationTest end end + + should "allow extending a Retainer's start and end months" do + @retainer_deliverable = RetainerDeliverable.spawn(:contract => @contract, :manager => @manager, :title => "Retainer") + @retainer_deliverable.labor_budgets << @labor_budget = LaborBudget.spawn(:deliverable => @retainer_deliverable, :budget => 1000, :hours => 10) + @retainer_deliverable.overhead_budgets << @overhead_budget = OverheadBudget.spawn(:deliverable => @retainer_deliverable, :budget => 1000, :hours => 10) + @retainer_deliverable.start_date = '2010-01-01' + @retainer_deliverable.end_date = '2010-12-31' + @retainer_deliverable.save! + assert_equal 12, @retainer_deliverable.months.length + + @first_labor_budget = @retainer_deliverable.labor_budgets.first + @first_overhead_budget = @retainer_deliverable.overhead_budgets.first + @last_labor_budget = @retainer_deliverable.labor_budgets.last + @last_overhead_budget = @retainer_deliverable.overhead_budgets.last + + visit_contract_page(@contract) + click_link_within "#deliverable_details_#{@retainer_deliverable.id}", 'Edit' + assert_response :success + assert_template 'deliverables/edit' + + # Extend the period + fill_in "Start", :with => '2009-01-13' # 12 new months + fill_in "End Date", :with => '2011-12-01' # 12 new months + click_button "Save" + assert_response :success + assert_template 'contracts/show' + + @labor_budgets = @retainer_deliverable.reload.labor_budgets + assert_equal 36, @labor_budgets.length + @labor_budgets[0,12].each do |labor_budget| # First 12 + assert_equal @first_labor_budget.hours, labor_budget.hours + assert_equal @first_labor_budget.budget, labor_budget.budget + end + @labor_budgets[-12,36].each do |labor_budget| # Last 12 + assert_equal @last_labor_budget.hours, labor_budget.hours + assert_equal @last_labor_budget.budget, labor_budget.budget + end + + + @overhead_budgets = @retainer_deliverable.reload.overhead_budgets + assert_equal 36, @overhead_budgets.length + @overhead_budgets[0,12].each do |overhead_budget| # First 12 + assert_equal @first_overhead_budget.hours, overhead_budget.hours + assert_equal @first_overhead_budget.budget, overhead_budget.budget + end + @overhead_budgets[-12,36].each do |overhead_budget| # Last 12 + assert_equal @last_overhead_budget.hours, overhead_budget.hours + assert_equal @last_overhead_budget.budget, overhead_budget.budget + end + + end end