class RubyLsp::TestUnitReporter

def add_fault(result)
  super
  return unless @current_uri && @current_test_id
  case result
  when ::Test::Unit::Failure
    LspReporter.instance.record_fail(id: @current_test_id, message: result.message, uri: @current_uri)
  when ::Test::Unit::Error
    LspReporter.instance.record_error(id: @current_test_id, message: result.message, uri: @current_uri)
  when ::Test::Unit::Pending
    LspReporter.instance.record_skip(id: @current_test_id, uri: @current_uri)
  end
end

def attach_to_mediator
  # Events we care about
  @mediator.add_listener(Test::Unit::TestResult::FAULT, &method(:add_fault))
  @mediator.add_listener(Test::Unit::TestCase::STARTED_OBJECT, &method(:test_started))
  @mediator.add_listener(Test::Unit::TestCase::FINISHED_OBJECT, &method(:test_finished))
  @mediator.add_listener(Test::Unit::UI::TestRunnerMediator::FINISHED, &method(:finished))
  # Other events needed for the console test runner to print
  @mediator.add_listener(Test::Unit::UI::TestRunnerMediator::STARTED, &method(:started))
  @mediator.add_listener(Test::Unit::TestSuite::STARTED_OBJECT, &method(:test_suite_started))
  @mediator.add_listener(Test::Unit::TestSuite::FINISHED_OBJECT, &method(:test_suite_finished))
end

def finished(elapsed_time)
  super
  LspReporter.instance.shutdown
end

def initialize(suite, options = {})
  super
  @current_uri = nil #: URI::Generic?
  @current_test_id = nil #: String?
end

def test_finished(test)
  super
  return unless test.passed? && @current_uri && @current_test_id
  LspReporter.instance.record_pass(id: @current_test_id, uri: @current_uri)
end

def test_started(test)
  super
  uri, line = LspReporter.instance.uri_and_line_for(test.method(test.method_name))
  return unless uri
  @current_uri = uri
  @current_test_id = "#{test.class.name}##{test.method_name}"
  LspReporter.instance.start_test(id: @current_test_id, uri: @current_uri, line: line)
end