Sha256: 27cdafbdbec9c82f75620469ffdb8bff8d43d39a6f90f7c336c7e4a76727febe

Contents?: true

Size: 1.81 KB

Versions: 10

Compression:

Stored size: 1.81 KB

Contents

module ParallelTests
  class Grouper
    class << self
      def by_steps(tests, num_groups, options)
        features_with_steps = build_features_with_steps(tests, options)
        in_even_groups_by_size(features_with_steps, num_groups)
      end

      def in_even_groups_by_size(items_with_sizes, num_groups, options = {})
        groups = Array.new(num_groups) { {:items => [], :size => 0} }

        # add all files that should run in a single process to one group
        (options[:single_process] || []).each do |pattern|
          matched, items_with_sizes = items_with_sizes.partition { |item, size| item =~ pattern }
          matched.each { |item, size| add_to_group(groups.first, item, size) }
        end

        groups_to_fill = (options[:isolate] ? groups[1..-1] : groups)

        # add all other files
        largest_first(items_with_sizes).each do |item, size|
          smallest = smallest_group(groups_to_fill)
          add_to_group(smallest, item, size)
        end

        groups.map!{|g| g[:items].sort }
      end

      private

      def largest_first(files)
        files.sort_by{|item, size| size }.reverse
      end

      def smallest_group(groups)
        groups.min_by{|g| g[:size] }
      end

      def add_to_group(group, item, size)
        group[:items] << item
        group[:size] += size
      end

      def build_features_with_steps(tests, options)
        require 'parallel_tests/gherkin/listener'
        listener = ParallelTests::Gherkin::Listener.new
        listener.ignore_tag_pattern = Regexp.compile(options[:ignore_tag_pattern]) if options[:ignore_tag_pattern]
        parser = ::Gherkin::Parser::Parser.new(listener, true, 'root')
        tests.each{|file|
          parser.parse(File.read(file), file, 0)
        }
        listener.collect.sort_by{|_,value| -value }
      end
    end
  end
end

Version data entries

10 entries across 10 versions & 1 rubygems

Version Path
parallel_tests-0.16.3 lib/parallel_tests/grouper.rb
parallel_tests-0.16.2 lib/parallel_tests/grouper.rb
parallel_tests-0.16.1 lib/parallel_tests/grouper.rb
parallel_tests-0.16.0 lib/parallel_tests/grouper.rb
parallel_tests-0.15.4 lib/parallel_tests/grouper.rb
parallel_tests-0.15.3 lib/parallel_tests/grouper.rb
parallel_tests-0.15.2 lib/parallel_tests/grouper.rb
parallel_tests-0.15.1 lib/parallel_tests/grouper.rb
parallel_tests-0.15.0 lib/parallel_tests/grouper.rb
parallel_tests-0.14.0 lib/parallel_tests/grouper.rb