spree/spree

module Spree
  module Shipments
    class ChangeState
      prepend Spree::ServiceModule::Base

      def call(shipment:, state:)
        shipment.send("#{state}!")
        success(shipment.reload)
      rescue ActiveRecord::Rollback, ActiveRecord::RecordInvalid, StateMachines::InvalidTransition
        failure(shipment)
      end
    end
  end
end