Add latest changes from gitlab-org/gitlab@master

2 files changed, 148 insertions, 0 deletions

diff --git a/spec/lib/gitlab/repository_size_checker_spec.rb b/spec/lib/gitlab/repository_size_checker_spec.rb
new file mode 100644
index 00000000000..61f76d716e5
--- /dev/null
+++ b/spec/lib/gitlab/repository_size_checker_spec.rb
@@ -0,0 +1,93 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe Gitlab::RepositorySizeChecker do
+  let(:current_size) { 0 }
+  let(:limit) { 50 }
+  let(:enabled) { true }
+
+  subject do
+    described_class.new(
+      current_size_proc: -> { current_size },
+      limit: limit,
+      enabled: enabled
+    )
+  end
+
+  describe '#enabled?' do
+    context 'when enabled' do
+      it 'returns true' do
+        expect(subject.enabled?).to be_truthy
+      end
+    end
+
+    context 'when limit is zero' do
+      let(:limit) { 0 }
+
+      it 'returns false' do
+        expect(subject.enabled?).to be_falsey
+      end
+    end
+  end
+
+  describe '#changes_will_exceed_size_limit?' do
+    let(:current_size) { 49 }
+
+    it 'returns true when changes go over' do
+      expect(subject.changes_will_exceed_size_limit?(2)).to be_truthy
+    end
+
+    it 'returns false when changes do not go over' do
+      expect(subject.changes_will_exceed_size_limit?(1)).to be_falsey
+    end
+  end
+
+  describe '#above_size_limit?' do
+    context 'when size is above the limit' do
+      let(:current_size) { 100 }
+
+      it 'returns true' do
+        expect(subject.above_size_limit?).to be_truthy
+      end
+    end
+
+    it 'returns false when not over the limit' do
+      expect(subject.above_size_limit?).to be_falsey
+    end
+  end
+
+  describe '#exceeded_size' do
+    context 'when current size is below or equal to the limit' do
+      let(:current_size) { 50 }
+
+      it 'returns zero' do
+        expect(subject.exceeded_size).to eq(0)
+      end
+    end
+
+    context 'when current size is over the limit' do
+      let(:current_size) { 51 }
+
+      it 'returns zero' do
+        expect(subject.exceeded_size).to eq(1)
+      end
+    end
+
+    context 'when change size will be over the limit' do
+      let(:current_size) { 50 }
+
+      it 'returns zero' do
+        expect(subject.exceeded_size(1)).to eq(1)
+      end
+    end
+
+    context 'when change size will not be over the limit' do
+      let(:current_size) { 49 }
+
+      it 'returns zero' do
+        expect(subject.exceeded_size(1)).to eq(0)
+      end
+    end
+  end
+end
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