Ensure finish event is emitted on test crashes

Author: vinistock

lib/ruby_lsp/test_reporters/lsp_reporter.rb

@@ -9,6 +9,9 @@ module RubyLsp
   class LspReporter
     include Singleton
 
+    #: bool
+    attr_reader :invoked_shutdown
+
     #: -> void
     def initialize
       port = ENV["RUBY_LSP_REPORTER_PORT"]
@@ -19,6 +22,8 @@ def initialize
         require "stringio"
         StringIO.new
       end #: IO | StringIO
+
+      @invoked_shutdown = false #: bool
     end
 
     #: -> void
@@ -34,6 +39,8 @@ def shutdown
     # reporter, use `shutdown` instead
     #: -> void
     def internal_shutdown
+      @invoked_shutdown = true
+
       send_message("finish")
       @io.close
     end
@@ -161,4 +168,10 @@ def send_message(method_name, **params)
     File.write(File.join(".ruby-lsp", "coverage_result.json"), coverage_results.to_json)
     RubyLsp::LspReporter.instance.internal_shutdown
   end
+elsif ENV["RUBY_LSP_TEST_RUNNER"] && !ENV["RUBY_LSP_ENV"] == "test"
+  at_exit do
+    # If the test process crashed immediately without finishing the tests, we still need to tell the extension that the
+    # execution ended so that it can clean up
+    RubyLsp::LspReporter.instance.internal_shutdown unless RubyLsp::LspReporter.instance.invoked_shutdown
+  end
 end