make sure populator requires a limit for credit-card accounts

jamis · Nov 12, 2009 · d62181b · d62181b
1 parent 3797771
commit d62181b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/lib/populator.rb b/lib/populator.rb
@@ -13,8 +13,9 @@ def initialize(subscription)
     @posts = []
   end
 
-  def account(name, role, starting_balance, balance_date)
-    @accounts << { :name => name, :role => role,
+  def account(name, role, starting_balance, balance_date, limit=nil)
+    raise "you have to specify a limit for credit-card accounts" if role == "credit-card" && limit.nil?
+    @accounts << { :name => name, :role => role, :limit => limit,
         :starting_balance => { :amount => starting_balance, :occurred_on => balance_date }}
   end
 

diff --git a/lib/tasks/demo.rake b/lib/tasks/demo.rake
@@ -35,7 +35,7 @@ namespace :demo do
 
       Populator.for(subscription) do |make|
         make.account("Checking", 'checking', 1_400_00, 45.days.ago)
-        make.account("Mastercard", 'credit-card', -500_00, 45.days.ago)
+        make.account("Mastercard", 'credit-card', -500_00, 45.days.ago, 5000_00)
         make.account("Savings", 'other', 750_00, 45.days.ago)
 
         # ---------------------------------------------------