1 files changed, 19 insertions, 9 deletions

diff --git a/spec/graphql/resolvers/blame_resolver_spec.rb b/spec/graphql/resolvers/blame_resolver_spec.rb
index a3344132928..ff6bfa97def 100644
--- a/spec/graphql/resolvers/blame_resolver_spec.rb
+++ b/spec/graphql/resolvers/blame_resolver_spec.rb
@@ -12,7 +12,7 @@ RSpec.describe Resolvers::BlameResolver, feature_category: :source_code_manageme
     let(:path) { 'files/ruby/popen.rb' }
     let(:commit) { project.commit('master') }
     let(:blob) { project.repository.blob_at(commit.id, path) }
-    let(:args) { { from_line: 1, to_line: 2 } }
+    let(:args) { { from_line: 1, to_line: 100 } }
 
     subject(:resolve_blame) { resolve(described_class, obj: blob, args: args, ctx: { current_user: user }) }
 
@@ -39,10 +39,10 @@ RSpec.describe Resolvers::BlameResolver, feature_category: :source_code_manageme
         end
       end
 
-      shared_examples 'argument error' do
+      shared_examples 'argument error' do |error_message|
         it 'raises an ArgumentError' do
           expect_graphql_error_to_be_created(Gitlab::Graphql::Errors::ArgumentError,
-            '`from_line` and `to_line` must be greater than or equal to 1') do
+            error_message) do
             resolve_blame
           end
         end
@@ -52,23 +52,33 @@ RSpec.describe Resolvers::BlameResolver, feature_category: :source_code_manageme
         context 'when from_line is below 1' do
           let(:args) { { from_line: 0, to_line: 2 } }
 
-          it_behaves_like 'argument error'
+          it_behaves_like 'argument error', '`from_line` and `to_line` must be greater than or equal to 1'
         end
 
         context 'when to_line is below 1' do
           let(:args) { { from_line: 1, to_line: 0 } }
 
-          it_behaves_like 'argument error'
+          it_behaves_like 'argument error', '`from_line` and `to_line` must be greater than or equal to 1'
         end
 
         context 'when to_line less than from_line' do
           let(:args) { { from_line: 3, to_line: 1 } }
 
+          it_behaves_like 'argument error', '`to_line` must be greater than or equal to `from_line`'
+        end
+
+        context 'when difference between to_line and from_line is greater then 99' do
+          let(:args) { { from_line: 3, to_line: 103 } }
+
+          it_behaves_like 'argument error',
+            '`to_line` must be greater than or equal to `from_line` and smaller than `from_line` + 100'
+        end
+
+        context 'when to_line and from_line are the same' do
+          let(:args) { { from_line: 1, to_line: 1 } }
+
           it 'returns blame object' do
-            expect_graphql_error_to_be_created(Gitlab::Graphql::Errors::ArgumentError,
-              '`to_line` must be greater than or equal to `from_line`') do
-              resolve_blame
-            end
+            expect(resolve_blame).to be_an_instance_of(Gitlab::Blame)
           end
         end