cucumber

account.rb

require File.join(File.dirname(__FILE__), '../../lib/account')

Given(/^I have \$(\d+) in my account$/) do |starting_balance|
  @account = Account.new(starting_balance)
end

When(/^I request my account balance$/) do
  @retrieved_balance = @account.balance
end

When(/^I deposit \$(\d+)$/) do |deposit_amount|
  @account.deposit(deposit_amount)
end

Then(/^a balance of \$(\d+) is reported$/) do |expected_balance|
  assert_equal expected_balance.to_f, @retrieved_balance
end

When(/^I withdraw \$(\d+)$/) do |withdrawal_amount|
  @withdrawal_result = @account.withdraw(withdrawal_amount)
end

Then(/^an error message appears$/) do
  assert_equal "OVERDRAWN", @withdrawal_result
end

account_steps.rb

require File.join(File.dirname(__FILE__), '../../lib/account')

Given(/^I have \$(\d+) in my account$/) do |starting_balance|
  @account = Account.new(starting_balance)
end

When(/^I request my account balance$/) do
  @retrieved_balance = @account.balance
end

When(/^I deposit \$(\d+)$/) do |deposit_amount|
  @account.deposit(deposit_amount)
end

Then(/^a balance of \$(\d+) is reported$/) do |expected_balance|
  assert_equal expected_balance.to_f, @retrieved_balance
end

When(/^I withdraw \$(\d+)$/) do |withdrawal_amount|
  @withdrawal_result = @account.withdraw(withdrawal_amount)
end

Then(/^an error message appears$/) do
  assert_equal "OVERDRAWN", @withdrawal_result
end