class Cucumber::Runtime::StepHooks

def after_step_hooks(test_step)
  @hooks.map do |hook|
    action = ->(*args) { hook.invoke('AfterStep', [args, test_step]) }
    hook_step = Hooks.after_step_hook(@id_generator.new_id, test_step, hook.location, &action)
    @event_bus.hook_test_step_created(hook_step, hook)
    hook_step
  end
end

def apply(test_steps)
  test_steps.flat_map do |test_step|
    [test_step] + after_step_hooks(test_step)
  end
end

def initialize(id_generator, hooks, event_bus)
  @hooks = hooks
  @id_generator = id_generator
  @event_bus = event_bus
end