1 files changed, 55 insertions, 0 deletions
diff --git a/spec/lib/gitlab/repository_size_error_message_spec.rb b/spec/lib/gitlab/repository_size_error_message_spec.rb
new file mode 100644
index 00000000000..9e4d19cc572
--- /dev/null
+++ b/spec/lib/gitlab/repository_size_error_message_spec.rb
@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe Gitlab::RepositorySizeErrorMessage do
+  let(:checker) do
+    Gitlab::RepositorySizeChecker.new(
+      current_size_proc: -> { 15.megabytes },
+      limit: 10.megabytes
+    )
+  end
+
+  let(:message) { checker.error_message }
+  let(:base_message) { 'because this repository has exceeded its size limit of 10 MB by 5 MB' }
+
+  describe 'error messages' do
+    describe '#commit_error' do
+      it 'returns the correct message' do
+        expect(message.commit_error).to eq("Your changes could not be committed, #{base_message}")
+      end
+    end
+
+    describe '#merge_error' do
+      it 'returns the correct message' do
+        expect(message.merge_error).to eq("This merge request cannot be merged, #{base_message}")
+      end
+    end
+
+    describe '#push_error' do
+      context 'with exceeded_limit value' do
+        let(:rejection_message) do
+          'because this repository has exceeded its size limit of 10 MB by 15 MB'
+        end
+
+        it 'returns the correct message' do
+          expect(message.push_error(10.megabytes))
+            .to eq("Your push has been rejected, #{rejection_message}. #{message.more_info_message}")
+        end
+      end
+
+      context 'without exceeded_limit value' do
+        it 'returns the correct message' do
+          expect(message.push_error)
+            .to eq("Your push has been rejected, #{base_message}. #{message.more_info_message}")
+        end
+      end
+    end
+
+    describe '#new_changes_error' do
+      it 'returns the correct message' do
+        expect(message.new_changes_error).to eq("Your push to this repository would cause it to exceed the size limit of 10 MB so it has been rejected. #{message.more_info_message}")
+      end
+    end
+  end
+end