# frozen_string_literal: true

require 'test_helper'

class StateCollectionWithStateBehaviorsTest < StateMachinesTest
  def setup
    @machine = StateMachines::Machine.new(Class.new)
    @states = StateMachines::StateCollection.new(@machine)

    @states << @parked = StateMachines::State.new(@machine, :parked)
    @states << @idling = StateMachines::State.new(@machine, :idling)
    @machine.states.concat(@states)

    @idling.context do
      def speed
        0
      end
    end
  end

  def test_should_order_states_after_initial_state
    @parked.initial = true

    assert_equal [@parked, @idling], @states.by_priority
  end

  def test_should_order_states_after_transition_states
    @machine.event :ignite do
      transition from: :parked
    end

    assert_equal [@parked, @idling], @states.by_priority
  end

  def test_should_order_states_before_other_states
    assert_equal [@idling, @parked], @states.by_priority
  end

  def test_should_order_state_before_callback_states
    @machine.before_transition from: :parked, do: -> {}

    assert_equal [@idling, @parked], @states.by_priority
  end
end
