Fix {flame,ice}graph for minitest and test-unit

Author: ajbouh

src/qa/runner-assets/ruby/minitest.rb

@@ -133,6 +133,9 @@ def report
     aggregate.default = [] # dumb. group_by should provide this
 
     corrected_time = ::Qa::Time.now_f - @suite_start_time
+
+    @trace.emit_final_stats
+
     emit(
         'type' => 'final',
         'time' => corrected_time,

src/qa/runner-assets/ruby/shared.rb

@@ -1183,7 +1183,6 @@ def main(args)
     trace_events = []
     qa_trace = ::Qa::Trace.new(Process.pid) { |e| trace_events.push(e) }
     trace_probes.each { |trace_probe| qa_trace.define_tracer(trace_probe) }
-    qa_trace.start
 
     socket = ::Qa::ClientSocket.connect(args.shift)
 
@@ -1193,6 +1192,11 @@ def main(args)
     worker_envs, initial_files = initial_config['workerEnvs'], initial_config['files']
     passthrough = initial_config['passthrough']
 
+    if passthrough['sampleStack']
+      Qa::Trace.enable_stackprof!
+    end
+    qa_trace.start
+
     # Delegate prefork actions.
     @prefork.call(initial_files)
 
@@ -1230,10 +1234,6 @@ def main(args)
       Qa::Stdcom.disable!
     end
 
-    if passthrough['sampleStack']
-      Qa::Trace.enable_stackprof!
-    end
-
     case passthrough['errorsCaptureLocals'].to_s
     when 'true'
       TracePoint.new(:raise) do |tp|

src/qa/runner-assets/ruby/test-unit.rb

@@ -47,6 +47,8 @@ def tapout_before_suite(result)
     end
 
     def tapout_after_suite(elapsed_time)
+      @trace.emit_final_stats
+
       emit(
           'type' => 'final',
           'time' => elapsed_time,