split_spec.rb in rspeed-0.6.0

- old
+ new

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-RSpec.describe RSpeed::Splitter, '.split' do
-  subject(:splitter) { described_class.new }
+require 'support/env_mock'
 
+RSpec.describe RSpeed::Splitter, '.split' do
   let!(:unsorted_data) do
     [
       { file: './spec/1_5_spec.rb', time: '1.5' },
       { file: './spec/1_1_spec.rb', time: '1.1' },
       { file: './spec/0_7_spec.rb', time: '0.7' },
@@ -13,50 +13,12 @@
       { file: './spec/0_2_spec.rb', time: '0.2' },
       { file: './spec/2_0_spec.rb', time: '2.0' },
     ]
   end
 
-  before { allow(RSpeed::Env).to receive(:pipes).and_return 3 }
-
-  context 'when diff is given' do
-    it 'splits the times between the pipes' do
-      expect(splitter.split(unsorted_data)).to eq(
-        rspeed_1: {
-          files: [{ file: './spec/2_0_spec.rb', time: 2.0 }],
-          number: 1,
-          total: 2.0,
-        },
-
-        rspeed_2: {
-          files: [
-            { file: './spec/1_5_spec.rb', time: 1.5 },
-            { file: './spec/0_4_spec.rb', time: 0.4 },
-            { file: './spec/0_2_spec.rb', time: 0.2 },
-          ],
-
-          number: 2,
-          total: 1.5 + 0.4 + 0.2, # 1.5 -> 1.9 -> 2.1
-        },
-
-        rspeed_3: {
-          files: [
-            { file: './spec/1_1_spec.rb', time: 1.1 },
-            { file: './spec/0_7_spec.rb', time: 0.7 },
-            { file: './spec/0_3_spec.rb', time: 0.3 },
-          ],
-
-          number: 3,
-          total: 1.1 + 0.7 + 0.3, # 1.1 -> 1.8 -> 2.1
-        }
-      )
-    end
-  end
-
-  context 'when diff is not given' do
-    before { allow(splitter).to receive(:diff).and_return(unsorted_data) }
-
-    it 'splits the diff result into times between the pipes' do
-      expect(splitter.split).to eq(
+  it 'splits the times between the pipes' do
+    EnvMock.mock(rspeed_pipes: 3) do
+      expect(described_class.split(data: unsorted_data)).to eq(
         rspeed_1: {
           files: [{ file: './spec/2_0_spec.rb', time: 2.0 }],
           number: 1,
           total: 2.0,
         },