module Bbq
  class TestUser
    module Within
      METHODS_USING_WITHIN = [
        :see?, :not_see?,
        :attach_file, :check, :choose, :click_link_or_button, :click_button,
        :click_link, :click_on, :fill_in, :select, :uncheck, :unselect
      ]

      METHODS_USING_WITHIN.each do |method_name|
        class_eval <<-RUBY
          def #{method_name}(*args)
            using_within(args) { super }
          end
        RUBY
      end

      def using_within(args)
        options = args.extract_options!
        locator = options.delete(:within)
        args.push(options) unless options.empty?

        if locator
          within(locator) { yield }
        else
          yield
        end
      end
    end
  end
end