refactor cycle analytics - updated based on MR feedback

1 files changed, 5 insertions, 7 deletions

diff --git a/lib/gitlab/cycle_analytics/test_stage.rb b/lib/gitlab/cycle_analytics/test_stage.rb
index 3ab93bebd87..58f72bb405e 100644
--- a/lib/gitlab/cycle_analytics/test_stage.rb
+++ b/lib/gitlab/cycle_analytics/test_stage.rb
@@ -1,16 +1,14 @@
 module Gitlab
   module CycleAnalytics
     class TestStage < BaseStage
-      def initialize(*args)
-        super(*args)
-
-        @description = "Total test time for all commits/merges"
+      def description
+        "Total test time for all commits/merges"
       end
 
       def median
-        @fetcher.calculate_metric(:test,
-                                  MergeRequest::Metrics.arel_table[:latest_build_started_at],
-                                  MergeRequest::Metrics.arel_table[:latest_build_finished_at])
+        @fetcher.median(:test,
+                        MergeRequest::Metrics.arel_table[:latest_build_started_at],
+                        MergeRequest::Metrics.arel_table[:latest_build_finished_at])
       end
     end
   end