1 files changed, 21 insertions, 0 deletions
diff --git a/spec/presenters/snippet_presenter_spec.rb b/spec/presenters/snippet_presenter_spec.rb
index 591d86652b6..423e9edc219 100644
--- a/spec/presenters/snippet_presenter_spec.rb
+++ b/spec/presenters/snippet_presenter_spec.rb
@@ -163,4 +163,25 @@ describe SnippetPresenter do
       end
     end
   end
+
+  describe '#blobs' do
+    let(:snippet) { personal_snippet }
+
+    subject { presenter.blobs }
+
+    context 'when snippet does not have a repository' do
+      it 'returns an array with one SnippetBlob' do
+        expect(subject.size).to eq(1)
+        expect(subject.first).to eq(snippet.blob)
+      end
+    end
+
+    context 'when snippet has a repository' do
+      let(:snippet) { create(:snippet, :repository, author: user) }
+
+      it 'returns an array with all repository blobs' do
+        expect(subject).to match_array(snippet.blobs)
+      end
+    end
+  end
 end