1 files changed, 10 insertions, 25 deletions
diff --git a/spec/finders/runner_jobs_finder_spec.rb b/spec/finders/runner_jobs_finder_spec.rb
index 9ed6f50ddfb..01f45a37ba8 100644
--- a/spec/finders/runner_jobs_finder_spec.rb
+++ b/spec/finders/runner_jobs_finder_spec.rb
@@ -37,38 +37,23 @@ describe RunnerJobsFinder do
     end
 
     context 'when order_by and sort are specified' do
-      context 'when order_by created_at' do
-        let(:params) { { order_by: 'created_at', sort: 'asc' } }
-        let!(:jobs) { Array.new(2) { create(:ci_build, runner: runner, project: project, user: create(:user)) } }
-
-        it 'sorts as created_at: :asc' do
-          is_expected.to match_array(jobs)
-        end
-
-        context 'when sort is invalid' do
-          let(:params) { { order_by: 'created_at', sort: 'invalid_sort' } }
-
-          it 'sorts as created_at: :desc' do
-            is_expected.to eq(jobs.sort_by { |p| -p.user.id })
-          end
-        end
-      end
-
-      context 'when order_by is invalid' do
-        let(:params) { { order_by: 'invalid_column', sort: 'asc' } }
-        let!(:jobs) { Array.new(2) { create(:ci_build, runner: runner, project: project, user: create(:user)) } }
+      context 'when order_by id and sort is asc' do
+        let(:params) { { order_by: 'id', sort: 'asc' } }
+        let!(:jobs) { create_list(:ci_build, 2, runner: runner, project: project, user: create(:user)) }
 
         it 'sorts as id: :asc' do
-          is_expected.to eq(jobs.sort_by { |p| p.id })
+          is_expected.to eq(jobs.sort_by(&:id))
         end
       end
+    end
 
-      context 'when both are nil' do
-        let(:params) { { order_by: nil, sort: nil } }
-        let!(:jobs) { Array.new(2) { create(:ci_build, runner: runner, project: project, user: create(:user)) } }
+    context 'when order_by is specified and sort is not specified' do
+      context 'when order_by id and sort is not specified' do
+        let(:params) { { order_by: 'id' } }
+        let!(:jobs) { create_list(:ci_build, 2, runner: runner, project: project, user: create(:user)) }
 
         it 'sorts as id: :desc' do
-          is_expected.to eq(jobs.sort_by { |p| -p.id })
+          is_expected.to eq(jobs.sort_by(&:id).reverse)
         end
       end
     end