module Buildkite::TestCollector::MinitestPlugin

def after_teardown
  tracer = Thread.current[:_buildkite_tracer]
  if !tracer.nil?
    Thread.current[:_buildkite_tracer] = nil
    tracer.finalize
    trace = Buildkite::TestCollector::MinitestPlugin::Trace.new(self, history: tracer.history)
    Buildkite::TestCollector.uploader.traces[trace.source_location] = trace
  end
  super
end

def before_setup
  super
  tracer = Buildkite::TestCollector::Tracer.new(
    min_duration: Buildkite::TestCollector.trace_min_duration,
  )
  # The _buildkite prefix here is added as a safeguard against name collisions
  # as we are in the main thread
  Thread.current[:_buildkite_tracer] = tracer
end