[#4183] Implemented Contract#profit_budget and Deliverable#profit_budget.

test/unit/contract_test.rb

@@ -80,4 +80,23 @@ class ContractTest < ActiveSupport::TestCase
       assert_equal 10_000 + (30 * 100), contract.total_budget
     end
   end
+
+  context "#profit_budget" do
+    should "sum all of the profit budgets of the Deliverables" do
+      contract = Contract.generate!(:billable_rate => 100.0)
+
+      contract.deliverables << @deliverable_1 = FixedDeliverable.generate!(:total => 10_000)
+      LaborBudget.generate!(:deliverable => @deliverable_1, :budget => 2000)
+      OverheadBudget.generate!(:deliverable => @deliverable_1, :budget => 2000)
+      
+      contract.deliverables << @deliverable_2 = HourlyDeliverable.generate!
+      LaborBudget.generate!(:deliverable => @deliverable_2, :hours => 10)
+      OverheadBudget.generate!(:deliverable => @deliverable_2, :hours => 20)
+
+      assert_equal 10_000 - 4000, @deliverable_1.profit_budget
+      assert_equal (30 * 100.0) - 0, @deliverable_2.profit_budget
+      assert_equal 9000, contract.profit_budget
+      
+    end
+  end
 end

test/unit/fixed_deliverable_test.rb

@@ -0,0 +1,24 @@
+require File.dirname(__FILE__) + '/../test_helper'
+
+class FixedDeliverableTest < ActiveSupport::TestCase
+  context "#profit_budget" do
+    context "with no labor budget, no overhead budget" do
+      should "equal the total" do
+        assert_equal 1000, FixedDeliverable.generate(:total => 1_000).profit_budget
+      end
+    end
+
+    should "be the total minus the sum of all of the budgets" do
+      deliverable = FixedDeliverable.generate(:total => 1_000)
+      LaborBudget.generate!(:deliverable => deliverable, :budget => 200)
+      LaborBudget.generate!(:deliverable => deliverable, :budget => 200)
+      OverheadBudget.generate!(:deliverable => deliverable, :budget => 200)
+
+      assert_equal 400, deliverable.profit_budget
+    end
+
+    should "be 0 if there is no total" do
+      assert_equal 0, FixedDeliverable.generate(:total => nil).profit_budget
+    end
+  end
+end

test/unit/hourly_deliverable_test.rb

@@ -49,4 +49,26 @@ class HourlyDeliverableTest < ActiveSupport::TestCase
       
     end
   end
+
+  context "#profit_budget" do
+    setup do
+      @contract = Contract.generate!(:billable_rate => 100.0)
+      @deliverable = HourlyDeliverable.generate!(:contract => @contract)
+    end
+    
+    context "with no labor budget, no overhead budget" do
+      should "be 0 (no hours available to bill)" do
+        assert_equal 0, @deliverable.profit_budget
+      end
+    end
+
+    should "be the total minus the sum of all of the budgets' amounts" do
+      LaborBudget.generate!(:deliverable => @deliverable, :hours => 10, :budget => 2000)
+      LaborBudget.generate!(:deliverable => @deliverable, :hours => 5, :budget => 1000)
+      OverheadBudget.generate!(:deliverable => @deliverable, :hours => 15, :budget => 2000)
+
+      assert_equal 30 * 100, @deliverable.total
+      assert_equal 3000 - (2000 + 1000 + 2000), @deliverable.profit_budget
+    end
+  end
 end