Fix visibility of private project snippets for members when searching

1 files changed, 28 insertions, 8 deletions

diff --git a/spec/models/snippet_spec.rb b/spec/models/snippet_spec.rb
index 57365571a7b..0621c6a06ce 100644
--- a/spec/models/snippet_spec.rb
+++ b/spec/models/snippet_spec.rb
@@ -89,22 +89,42 @@ describe Snippet, models: true do
   end
 
   describe '.accessible_to' do
-    let(:author) { create(:author) }
-    let(:user) { create(:user) }
+    let(:author)  { create(:author) }
+    let(:project) { create(:empty_project) }
+
     let!(:public_snippet)   { create(:snippet, :public) }
     let!(:internal_snippet) { create(:snippet, :internal) }
     let!(:private_snippet)  { create(:snippet, :private, author: author) }
 
-    it 'returns only public snippets when user is nil' do
-      expect(described_class.accessible_to(nil)).to eq [public_snippet]
+    let!(:project_public_snippet)   { create(:snippet, :public, project: project) }
+    let!(:project_internal_snippet) { create(:snippet, :internal, project: project) }
+    let!(:project_private_snippet)  { create(:snippet, :private, project: project) }
+
+    it 'returns only public snippets when user is blank' do
+      expect(described_class.accessible_to(nil)).to match_array [public_snippet, project_public_snippet]
+    end
+
+    it 'returns only public, and internal snippets for regular users' do
+      user = create(:user)
+
+      expect(described_class.accessible_to(user)).to match_array [public_snippet, internal_snippet, project_public_snippet, project_internal_snippet]
     end
 
-    it 'returns only public, and internal snippets when user is not nil' do
-      expect(described_class.accessible_to(user)).to match_array [public_snippet, internal_snippet]
+    it 'returns public, internal snippets and project private snippets for project members' do
+      member = create(:user)
+      project.team << [member, :developer]
+
+      expect(described_class.accessible_to(member)).to match_array [public_snippet, internal_snippet, project_public_snippet, project_internal_snippet, project_private_snippet]
     end
 
-    it 'returns snippets where the user is the author' do
-      expect(described_class.accessible_to(author)).to match_array [public_snippet, internal_snippet, private_snippet]
+    it 'returns private snippets where the user is the author' do
+      expect(described_class.accessible_to(author)).to match_array [public_snippet, internal_snippet, private_snippet, project_public_snippet, project_internal_snippet]
+    end
+
+    it 'returns all snippets when for admins' do
+      admin = create(:admin)
+
+      expect(described_class.accessible_to(admin)).to match_array [public_snippet, internal_snippet, private_snippet, project_public_snippet, project_internal_snippet, project_private_snippet]
     end
   end