budget 1.0.0

data/lib/budget.rb

@@ -0,0 +1,98 @@
+class Budget
+  attr_reader :income, :expense, :remaining, :total_income, :total_expenses, :month, :year
+  # Your code goes here...
+  def initialize
+    @income = Array.new
+    @expense = Array.new
+    @remaining = 0
+    @total_income = 0
+    @total_expenses = 0
+    @month = Time.now.month
+    @year = Time.now.year
+  end
+
+  def set_month(liMonth, liYear)
+    raise RangeError, 'Month must be an integer 1 - 12' if  ((!liMonth.is_a? Integer) || liMonth < 1 || liMonth > 12)
+    raise RangeError, 'Year must be an integer' if (!liYear.is_a? Integer)
+    @month = liMonth
+    @year = liYear
+  end
+
+  def add_income(pfIncome, psMemo = "Income Item #{self.income.count + 1}", pdDate = Time.now)
+    raise TypeError, 'Amount parameter must be numeric' if !is_numeric?(pfIncome)
+    lhIncome = Hash.new
+    lhIncome["amount"] = pfIncome.round(2)
+    lhIncome["memo"] = psMemo.to_s
+    lhIncome["date"] = pdDate
+    @income << lhIncome
+    set_total_income()
+    set_difference()
+  end
+ 
+  def set_total_income
+    lfIncome = get_income()
+    @total_income = lfIncome.round(2)
+  end
+
+  def add_expense(pfAmount, psMemo = "Expense Item #{self.expense.count + 1}", pdDate = Time.now)
+    raise TypeError, 'Amount parameter must be numeric' if !is_numeric?(pfAmount)
+    lhExpense = Hash.new
+    lhExpense["amount"] = pfAmount.round(2)
+    lhExpense["memo"] = psMemo.to_s
+    lhExpense["date"] = pdDate
+    @expense << lhExpense
+    set_total_expenses()
+    set_difference()
+  end
+
+  def set_total_expenses
+    lfTotalExpenses = get_expenses()
+    @total_expenses = lfTotalExpenses.round(2)
+  end
+
+  def get_income
+    lfTotalIncome = 0
+    @income.each do |i|
+      lfTotalIncome = lfTotalIncome + i["amount"].round(2)
+    end
+    return lfTotalIncome
+  end
+
+  def get_expenses
+    lfTotalExpenses = 0
+    @expense.each do |i|
+      lfTotalExpenses = lfTotalExpenses + i["amount"].round(2)
+    end
+    return lfTotalExpenses
+  end
+
+  def set_difference
+    lfIncome = get_income()
+    lfExpenses = get_expenses()
+    lfDifference = lfIncome - lfExpenses
+    @remaining = lfDifference
+  end
+
+  def delete_income(piIndex)
+    @income.delete_at(piIndex)
+    set_total_income()
+    set_difference()
+  end
+
+  def delete_expense(piIndex)
+    @expense.delete_at(piIndex)
+    set_total_expenses()
+    set_difference()
+  end
+  
+  def complete?
+    if @total_expenses == @total_income
+      return true
+    end
+    return false
+  end
+
+  def is_numeric?(obj) 
+   obj.to_s.match(/\A[+-]?\d+?(\.\d+)?\Z/) == nil ? false : true
+  end
+end

data/test/test_budget.rb

@@ -0,0 +1,56 @@
+require 'test/unit'
+require 'budget'
+
+class BudgetTest < Test::Unit::TestCase
+  def test_budget
+    foo = Budget.new
+    # Test the inital month and year is the current month and year
+    assert_equal Time.now.month, foo.month
+    assert_equal Time.now.year, foo.year
+    # Test we can change month to May 2015
+    foo.set_month(5,2015)
+    assert_equal 5, foo.month
+    assert_equal 2015, foo.year
+    # Test we can add $500 income
+    foo.add_income(500)
+    assert_equal 500, foo.total_income
+    assert_equal 500, foo.remaining
+    assert_equal 0, foo.total_expenses
+    foo.add_income(500)
+    assert_equal 1000, foo.total_income
+    assert_equal 1000, foo.remaining
+    assert_equal 0, foo.total_expenses
+    # Test to make sure memo is set to default with the right number 
+    assert_equal 'Income Item 1', foo.income[0]["memo"]
+    assert_equal 'Income Item 2', foo.income[1]["memo"]
+    foo.add_income(500, 'Paycheck 3')
+    assert_equal 'Paycheck 3', foo.income[2]["memo"]
+    # Test to make sure budget is not complete at the moment
+    assert !foo.complete?
+    # Add a few expenses
+    foo.add_expense(200)
+    foo.add_expense(400)
+    assert_equal 600, foo.total_expenses
+    assert_equal 200, foo.expense[0]["amount"]
+    assert_equal 400, foo.expense[1]["amount"]
+    # Make sure budget still is not complete
+    assert !foo.complete?
+    foo.add_expense(900, 'Mortgage')
+    assert_equal 1500, foo.total_expenses
+    assert foo.complete?
+    assert_equal 0, foo.remaining
+    # Remove last first expense
+    foo.delete_expense(0)
+    assert !foo.complete?
+    assert_equal 1300, foo.total_expenses
+    assert_equal 200, foo.remaining
+    foo.add_expense(200)
+    assert_equal 'Expense Item 3', foo.expense[2]["memo"]
+    foo.delete_income(2)
+    assert !foo.complete?
+    assert_equal -500, foo.remaining
+    assert_equal 1000, foo.total_income
+    foo.add_income(500)
+    assert foo.complete?    
+  end
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: budget
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- DrSayre
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-09-08 00:00:00.000000000 Z
+dependencies: []
+description: A class to create a budget
+email:
+- DrSayre2002@yahoo.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/budget.rb
+- test/test_budget.rb
+homepage: https://github.com/DrSayre/budget
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: A class to create a budget
+test_files:
+- test/test_budget.rb