DEBUG

Author: vinistock

test/ruby_lsp_rails/server_test.rb

@@ -3,7 +3,6 @@
 
 require "test_helper"
 require "ruby_lsp/ruby_lsp_rails/server"
-require "timeout"
 
 class ServerTest < ActiveSupport::TestCase
   setup do
@@ -276,36 +275,41 @@ def print_it!
 
     addon_path = File.expand_path("my_addon.rb")
     File.write(addon_path, <<~RUBY)
+      module Patch
+        def setproctitle(title)
+          $TEST_TITLE = title
+          super
+        end
+
+        Process.singleton_class.prepend(Patch)
+      end
+
       class MyServerAddon < RubyLsp::Rails::ServerAddon
         def name
           "MyAddon"
         end
 
         def execute(request, params)
-          parent_process_title = `ps -p \#{Process.pid} -o comm=`.lines.last.strip
           file = "process_name.txt"
+
           pid = fork do
             # We can't directly send a message in these tests because we're using a StringIO as stdout instead of the
             # actual pipe, which means that the child process doesn't have access to the same object
-            process_title = `ps -p \#{Process.pid} -o comm=`.lines.last.strip
-            File.write(file, process_title)
+            File.write(file, $TEST_TITLE)
           end
 
           Process.wait(pid)
 
-          parent_process_title_changed = `ps -p \#{Process.pid} -o comm=`.lines.last.strip != parent_process_title
-          send_message({ process_name: File.read(file), changed_parent_title: parent_process_title_changed })
+          send_message({ process_name: File.read(file) })
           File.delete(file)
-        rescue => e
-          send_message({ error: e.full_message })
         end
       end
     RUBY
 
     begin
       @server.execute("server_addon/register", server_addon_path: addon_path)
       @server.execute("server_addon/delegate", server_addon_name: "MyAddon", request_name: "dsl")
-      assert_equal(response, { process_name: "ruby-lsp-rails: #{addon_path}", changed_parent_title: false })
+      assert_equal(response, { process_name: "ruby-lsp-rails: #{addon_path}" })
     ensure
       FileUtils.rm(addon_path)
     end
@@ -316,38 +320,41 @@ def execute(request, params)
 
     addon_path = File.expand_path("my_other_addon.rb")
     File.write(addon_path, <<~RUBY)
+      module Patch
+        def setproctitle(title)
+          $TEST_TITLE = title
+          super
+        end
+
+        Process.singleton_class.prepend(Patch)
+      end
+
       class MyOtherServerAddon < RubyLsp::Rails::ServerAddon
         def name
           "MyOtherAddon"
         end
 
         def execute(request, params)
-          parent_process_title = `ps -p \#{Process.pid} -o comm=`.lines.last.strip
           file = "other_process_name.txt"
           pid = fork
 
           if pid
             Process.wait(pid)
-            parent_process_title_changed = `ps -p \#{Process.pid} -o comm=`.lines.last.strip != parent_process_title
-            send_message({ process_name: File.read(file), changed_parent_title: parent_process_title_changed })
+            send_message({ process_name: File.read(file) })
             File.delete(file)
           else
-            process_title = `ps -p \#{Process.pid} -o comm=`.lines.last.strip
-            File.write(file, process_title)
-
+            File.write(file, $TEST_TITLE)
             # Exit from the child process or else we're stuck in the infinite loop of the server
             exit!
           end
-        rescue => e
-          send_message({ error: e.full_message })
         end
       end
     RUBY
 
     begin
       @server.execute("server_addon/register", server_addon_path: addon_path)
       @server.execute("server_addon/delegate", server_addon_name: "MyOtherAddon", request_name: "dsl")
-      assert_equal(response, { process_name: "ruby-lsp-rails: #{addon_path}", changed_parent_title: false })
+      assert_equal(response, { process_name: "ruby-lsp-rails: #{addon_path}" })
     ensure
       FileUtils.rm(addon_path)
     end
@@ -356,9 +363,7 @@ def execute(request, params)
   private
 
   def response
-    Timeout.timeout(2) do
-      _headers, content = @stdout.string.split("\r\n\r\n")
-      JSON.parse(content, symbolize_names: true)
-    end
+    _headers, content = @stdout.string.split("\r\n\r\n")
+    JSON.parse(content, symbolize_names: true)
   end
 end