1 files changed, 18 insertions, 6 deletions
diff --git a/spec/models/ci/pipeline_spec.rb b/spec/models/ci/pipeline_spec.rb
index 20520b0a111..56cc10fa083 100644
--- a/spec/models/ci/pipeline_spec.rb
+++ b/spec/models/ci/pipeline_spec.rb
@@ -503,13 +503,13 @@ describe Ci::Pipeline, models: true do
     end
   end
 
-  describe '#with_warnings?' do
-    subject { pipeline.with_warnings? }
+  describe '#has_warnings?' do
+    subject { pipeline.has_warnings? }
 
     context 'build which is allowed to fail fails' do
       before do
-        FactoryGirl.create :ci_build, :success, pipeline: pipeline, name: 'rspec'
-        FactoryGirl.create :ci_build, :allowed_to_fail, :failed, pipeline: pipeline, name: 'rubocop'
+        create :ci_build, :success, pipeline: pipeline, name: 'rspec'
+        create :ci_build, :allowed_to_fail, :failed, pipeline: pipeline, name: 'rubocop'
       end
       
       it 'returns true' do
@@ -519,13 +519,25 @@ describe Ci::Pipeline, models: true do
 
     context 'build which is allowed to fail succeeds' do
       before do
-        FactoryGirl.create :ci_build, :success, pipeline: pipeline, name: 'rspec'
-        FactoryGirl.create :ci_build, :allowed_to_fail, :success, pipeline: pipeline, name: 'rubocop'
+        create :ci_build, :success, pipeline: pipeline, name: 'rspec'
+        create :ci_build, :allowed_to_fail, :success, pipeline: pipeline, name: 'rubocop'
       end
       
       it 'returns false' do
         is_expected.to be_falsey
       end
     end
+
+    context 'build is retried and succeeds' do
+      before do
+        create :ci_build, :success, pipeline: pipeline, name: 'rubocop'
+        create :ci_build, :failed, pipeline: pipeline, name: 'rspec'
+        create :ci_build, :success, pipeline: pipeline, name: 'rspec'
+      end
+
+      it 'returns false' do
+        is_expected.to be_falsey
+      end
+    end
   end
 end