lib/knapsack_pro/tracker.rb in knapsack_pro-0.46.0 vs lib/knapsack_pro/tracker.rb in knapsack_pro-0.47.0

@@ -1,13 +1,14 @@
 module KnapsackPro
   class Tracker
     include Singleton
 
-    attr_reader :global_time, :test_files_with_time
+    attr_reader :global_time_since_beginning, :global_time, :test_files_with_time
     attr_writer :current_test_path
 
     def initialize
+      @global_time_since_beginning = 0
       set_defaults
     end
 
     def reset!
       set_defaults
@@ -16,14 +17,14 @@
     def start_timer
       @start_time = now_without_mock_time.to_f
     end
 
     def stop_timer
-      @execution_time = @start_time ? now_without_mock_time.to_f - @start_time : 0.0
-      update_global_time
-      update_test_file_time
-      @execution_time
+      execution_time = @start_time ? now_without_mock_time.to_f - @start_time : 0.0
+      update_global_time(execution_time)
+      update_test_file_time(execution_time)
+      execution_time
     end
 
     def current_test_path
       raise("current_test_path needs to be set by Knapsack Pro Adapter's bind method") unless @current_test_path
       @current_test_path.sub(/^\.\//, '')
@@ -46,16 +47,17 @@
       @global_time = 0
       @test_files_with_time = {}
       @test_path = nil
     end
 
-    def update_global_time
-      @global_time += @execution_time
+    def update_global_time(execution_time)
+      @global_time += execution_time
+      @global_time_since_beginning += execution_time
     end
 
-    def update_test_file_time
+    def update_test_file_time(execution_time)
       @test_files_with_time[current_test_path] ||= 0
-      @test_files_with_time[current_test_path] += @execution_time
+      @test_files_with_time[current_test_path] += execution_time
     end
 
     def now_without_mock_time
       if defined?(Timecop)
         Time.now_without_mock_time