test/transition_test.rb in simplificator-fsm-0.2.1 vs test/transition_test.rb in simplificator-fsm-0.2.2

@@ -1,38 +1,44 @@
 require 'test_helper'
 
 class TransitionTest < Test::Unit::TestCase
   context 'Initializer' do
+    
+    
     should 'require name, from and to' do
+      bli_state = FSM::State.new('bli')
+      blo_state = FSM::State.new('blo')
       assert_raise(ArgumentError) do
         FSM::Transition.new(nil, nil, nil)
       end
       assert_raise(ArgumentError) do
-        FSM::Transition.new('bla', nil, nil)
+        FSM::Transition.new(:name, nil, nil)
       end
       assert_raise(ArgumentError) do
-        FSM::Transition.new(nil, 'bli', nil)
+        FSM::Transition.new(nil, bli_state, nil)
       end
       assert_raise(ArgumentError) do
-        FSM::Transition.new(nil, nil, 'blo')
+        FSM::Transition.new(nil, nil, blo_state)
       end
       assert_raise(ArgumentError) do
-        FSM::Transition.new('bli', 'bli', nil)
+        FSM::Transition.new(:name, bli_state, nil)
       end
       assert_raise(ArgumentError) do
-        FSM::Transition.new(nil, 'blo', 'blo')
+        FSM::Transition.new(nil, blo_state, blo_state)
       end
       assert_raise(ArgumentError) do
-        FSM::Transition.new('blo', nil, 'bli')
+        FSM::Transition.new(:name, nil, bli_state)
       end
       
-      FSM::Transition.new('bla', 'bli', 'blo')
+      FSM::Transition.new(:name, bli_state, blo_state)
     end
     
     should 'allow only valid options' do
+      bli_state = FSM::State.new('bli')
+      blo_state = FSM::State.new('blo')
       assert_raise(ArgumentError) do
-        FSM::Transition.new('bla', 'bli', 'blo', :foo => 12)
+        FSM::Transition.new(:name, bli_state, blo_state, :foo => 12)
       end
-      FSM::Transition.new('bla', 'bli', 'blo', :event => :some)
+      FSM::Transition.new(:name, bli_state, blo_state, :event => :some)
     end
   end
 end