Lynn Trickey's Bank Account

Rakefile

@@ -2,7 +2,7 @@ require 'rake/testtask'
 
 Rake::TestTask.new do |t|
   t.libs = ["lib"]
-  t.warning = true
+  t.warning = false
   t.test_files = FileList['specs/*_spec.rb']
 end
 

lib/account.rb

@@ -0,0 +1,152 @@
+require 'csv'
+
+#Adding this in after the reading this weekend.
+#b/c both savings and money market have interest
+#but are sisters not mother/daughter classes!
+module Interest
+
+  def add_interest(rate)
+    argument("Interest rate >=0") if rate < 0
+
+    interest = @balance * (rate/100)
+    @balance += interest
+
+    return interest
+  end
+
+end
+
+module Bank
+
+  class Account
+    attr_accessor :balance
+    attr_reader :id, :owner
+
+    def initialize(id, start_balance, opendate = nil)
+
+      @id = id
+      @opendate = opendate
+      @balance = set_balance(start_balance)
+    end
+
+    def set_balance(start_balance)
+      if start_balance < 0
+        argument("You cannot initialize a new account with a negative balance.")
+      else
+        start_balance
+      end
+    end
+
+    def self.all
+      accounts = []
+
+      CSV.open("support/accounts.csv").each do |account|
+        accounts << Bank::Account.new(account[0].to_i, account[1].to_i, account[2])
+      end
+      return accounts
+    end
+
+    def self.find(id)
+      @account = nil
+
+      CSV.open("support/accounts.csv").each do |line|
+        if line[0].to_i == id
+          @account = Bank::Account.new(line[0].to_i, line[1].to_i, line[2])
+        end
+      end
+
+      if @account == nil
+        raise ArgumentError.new "This account does not exist!"
+      else
+        return @account
+      end
+    end
+
+    def add_owner(owner)
+      if owner.class == Owner
+        @owner = owner
+      else
+        argument("You must add a class type of Owner.")
+      end
+    end
+
+    def withdraw(withdrawal_amount)
+      withdraw_positive(withdrawal_amount)
+
+      if balance - withdrawal_amount < 0
+        puts "You are going negative."
+        return balance
+      else
+        @balance -= withdrawal_amount
+      end
+    end
+
+    #Should this be private??
+    #b/c only used in check_withdrawals?
+    def withdraw_positive(withdrawal_amount)
+      argument("Withdrawal must be >= 0") if withdrawal_amount < 0
+    end
+
+    def deposit(deposit_amount)
+      if deposit_amount > 0
+        @balance += deposit_amount
+      else
+        argument("Your deposit must be greater than zero.")
+      end
+    end
+
+    def argument(output)
+      raise ArgumentError.new "#{ output }"
+    end
+
+  end
+
+  class Owner
+    attr_reader :last_name, :first_name, :street_address, :city, :state
+
+    def initialize(
+      id = nil, last_name = nil,
+      first_name = nil, street_address = nil,
+      city = nil, state = nil)
+      @id = id
+      @last_name = last_name
+      @first_name = first_name
+      @street_address = street_address
+      @city = city
+      @state = state
+    end
+
+    def self.all
+      owners = []
+      CSV.open("support/owners.csv").each do |owner|
+        owners << Bank::Owner.new(
+        owner[0].to_i, owner[1],
+        owner[2], owner[3],
+        owner[4], owner[5])
+      end
+
+      return owners
+    end
+
+    def self.find(id)
+      @owner = nil
+      CSV.open("support/owners.csv").each do |owner|
+        if owner[0].to_i == id
+          @owner = Bank::Owner.new(
+          owner[0].to_i, owner[1],
+          owner[2], owner[3],
+          owner[4], owner[5])
+        end
+      end
+
+      if @owner == nil
+        raise ArgumentError.new "This owner does not exist!"
+      else
+        return @owner
+      end
+
+    end
+
+  end
+
+end

lib/checking_account.rb

@@ -0,0 +1,57 @@
+module Bank
+
+  class CheckingAccount < Account
+
+    def initialize(id, balance, opendate = nil)
+      super
+      @check_withdrawals = 0
+    end
+
+    def withdraw(withdrawal_amount)
+      original_balance = balance
+
+      super
+
+      if balance == original_balance
+        balance
+      elsif balance - 1 < 0
+        puts "This withdrawal and fee will take your balance below 0."
+        @balance = original_balance
+      else
+        withdrawal_fee
+      end
+    end
+
+    def withdrawal_fee
+      @balance -= 1
+    end
+
+    def withdraw_using_check(withdrawal_amount)
+      withdraw_positive(withdrawal_amount)
+
+      if @balance - withdrawal_amount < -10
+        puts "You can only go negative up to -$10"
+        return balance
+      end
+
+      @check_withdrawals += 1
+      @balance -= withdrawal_amount
+
+      check_fee
+    end
+
+    def check_fee
+      if @check_withdrawals > 3
+        @balance -= 2
+      else
+        @balance
+      end
+    end
+
+    def reset_checks
+      @check_withdrawals = 0
+    end
+
+  end
+
+end

lib/money_market_account.rb

@@ -0,0 +1,60 @@
+require_relative 'account'
+
+module Bank
+
+  class MoneyMarketAccount < Bank::Account
+
+    include Interest
+
+    attr_reader :total_transactions
+
+    def initialize(id, balance, opendate = nil)
+      super
+      @total_transactions = 0
+    end
+
+    def set_balance(start_balance)
+      if start_balance < 10000
+        argument("You cannot initialize a new Money Market account with less than 10k.")
+      else
+        start_balance
+      end
+    end
+
+    def withdraw(withdrawal_amount)
+      @total_transactions += 1
+
+      if @total_transactions > 6
+        argument("You cannot make more than six transactions per month.")
+      end
+
+      if @balance < 10000
+        argument("You cannot make another withdrawal until you make a deposit")
+      end
+
+      super
+
+      if @balance < 10000
+        @balance -= 100
+      end
+    end
+
+    def deposit(deposit_amount)
+      if @balance > 10000
+        @total_transactions += 1
+      end
+
+      if @total_transactions > 6
+        argument("You cannot make more than six transactions per month.")
+      end
+
+      super
+    end
+
+    def reset_transactions
+      @total_transactions = 0
+    end
+
+  end
+
+end

lib/savings_account.rb

@@ -0,0 +1,35 @@
+module Bank
+
+  class SavingsAccount < Account
+
+    include Interest
+
+    def initialize(id, balance, opendate = nil)
+      super
+
+      if @balance < 10
+        argument("You must initially deposit at least $10.00")
+      end
+    end
+
+    def withdraw(withdrawal_amount)
+      original_balance = balance
+      super
+
+      if balance == original_balance
+        balance
+      elsif balance - 2 <= 10
+        puts "This withdrawal and fee will take your balance below $10."
+        @balance = original_balance
+      else
+        withdrawal_fee
+      end
+    end
+
+    def withdrawal_fee
+      @balance -= 2
+    end
+
+  end
+
+end