Isolate CheckGcpProjectBillingWorker specreturns

1 files changed, 5 insertions, 4 deletions

diff --git a/spec/workers/check_gcp_project_billing_worker_spec.rb b/spec/workers/check_gcp_project_billing_worker_spec.rb
index c6e624f65be..70738f36324 100644
--- a/spec/workers/check_gcp_project_billing_worker_spec.rb
+++ b/spec/workers/check_gcp_project_billing_worker_spec.rb
@@ -12,12 +12,13 @@ describe CheckGcpProjectBillingWorker do
     end
 
     it 'stores billing status in redis' do
+      redis_double = double
+
       expect(CheckGcpProjectBillingService).to receive_message_chain(:new, :execute).and_return(true)
-      subject
+      expect(Gitlab::Redis::SharedState).to receive(:with).and_yield(redis_double)
+      expect(redis_double).to receive(:set).with(CheckGcpProjectBillingWorker.redis_shared_state_key_for(token), anything)
 
-      Gitlab::Redis::SharedState.with do |redis|
-        expect(redis.get("gitlab:gcp:#{token}:billing_enabled")).to eq('true')
-      end
+      subject
     end
   end
 end