Add WikiDirectory#to_partial_path

2 files changed, 14 insertions, 0 deletions

diff --git a/app/models/wiki_directory.rb b/app/models/wiki_directory.rb
index c126a4d0421..561e5a497bc 100644
--- a/app/models/wiki_directory.rb
+++ b/app/models/wiki_directory.rb
@@ -10,4 +10,10 @@ class WikiDirectory
     @pages = pages
     @directories = directories
   end
+
+  # Relative path to the partial to be used when rendering collections
+  # of this object.
+  def to_partial_path
+    'projects/wikis/wiki_directory'
+  end
 end
diff --git a/spec/models/wiki_directory_spec.rb b/spec/models/wiki_directory_spec.rb
index 8362a285c54..fac70f8d3c7 100644
--- a/spec/models/wiki_directory_spec.rb
+++ b/spec/models/wiki_directory_spec.rb
@@ -42,4 +42,12 @@ RSpec.describe WikiDirectory, models: true do
       end
     end
   end
+
+  describe '#to_partial_path' do
+    it 'returns the relative path to the partial to be used' do
+      directory = build(:wiki_directory)
+
+      expect(directory.to_partial_path).to eq('projects/wikis/wiki_directory')
+    end
+  end
 end