From e96bdccb428cf630d64fd35c03fc00732ef9e11d Mon Sep 17 00:00:00 2001
From: Eric Davis <edavis@littlestreamsoftware.com>
Date: Wed, 28 May 2008 13:13:08 -0700
Subject: [PATCH] Cleanup of values when a project is empty. #1137

---
 app/models/budget.rb                  |  9 +++++++--
 app/views/deliverables/index.html.erb |  4 ++--
 spec/models/budget_spec.rb            | 11 +++++++++++
 3 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/app/models/budget.rb b/app/models/budget.rb
index a951994..2828265 100644
--- a/app/models/budget.rb
+++ b/app/models/budget.rb
@@ -38,7 +38,12 @@ class Budget
   end
   
   def budget_ratio
-    return ((self.spent / self.budget) * 100).round
+    budget = self.budget # cache result
+    if budget > 0.0
+      return ((self.spent / budget) * 100).round 
+    else
+      self.progress
+    end
   end
   
   def spent
@@ -50,7 +55,7 @@ class Budget
   end
   
   def overruns
-    if self.left > 0
+    if self.left >= 0
       return 0
     else
       return self.left * -1
diff --git a/app/views/deliverables/index.html.erb b/app/views/deliverables/index.html.erb
index d539e0a..d65822b 100644
--- a/app/views/deliverables/index.html.erb
+++ b/app/views/deliverables/index.html.erb
@@ -20,10 +20,10 @@
   Overruns: <%= h number_to_currency(@budget.overruns, :precision => 0) %>
 </p>
 <p>
-  Next Due Date: <%= h distance_of_time_in_words_to_now(@budget.next_due_date) %>
+  Next Due Date: <%= h distance_of_time_in_words_to_now(@budget.next_due_date) if @budget.next_due_date %>
 </p>
 <p>
-  Completion: <%= h distance_of_time_in_words_to_now(@budget.final_due_date) %>
+  Completion: <%= h distance_of_time_in_words_to_now(@budget.final_due_date) if @budget.final_due_date %>
 </p>
 <p>
   Profit: TODO
diff --git a/spec/models/budget_spec.rb b/spec/models/budget_spec.rb
index 2e8add8..b78afeb 100644
--- a/spec/models/budget_spec.rb
+++ b/spec/models/budget_spec.rb
@@ -224,6 +224,17 @@ describe Budget,'.budget_ratio' do
     @budget.should_receive(:spent).and_return(2000.00)
     @budget.budget_ratio.should eql(40)
   end
+  
+  it 'should return progress if the budget is < 0' do
+    @project = mock_model(Project)
+    Project.stub!(:find).with(@project.id).and_return(@project)
+    
+    @budget = Budget.new(@project.id)  
+    @budget.should_receive(:budget).and_return(0.0)
+    @budget.should_receive(:progress).and_return(50)
+    @budget.budget_ratio.should eql(50)
+  end
+
 end