Add spec

Signed-off-by: Baron Bloomer <[email protected]>

Author: bazay

lib/annotate/annotate_models.rb

@@ -614,11 +614,11 @@ def get_loaded_model(model_path, file)
 
       # Handle Rails apps with collapsed model paths
       model_paths = model_paths
-        .select do |model_path|
-          defined?(Rails) &&
-          Rails.autoloaders.main.collapse_dirs.any? &&
-          Rails.autoloaders.main.collapse_dirs.select { |path| path.match(model_path) }
-        end.map { |model_path| model_path.sub(/\/models/, '') }
+                    .select do |mpath|
+                      defined?(Rails) &&
+                      Rails.autoloaders.main.collapse_dirs.any? &&
+                      Rails.autoloaders.main.collapse_dirs.select { |path| path.match(mpath) }
+                    end.map { |mpath| mpath.sub(/\/models/, '') }
 
       model_paths
         .map { |path| get_loaded_model_by_path(path) }

spec/lib/annotate/annotate_models_spec.rb

@@ -1892,6 +1892,37 @@ class FooInsideBar < ActiveRecord::Base
           expect(klass.name).to eq('Bar::FooInsideBar')
         end
       end
+
+      context 'when class Bar::Foo is defined in Rails collapsed directory' do
+        before do
+          Rails = double('Rails')
+          # rubocop:disable RSpec/MessageChain
+          allow(Rails).to receive_message_chain(:autoloaders, :main, :collapse_dirs) { Set.new(["bar/models"]) }
+          # rubocop:enable RSpec/MessageChain
+          $LOAD_PATH.unshift(dir)
+        end
+
+        let :dir do
+          AnnotateModels.model_dir[0]
+        end
+
+        let :filename do
+          'bar/models/foo.rb'
+        end
+
+        let :file_content do
+          <<~EOS
+            module Bar
+              class Foo < ActiveRecord::Base
+              end
+            end
+          EOS
+        end
+
+        it 'works' do
+          expect(klass.name).to eql('Bar::Foo')
+        end
+      end
     end
 
     context 'when class is defined inside module and class name is not capitalized normally' do