spec_helpers.rb 741 Bytes
Newer Older
1 2 3 4 5 6 7 8
module RuboCop
  module SpecHelpers
    SPEC_HELPERS = %w[spec_helper.rb rails_helper.rb].freeze

    # Returns true if the given node originated from the spec directory.
    def in_spec?(node)
      path = node.location.expression.source_buffer.name

9 10 11 12 13 14 15 16 17 18 19 20
      !SPEC_HELPERS.include?(File.basename(path)) &&
        path.start_with?(File.join(Dir.pwd, 'spec'), File.join(Dir.pwd, 'ee', 'spec'))
    end

    # Returns true if the given node originated from a migration spec.
    def in_migration_spec?(node)
      path = node.location.expression.source_buffer.name

      in_spec?(node) &&
        path.start_with?(
          File.join(Dir.pwd, 'spec', 'migrations'),
          File.join(Dir.pwd, 'ee', 'spec', 'migrations'))
21 22 23
    end
  end
end