dev

celsopitta · May 2, 2011 · 1653efa · 1653efa
1 parent 0bd7791
commit 1653efa
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/btc_trader/trader.rb b/lib/btc_trader/trader.rb
@@ -27,7 +27,7 @@ def execute!(rec)
     end
 
     def buy(amount)
-      raise "Order exceeds balance" if amount > @cash_balance
+      raise "Buy order exceeds balance" if amount > @cash_balance
       @cash_balance -= amount
       btc = amount / @price
       @btc_balance  += btc * TRADE_FEE_MULTIPLIER
@@ -36,6 +36,7 @@ def buy(amount)
     end
 
     def sell(amount)
+      raise "Sell order exceeds balance" if amount > @btc_balance
       @btc_balance  -= amount
       cash =  amount * @price
       @cash_balance += cash * TRADE_FEE_MULTIPLIER
@@ -56,7 +57,7 @@ def ror_no_fee
     end
   end
 
-  # A trivial example that buys at the outset and hold until the end
+  # A trivial example that buys at the outset and holds until the end
   module Strategies
     class BuyAndHoldTrader < Trader
       def execute_strategy!