require 'erb'

require 'capybara_screenshot_idobata'
require 'capybara_screenshot_idobata/dsl'

RSpec.configure do |config|
  config.include CapybaraScreenshotIdobata::DSL, type: :feature
end

CapybaraScreenshotIdobata.configure do |config|
  config.message_formatter = proc {|file, line, context|
    example = context.respond_to?(:example) ? context.example : RSpec.current_example # For RSpec 2/3 conpatiblity

<<-MESSAGE
Screenshot from: <code>#{ERB::Util.h(file)}:#{line}</code>
<br>
<b>#{ERB::Util.h(example.full_description)}</b>
MESSAGE
  }
end

require 'erb'

require 'capybara_screenshot_idobata'
require 'capybara_screenshot_idobata/dsl'

RSpec.configure do |config|
  config.include CapybaraScreenshotIdobata::DSL, type: :feature
end

CapybaraScreenshotIdobata.configure do |config|
  config.message_formatter = proc {|file, line, context|
<<-MESSAGE
Screenshot from: <code>#{ERB::Util.h(file)}:#{line}</code>
<br>
<b>#{ERB::Util.h(RSpec.current_example.full_description)}</b>
MESSAGE
  }
end

Remove the code for Rspec2 conpatibility
Remove the code for Rspec2 conpatibility
