Class: TestSuiteSplitter::RspecHelper

Inherits:
Object
  • Object
show all
Defined in:
lib/test_suite_splitter/rspec_helper.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(groups:, group_number:, exclude_types: nil, only_types: nil, tags: nil) ⇒ RspecHelper

Returns a new instance of RspecHelper.



4
5
6
7
8
9
10
11
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 4

def initialize(groups:, group_number:, exclude_types: nil, only_types: nil, tags: nil)
  @exclude_types = exclude_types
  @groups = groups
  @group_number = group_number
  @example_data_exists = File.exist?("spec/examples.txt")
  @only_types = only_types
  @tags = tags
end

Instance Attribute Details

#exclude_types ⇒ Object (readonly)

Returns the value of attribute exclude_types.



2
3
4
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 2

def exclude_types
  @exclude_types
end

#only_types ⇒ Object (readonly)

Returns the value of attribute only_types.



2
3
4
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 2

def only_types
  @only_types
end

#tags ⇒ Object (readonly)

Returns the value of attribute tags.



2
3
4
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 2

def tags
  @tags
end

Instance Method Details

#example_data ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 17

def example_data
  @example_data ||= begin
    raw_data = File.read("spec/examples.txt")

    result = []
    raw_data.scan(/^\.\/(.+)\[(.+?)\]\s+\|\s+(.+?)\s+\|\s+((.+?) seconds|)\s+\|$/) do |match|
      file_path = match[0]
      spec_result = match[1]
      seconds = match[4]&.to_f

      spec_data = {
        file_path: file_path,
        spec_result: spec_result,
        seconds: seconds
      }

      result << spec_data
    end

    result
  end
end

#example_data_exists? ⇒ Boolean

Returns:

  • (Boolean) 


13
14
15
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 13

def example_data_exists?
  @example_data_exists
end

#example_file(path) ⇒ Object 



56
57
58
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 56

def example_file(path)
  example_files[path]
end

#example_files ⇒ Object 



40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 40

def example_files
  @example_files ||= begin
    files = {}
    example_data.each do |spec_data|
      file_path = spec_data.fetch(:file_path)
      seconds = spec_data.fetch(:seconds)

      files[file_path] ||= {examples: 0, seconds: 0.0}
      files[file_path][:examples] += 1
      files[file_path][:seconds] += seconds if seconds
    end

    files
  end
end

#group_files ⇒ Object 



60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 60

def group_files
  return @group_files if @group_files

  sorted_files.each do |file|
    file_path = file.fetch(:path)
    file_data = example_file(file_path) if example_data_exists?

    if file_data
      examples = file_data.fetch(:examples)
      seconds = file_data.fetch(:seconds)
    else
      examples = file.fetch(:examples)
    end

    group = group_with_least
    group[:examples] += examples
    group[:files] << file
    group[:seconds] += seconds if seconds
  end

  @group_files = group_orders[@group_number - 1].fetch(:files)
end

#group_orders ⇒ Object 



83
84
85
86
87
88
89
90
91
92
93
94
95
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 83

def group_orders
  @group_orders ||= begin
    group_orders = []
    @groups.times do
      group_orders << {
        examples: 0,
        files: [],
        seconds: 0.0
      }
    end
    group_orders
  end
end

#group_with_least ⇒ Object 



97
98
99
100
101
102
103
104
105
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 97

def group_with_least
  group_orders.min do |group1, group2|
    if example_data_exists? && group1.fetch(:seconds) != 0.0 && group2.fetch(:seconds) != 0.0
      group1.fetch(:seconds) <=> group2.fetch(:seconds)
    else
      group1.fetch(:examples) <=> group2.fetch(:examples)
    end
  end
end

#sorted_files ⇒ Object

Sort them so that they are sorted by file path in three groups so each group have an equal amount of controller specs, features specs and so on



108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
 
# File 'lib/test_suite_splitter/rspec_helper.rb', line 108

def sorted_files
  files.values.sort do |file1, file2|
    file1_path = file1.fetch(:path)
    file2_path = file2.fetch(:path)

    file1_data = example_file(file1_path) if example_data_exists?
    file2_data = example_file(file2_path) if example_data_exists?

    if file1_data && file2_data && file1_data.fetch(:seconds) != 0.0 && file2_data.fetch(:seconds) != 0.0
      value1 = file1_data[:seconds]
    else
      value1 = file1.fetch(:points)
    end

    if file2_data && file1_data && file2_data.fetch(:seconds) != 0.0 && file2_data.fetch(:seconds) != 0.0
      value2 = file2_data[:seconds]
    else
      value2 = file2.fetch(:points)
    end

    if value1 == value2
      value2 = file1_path
      value1 = file2_path
    end

    value2 <=> value1
  end
end