class Cucumber::Core::Test::Mapper::HookFactory

def after(block)
  build_hook_step(block, Hooks::AfterHook, Test::UnskippableAction)
end

def after_step(block)
  build_hook_step(block, Hooks::AfterStepHook, Test::Action)
end

def before(block)
  build_hook_step(block, Hooks::BeforeHook, Test::UnskippableAction)
end

def before_step(block)
  build_hook_step(block, Hooks::BeforeStepHook, Test::UnskippableAction)
end

def build_hook_step(block, hook_type, mapping_type)
  mapping = mapping_type.new(&block)
  hook = hook_type.new(mapping.location)
  Step.new(source + [hook], mapping)
end