298 lines
8.6 KiB
Ruby
298 lines
8.6 KiB
Ruby
require 'rubygems'
|
|
require 'active_record'
|
|
require 'logger'
|
|
|
|
load_schema
|
|
|
|
ActiveRecord::Base.logger = Logger.new(STDERR)
|
|
|
|
class Gate < ActiveRecord::Base
|
|
include AASM
|
|
|
|
# Fake this column for testing purposes
|
|
attr_accessor :aasm_state
|
|
|
|
aasm do
|
|
state :opened
|
|
state :closed
|
|
|
|
event :view do
|
|
transitions :to => :read, :from => [:needs_attention]
|
|
end
|
|
end
|
|
end
|
|
|
|
class Reader < ActiveRecord::Base
|
|
def aasm_read_state
|
|
"fi"
|
|
end
|
|
include AASM
|
|
end
|
|
|
|
class Writer < ActiveRecord::Base
|
|
def aasm_write_state(state)
|
|
"fo"
|
|
end
|
|
include AASM
|
|
end
|
|
|
|
class Transient < ActiveRecord::Base
|
|
def aasm_write_state_without_persistence(state)
|
|
"fum"
|
|
end
|
|
include AASM
|
|
end
|
|
|
|
class Simple < ActiveRecord::Base
|
|
include AASM
|
|
aasm_column :status
|
|
end
|
|
|
|
class SimpleNewDsl < ActiveRecord::Base
|
|
include AASM
|
|
aasm :column => :status
|
|
end
|
|
|
|
class Derivate < Simple
|
|
end
|
|
|
|
class DerivateNewDsl < SimpleNewDsl
|
|
end
|
|
|
|
class Thief < ActiveRecord::Base
|
|
set_table_name "thieves"
|
|
include AASM
|
|
aasm_initial_state Proc.new { |thief| thief.skilled ? :rich : :jailed }
|
|
aasm_state :rich
|
|
aasm_state :jailed
|
|
attr_accessor :skilled, :aasm_state
|
|
end
|
|
|
|
shared_examples_for "aasm model" do
|
|
it "should include AASM::Persistence::ActiveRecordPersistence" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence)
|
|
end
|
|
it "should include AASM::Persistence::ActiveRecordPersistence::InstanceMethods" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence::InstanceMethods)
|
|
end
|
|
end
|
|
|
|
describe Gate, "class methods" do
|
|
before(:each) do
|
|
@klass = Gate
|
|
end
|
|
it_should_behave_like "aasm model"
|
|
it "should include AASM::Persistence::ActiveRecordPersistence::ReadState" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence::ReadState)
|
|
end
|
|
it "should include AASM::Persistence::ActiveRecordPersistence::WriteState" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence::WriteState)
|
|
end
|
|
it "should include AASM::Persistence::ActiveRecordPersistence::WriteStateWithoutPersistence" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence::WriteStateWithoutPersistence)
|
|
end
|
|
end
|
|
|
|
describe Reader, "class methods" do
|
|
before(:each) do
|
|
@klass = Reader
|
|
end
|
|
it_should_behave_like "aasm model"
|
|
it "should not include AASM::Persistence::ActiveRecordPersistence::ReadState" do
|
|
@klass.included_modules.should_not be_include(AASM::Persistence::ActiveRecordPersistence::ReadState)
|
|
end
|
|
it "should include AASM::Persistence::ActiveRecordPersistence::WriteState" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence::WriteState)
|
|
end
|
|
it "should include AASM::Persistence::ActiveRecordPersistence::WriteStateWithoutPersistence" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence::WriteStateWithoutPersistence)
|
|
end
|
|
end
|
|
|
|
describe Writer, "class methods" do
|
|
before(:each) do
|
|
@klass = Writer
|
|
end
|
|
it_should_behave_like "aasm model"
|
|
it "should include AASM::Persistence::ActiveRecordPersistence::ReadState" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence::ReadState)
|
|
end
|
|
it "should not include AASM::Persistence::ActiveRecordPersistence::WriteState" do
|
|
@klass.included_modules.should_not be_include(AASM::Persistence::ActiveRecordPersistence::WriteState)
|
|
end
|
|
it "should include AASM::Persistence::ActiveRecordPersistence::WriteStateWithoutPersistence" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence::WriteStateWithoutPersistence)
|
|
end
|
|
end
|
|
|
|
describe Transient, "class methods" do
|
|
before(:each) do
|
|
@klass = Transient
|
|
end
|
|
it_should_behave_like "aasm model"
|
|
it "should include AASM::Persistence::ActiveRecordPersistence::ReadState" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence::ReadState)
|
|
end
|
|
it "should include AASM::Persistence::ActiveRecordPersistence::WriteState" do
|
|
@klass.included_modules.should be_include(AASM::Persistence::ActiveRecordPersistence::WriteState)
|
|
end
|
|
it "should not include AASM::Persistence::ActiveRecordPersistence::WriteStateWithoutPersistence" do
|
|
@klass.included_modules.should_not be_include(AASM::Persistence::ActiveRecordPersistence::WriteStateWithoutPersistence)
|
|
end
|
|
end
|
|
|
|
describe Gate, "instance methods" do
|
|
|
|
before do
|
|
@gate = Gate.new
|
|
end
|
|
|
|
it "should respond to aasm read state when not previously defined" do
|
|
@gate.should respond_to(:aasm_read_state)
|
|
end
|
|
|
|
it "should respond to aasm write state when not previously defined" do
|
|
@gate.should respond_to(:aasm_write_state)
|
|
end
|
|
|
|
it "should respond to aasm write state without persistence when not previously defined" do
|
|
@gate.should respond_to(:aasm_write_state_without_persistence)
|
|
end
|
|
|
|
it "should return the initial state when new and the aasm field is nil" do
|
|
@gate.aasm_current_state.should == :opened
|
|
end
|
|
|
|
it "should return the aasm column when new and the aasm field is not nil" do
|
|
@gate.aasm_state = "closed"
|
|
@gate.aasm_current_state.should == :closed
|
|
end
|
|
|
|
it "should return the aasm column when not new and the aasm_column is not nil" do
|
|
@gate.stub!(:new_record?).and_return(false)
|
|
@gate.aasm_state = "state"
|
|
@gate.aasm_current_state.should == :state
|
|
end
|
|
|
|
it "should allow a nil state" do
|
|
@gate.stub!(:new_record?).and_return(false)
|
|
@gate.aasm_state = nil
|
|
@gate.aasm_current_state.should be_nil
|
|
end
|
|
|
|
it "should have aasm_ensure_initial_state" do
|
|
@gate.send :aasm_ensure_initial_state
|
|
end
|
|
|
|
it "should call aasm_ensure_initial_state on validation before create" do
|
|
@gate.should_receive(:aasm_ensure_initial_state).and_return(true)
|
|
@gate.valid?
|
|
end
|
|
|
|
it "should call aasm_ensure_initial_state on validation before create" do
|
|
@gate.stub!(:new_record?).and_return(false)
|
|
@gate.should_not_receive(:aasm_ensure_initial_state)
|
|
@gate.valid?
|
|
end
|
|
|
|
end
|
|
|
|
describe 'Derivates' do
|
|
it "should have the same states as its parent" do
|
|
Derivate.aasm_states.should == Simple.aasm_states
|
|
end
|
|
|
|
it "should have the same events as its parent" do
|
|
Derivate.aasm_events.should == Simple.aasm_events
|
|
end
|
|
|
|
it "should have the same column as its parent" do
|
|
Derivate.aasm_column.should == :status
|
|
end
|
|
|
|
it "should have the same column as its parent even for the new dsl" do
|
|
SimpleNewDsl.aasm_column.should == :status
|
|
DerivateNewDsl.aasm_column.should == :status
|
|
end
|
|
end
|
|
|
|
describe AASM::Persistence::ActiveRecordPersistence::NamedScopeMethods do
|
|
|
|
context "Does not already respond_to? the scope name" do
|
|
it "should add a scope" do
|
|
Simple.should_not respond_to(:unknown_scope)
|
|
Simple.aasm_state :unknown_scope
|
|
Simple.should respond_to(:unknown_scope)
|
|
Simple.unknown_scope.class.should == ActiveRecord::Relation
|
|
end
|
|
end
|
|
|
|
context "Already respond_to? the scope name" do
|
|
it "should not add a scope" do
|
|
Simple.aasm_state :new
|
|
Simple.should respond_to(:new)
|
|
Simple.new.class.should == Simple
|
|
end
|
|
end
|
|
end
|
|
|
|
describe 'Thieves' do
|
|
|
|
it 'should be rich if they\'re skilled' do
|
|
Thief.new(:skilled => true).aasm_current_state.should == :rich
|
|
end
|
|
|
|
it 'should be jailed if they\'re unskilled' do
|
|
Thief.new(:skilled => false).aasm_current_state.should == :jailed
|
|
end
|
|
end
|
|
|
|
describe 'transitions with persistence' do
|
|
|
|
it 'should not store states for invalid models' do
|
|
validator = Validator.create(:name => 'name')
|
|
validator.should be_valid
|
|
validator.should be_sleeping
|
|
|
|
validator.name = nil
|
|
validator.should_not be_valid
|
|
validator.run!.should be_false
|
|
validator.should be_sleeping
|
|
|
|
validator.reload
|
|
validator.should_not be_running
|
|
validator.should be_sleeping
|
|
|
|
validator.name = 'another name'
|
|
validator.should be_valid
|
|
validator.run!.should be_true
|
|
validator.should be_running
|
|
|
|
validator.reload
|
|
validator.should be_running
|
|
validator.should_not be_sleeping
|
|
end
|
|
|
|
it 'should store states for invalid models if configured' do
|
|
persistor = InvalidPersistor.create(:name => 'name')
|
|
persistor.should be_valid
|
|
persistor.should be_sleeping
|
|
|
|
persistor.name = nil
|
|
persistor.should_not be_valid
|
|
persistor.run!.should be_true
|
|
persistor.should be_running
|
|
|
|
persistor = InvalidPersistor.find(persistor.id)
|
|
persistor.valid?
|
|
persistor.should be_valid
|
|
persistor.should be_running
|
|
persistor.should_not be_sleeping
|
|
|
|
persistor.reload
|
|
persistor.should be_running
|
|
persistor.should_not be_sleeping
|
|
end
|
|
|
|
end
|