50 lines
1.3 KiB
Ruby
50 lines
1.3 KiB
Ruby
require 'cucumber/formatter/io'
|
|
require 'cucumber/formatter/ansicolor'
|
|
|
|
module Cucumber
|
|
class StepWriter
|
|
include Cucumber::Formatter::Io
|
|
include Cucumber::Formatter::ANSIColor
|
|
|
|
class << self
|
|
def after_write(&block)
|
|
if block
|
|
@after_write = block
|
|
else
|
|
@after_write
|
|
end
|
|
end
|
|
end
|
|
|
|
def initialize(step_mother, path_or_io, options)
|
|
@step_mother, @io = step_mother, ensure_dir(path_or_io, 'step writer')
|
|
end
|
|
|
|
def after_features(features)
|
|
undefined = @step_mother.steps(:undefined)
|
|
return if undefined.empty?
|
|
|
|
undefined.each do |step|
|
|
step_name = Cucumber::Undefined === step.exception ? step.exception.step_name : step.name
|
|
step_multiline_class = step.multiline_arg ? step.multiline_arg.class : nil
|
|
|
|
path = Pathname(@io).join(step.actual_keyword.downcase.strip, step_name.downcase.gsub(%r{[^\w]+}, '_').gsub(%r{^_|_$}, '') + '.rb')
|
|
|
|
if !path.file?
|
|
puts yellow("Writing new step to #{path}.")
|
|
|
|
path.parent.mkpath
|
|
path.open('wb') { |fh| fh.print @step_mother.snippet_text(step.actual_keyword, step_name, step_multiline_class) }
|
|
end
|
|
end
|
|
|
|
after_write.call(@io) if after_write
|
|
end
|
|
|
|
def after_write
|
|
self.class.after_write
|
|
end
|
|
end
|
|
end
|
|
|