class Cucumber::Runtime::AfterHooks

def after_hooks
  @hooks.map do |hook|
    action = ->(result) { hook.invoke('After', @scenario.with_result(result)) }
    Hooks.after_hook(hook.location, &action)
  end
end

def apply_to(test_case)
  test_case.with_steps(
    test_case.test_steps + after_hooks.reverse
  )
end

def initialize(hooks, scenario)
  @hooks = hooks
  @scenario = scenario
end