Clear the model's association cache before resolving its association

Author: st0012

lib/ruby_lsp/ruby_lsp_rails/server.rb

@@ -152,6 +152,8 @@ def resolve_association_target(params)
           }
         end
 
+        # Clear the cache so that we get the latest associations.
+        const.clear_reflections_cache
         association_klass = const.reflect_on_association(params[:association_name].intern).klass
 
         source_location = Object.const_source_location(association_klass.to_s)

sorbet/rbi/shims/user.rbi

@@ -0,0 +1,14 @@
+# typed: strict
+# frozen_string_literal: true
+
+class User
+  class << self
+    sig { params(association_name: Symbol).void }
+    def has_many(association_name)
+    end
+
+    sig { returns(T::Hash[String, T.untyped]) }
+    def reflections
+    end
+  end
+end

test/ruby_lsp_rails/server_test.rb

@@ -103,6 +103,26 @@ def <(other)
     assert_nil(response.fetch(:result))
   end
 
+  test "resolve association reflect the latest associations" do
+    response = @server.execute(
+      "association_target_location",
+      { model_name: "User", association_name: :memberships },
+    )
+    assert_nil(response.fetch(:result))
+
+    User.has_many(:memberships)
+
+    response = @server.execute(
+      "association_target_location",
+      { model_name: "User", association_name: :memberships },
+    )
+
+    location = response[:result][:location]
+    assert_match(%r{test/dummy/app/models/membership.rb:3$}, location)
+  ensure
+    User.reflections.delete("memberships")
+  end
+
   test "resolve association handles class_name option" do
     response = @server.execute(
       "association_target_location",