Move state transition test to separate file