aboutsummaryrefslogtreecommitdiffstats
path: root/railties/lib/rails/test_unit/line_filtering.rb
blob: 65f76f21a6e11201eed144d43e800e5dd9ba50a3 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
require 'method_source'

module Rails
  module LineFiltering # :nodoc:
    def run(reporter, options = {})
      if options[:patterns] && options[:patterns].any? { |p| p =~ /:\d+/ }
        options[:filter] = \
          CompositeFilter.new(self, options[:filter], options[:patterns])
      end

      super
    end
  end

  class CompositeFilter # :nodoc:
    def initialize(runnable, filter, patterns)
      @runnable = runnable
      @filters = [ derive_regexp(filter), *derive_line_filters(patterns) ].compact
    end

    def ===(method)
      @filters.any? { |filter| filter === method }
    end

    private
      def derive_regexp(filter)
        filter =~ %r%/(.*)/% ? Regexp.new($1) : filter
      end

      def derive_line_filters(patterns)
        patterns.map do |file_and_line|
          file, line = file_and_line.split(':')
          Filter.new(@runnable, file, line) if file
        end
      end
  end

  class Filter # :nodoc:
    def initialize(runnable, file, line)
      @runnable, @file = runnable, File.expand_path(file)
      @line = line.to_i if line
    end

    def ===(method)
      return unless @runnable.method_defined?(method)

      if @line
        test_file, test_range = definition_for(@runnable.instance_method(method))
        test_file == @file && test_range.include?(@line)
      else
        @runnable.instance_method(method).source_location.first == @file
      end
    end

    private
      def definition_for(method)
        file, start_line = method.source_location
        end_line = method.source.count("\n") + start_line - 1

        return file, start_line..end_line
      end
  end
end