1 files changed, 30 insertions, 10 deletions
diff --git a/spec/lib/gitlab/checks/snippet_check_spec.rb b/spec/lib/gitlab/checks/snippet_check_spec.rb
index 037de8e9369..89417aaca4d 100644
--- a/spec/lib/gitlab/checks/snippet_check_spec.rb
+++ b/spec/lib/gitlab/checks/snippet_check_spec.rb
@@ -9,19 +9,30 @@ RSpec.describe Gitlab::Checks::SnippetCheck do
 
   let(:user_access) { Gitlab::UserAccessSnippet.new(user, snippet: snippet) }
   let(:default_branch) { snippet.default_branch }
+  let(:branch_name) { default_branch }
+  let(:creation) { false }
+  let(:deletion) { false }
 
-  subject { Gitlab::Checks::SnippetCheck.new(changes, default_branch: default_branch, logger: logger) }
+  subject { Gitlab::Checks::SnippetCheck.new(changes, default_branch: default_branch, root_ref: snippet.repository.root_ref, logger: logger) }
 
   describe '#validate!' do
     it 'does not raise any error' do
       expect { subject.validate! }.not_to raise_error
     end
 
+    shared_examples 'raises and logs error' do
+      specify do
+        expect(Gitlab::ErrorTracking).to receive(:log_exception).with(instance_of(Gitlab::GitAccess::ForbiddenError), default_branch: default_branch, branch_name: branch_name, creation: creation, deletion: deletion)
+
+        expect { subject.validate! }.to raise_error(Gitlab::GitAccess::ForbiddenError, 'You can not create or delete branches.')
+      end
+    end
+
     context 'trying to delete the branch' do
       let(:newrev) { '0000000000000000000000000000000000000000' }
 
-      it 'raises an error' do
-        expect { subject.validate! }.to raise_error(Gitlab::GitAccess::ForbiddenError, 'You can not create or delete branches.')
+      it_behaves_like 'raises and logs error' do
+        let(:deletion) { true }
       end
     end
 
@@ -29,14 +40,23 @@ RSpec.describe Gitlab::Checks::SnippetCheck do
       let(:oldrev) { '0000000000000000000000000000000000000000' }
       let(:ref) { 'refs/heads/feature' }
 
-      it 'raises an error' do
-        expect { subject.validate! }.to raise_error(Gitlab::GitAccess::ForbiddenError, 'You can not create or delete branches.')
+      it_behaves_like 'raises and logs error' do
+        let(:creation) { true }
+        let(:branch_name) { 'feature' }
       end
 
-      context "when branch is 'master'" do
-        let(:ref) { 'refs/heads/master' }
+      context 'when branch is the same as the default branch' do
+        let(:ref) { "refs/heads/#{default_branch}" }
 
-        it "allows the operation" do
+        it 'allows the operation' do
+          expect { subject.validate! }.not_to raise_error
+        end
+      end
+
+      context 'when snippet has an empty repo' do
+        let_it_be(:snippet) { create(:personal_snippet, :empty_repo) }
+
+        it 'allows the operation' do
           expect { subject.validate! }.not_to raise_error
         end
       end
@@ -45,8 +65,8 @@ RSpec.describe Gitlab::Checks::SnippetCheck do
     context 'when default_branch is nil' do
       let(:default_branch) { nil }
 
-      it 'raises an error' do
-        expect { subject.validate! }.to raise_error(Gitlab::GitAccess::ForbiddenError, 'You can not create or delete branches.')
+      it_behaves_like 'raises and logs error' do
+        let(:branch_name) { 'master' }
       end
     end
   end