lib/state_machina/definitions.rb in state_machina-0.1.0 vs lib/state_machina/definitions.rb in state_machina-0.1.1

- old
+ new

@@ -14,7 +14,13 @@ yield(StateMachina::Registry.find_machine!(model_name, machine_name)) end alias_method :extend_state_machine, :extend_machine + + def find_machine(machine_name = 'default') + model_name = StateMachina::Util.normalized_klass_to_name(self) + + StateMachina::Registry.find_machine!(model_name, machine_name) + end end end