mirror of
https://github.com/aasm/aasm
synced 2023-03-27 23:22:41 -04:00
27 lines
775 B
Ruby
27 lines
775 B
Ruby
require 'spec_helper'
|
|
|
|
class Banker
|
|
include AASM
|
|
aasm do
|
|
state :retired
|
|
state :selling_bad_mortgages
|
|
initial_state Proc.new { |banker| banker.rich? ? :retired : :selling_bad_mortgages }
|
|
end
|
|
RICH = 1_000_000
|
|
attr_accessor :balance
|
|
def initialize(balance = 0); self.balance = balance; end
|
|
def rich?; self.balance >= RICH; end
|
|
end
|
|
|
|
describe 'initial states' do
|
|
let(:bar) {Bar.new}
|
|
|
|
it 'should use the first state defined if no initial state is given' do
|
|
expect(bar.aasm.current_state).to eq(:read)
|
|
end
|
|
|
|
it 'should determine initial state from the Proc results' do
|
|
expect(Banker.new(Banker::RICH - 1).aasm.current_state).to eq(:selling_bad_mortgages)
|
|
expect(Banker.new(Banker::RICH + 1).aasm.current_state).to eq(:retired)
|
|
end
|
|
end
|