Wrap updating of cache after pipeline transition in class method

2 files changed, 16 insertions, 6 deletions

diff --git a/spec/lib/gitlab/cache/ci/project_build_status_spec.rb b/spec/lib/gitlab/cache/ci/project_build_status_spec.rb
index b7504788255..7b9e959b087 100644
--- a/spec/lib/gitlab/cache/ci/project_build_status_spec.rb
+++ b/spec/lib/gitlab/cache/ci/project_build_status_spec.rb
@@ -12,6 +12,20 @@ describe Gitlab::Cache::Ci::ProjectBuildStatus do
     end
   end
 
+  describe '.update_for_pipeline' do
+    it 'refreshes the cache if nescessary' do
+      pipeline = build_stubbed(:ci_pipeline, sha: '123456', status: 'success')
+      fake_status = double
+      expect(described_class).to receive(:new).
+                                   with(pipeline.project, sha: '123456', status: 'success').
+                                   and_return(fake_status)
+
+      expect(fake_status).to receive(:store_in_cache_if_needed)
+
+      described_class.update_for_pipeline(pipeline)
+    end
+  end
+
   describe '#has_status?' do
     it "is false when the status wasn't loaded yet" do
       expect(pipeline_status.has_status?).to be_falsy
diff --git a/spec/models/ci/pipeline_spec.rb b/spec/models/ci/pipeline_spec.rb
index 5caba6ae703..504442392ea 100644
--- a/spec/models/ci/pipeline_spec.rb
+++ b/spec/models/ci/pipeline_spec.rb
@@ -1083,12 +1083,8 @@ describe Ci::Pipeline, models: true do
     let(:pipeline) { create(:ci_pipeline, sha: '123456') }
 
     it 'updates the cached status' do
-      fake_status = double
-      expect(Gitlab::Cache::Ci::ProjectBuildStatus).to receive(:new).
-                                                         with(pipeline.project, sha: '123456', status: 'manual').
-                                                         and_return(fake_status)
-
-      expect(fake_status).to receive(:store_in_cache_if_needed)
+      expect(Gitlab::Cache::Ci::ProjectBuildStatus).to receive(:update_for_pipeline).
+                                                         with(pipeline)
 
       pipeline.block
     end