Use token from redis in gcp project billing worker

app/workers/check_gcp_project_billing_worker.rb

+require 'securerandom'
+
 class CheckGcpProjectBillingWorker
   include ApplicationWorker
   include ClusterQueue
 
   LEASE_TIMEOUT = 15.seconds.to_i
 
+  def self.generate_redis_token_key
+    SecureRandom.uuid
+  end
+
   def self.redis_shared_state_key_for(token)
     "gitlab:gcp:#{token.hash}:billing_enabled"
   end
 
-  def perform(token)
+  def perform(token_key)
+    return unless token_key
+
+    token = get_token(token_key)
     return unless token
     return unless try_obtain_lease_for(token)
 
@@ -20,6 +29,10 @@ class CheckGcpProjectBillingWorker
 
   private
 
+  def get_token(token_key)
+    Gitlab::Redis::SharedState.with { |redis| redis.get(token_key) }
+  end
+
   def try_obtain_lease_for(token)
     Gitlab::ExclusiveLease
       .new("check_gcp_project_billing_worker:#{token.hash}", timeout: LEASE_TIMEOUT)

spec/workers/check_gcp_project_billing_worker_spec.rb

@@ -3,7 +3,12 @@ require 'spec_helper'
 describe CheckGcpProjectBillingWorker do
   describe '.perform' do
     let(:token) { 'bogustoken' }
-    subject { described_class.new.perform(token) }
+    subject { described_class.new.perform('token_key') }
+
+    context 'when there is a token in redis' do
+      before do
+        allow_any_instance_of(described_class).to receive(:get_token).and_return(token)
+      end
 
       context 'when there is no lease' do
         before do
@@ -39,4 +44,17 @@ describe CheckGcpProjectBillingWorker do
         end
       end
     end
+
+    context 'when there is no token in redis' do
+      before do
+        allow_any_instance_of(described_class).to receive(:get_token).and_return(nil)
+      end
+
+      it 'does not call the service' do
+        expect(CheckGcpProjectBillingService).not_to receive(:new)
+
+        subject
+      end
+    end
+  end
 end