From c425a30a43e95a38c0c57e0654ca47d656666b5a Mon Sep 17 00:00:00 2001
From: Annie Caron <annie.caron@shopify.com>
Date: Thu, 30 Apr 2026 11:37:30 -0400
Subject: [PATCH] Make HeartbeatProcess#tick! lease argument
 backward-compatible

The lease parameter was added to tick! in v0.85.0 but callers in the
monolith (MultiTick#heartbeat) still use the old 1-arg signature.
Make lease optional so existing callers work without changes.

Bump ci-queue to v0.94.0.
---
 ruby/Gemfile.lock                             |  2 +-
 ruby/lib/ci/queue/redis/base.rb               |  6 ++-
 ruby/lib/ci/queue/version.rb                  |  2 +-
 .../ci/queue/redis/heartbeat_process_test.rb  | 38 ++++++++++++++++---
 4 files changed, 38 insertions(+), 10 deletions(-)

diff --git a/ruby/Gemfile.lock b/ruby/Gemfile.lock
index 6b909ee5..592956fe 100644
--- a/ruby/Gemfile.lock
+++ b/ruby/Gemfile.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ci-queue (0.93.0)
+    ci-queue (0.94.0)
       logger
 
 GEM
diff --git a/ruby/lib/ci/queue/redis/base.rb b/ruby/lib/ci/queue/redis/base.rb
index 8e308000..1f454f27 100644
--- a/ruby/lib/ci/queue/redis/base.rb
+++ b/ruby/lib/ci/queue/redis/base.rb
@@ -318,8 +318,10 @@ def shutdown!
             end
           end
 
-          def tick!(id, lease)
-            send_message(TICK_COMMAND, id: id, lease: lease.to_s)
+          def tick!(id, lease = nil)
+            payload = { id: id }
+            payload[:lease] = lease.to_s if lease
+            send_message(TICK_COMMAND, **payload)
             @restart_attempts = 0
           rescue IOError, Errno::EPIPE => error
             @restart_attempts = (@restart_attempts || 0) + 1
diff --git a/ruby/lib/ci/queue/version.rb b/ruby/lib/ci/queue/version.rb
index d4abea01..03874a73 100644
--- a/ruby/lib/ci/queue/version.rb
+++ b/ruby/lib/ci/queue/version.rb
@@ -2,7 +2,7 @@
 
 module CI
   module Queue
-    VERSION = '0.93.0'
+    VERSION = '0.94.0'
     DEV_SCRIPTS_ROOT = ::File.expand_path('../../../../../redis', __FILE__)
     RELEASE_SCRIPTS_ROOT = ::File.expand_path('../redis', __FILE__)
   end
diff --git a/ruby/test/ci/queue/redis/heartbeat_process_test.rb b/ruby/test/ci/queue/redis/heartbeat_process_test.rb
index 283e0cf6..e9cba8a9 100644
--- a/ruby/test/ci/queue/redis/heartbeat_process_test.rb
+++ b/ruby/test/ci/queue/redis/heartbeat_process_test.rb
@@ -21,6 +21,18 @@ def write(*parts)
     end
   end
 
+  class PipedHeartbeatProcess < CI::Queue::Redis::Base::HeartbeatProcess
+    attr_reader :pipe
+
+    def initialize(pipe, *args)
+      super(*args)
+      @pipe = pipe
+    end
+
+    def boot!; end
+    def restart!; end
+  end
+
   def setup
     @hp = CI::Queue::Redis::Base::HeartbeatProcess.new(
       'redis://localhost:6379/0',
@@ -71,12 +83,12 @@ def test_restart_counter_resets_after_success
   end
 
   def test_tick_does_not_allocate_tick_marker_string
-    @hp.instance_variable_set(:@pipe, FakePipe.new)
-    @hp.tick!("test_id", "lease_id") # warm up any one-time caches
+    hp = PipedHeartbeatProcess.new(FakePipe.new, 'redis://localhost:6379/0', 'zset', 'owners', 'leases')
+    hp.tick!("test_id", "lease_id") # warm up any one-time caches
 
     ObjectSpace.trace_object_allocations_start
     begin
-      @hp.tick!("test_id", "lease_id")
+      hp.tick!("test_id", "lease_id")
     ensure
       ObjectSpace.trace_object_allocations_stop
     end
@@ -97,15 +109,29 @@ def test_tick_does_not_allocate_tick_marker_string
 
   def test_tick_sends_valid_tick_payload
     pipe = FakePipe.new
-    @hp.instance_variable_set(:@pipe, pipe)
+    hp = PipedHeartbeatProcess.new(pipe, 'redis://localhost:6379/0', 'zset', 'owners', 'leases')
 
-    @hp.tick!("test_id", "lease_id")
+    hp.tick!("test_id", "lease_id")
+
+    raw = pipe.buffer
+    header_size = [0].pack("L").bytesize
+    size = raw.byteslice(0, header_size).unpack1("L")
+    payload = raw.byteslice(header_size, size)
+
+    assert_equal(["tick!", { "id" => "test_id", "lease" => "lease_id" }], JSON.parse(payload))
+  end
+
+  def test_tick_without_lease_omits_lease_from_payload
+    pipe = FakePipe.new
+    hp = PipedHeartbeatProcess.new(pipe, 'redis://localhost:6379/0', 'zset', 'owners', 'leases')
+
+    hp.tick!("test_id")
 
     raw = pipe.buffer
     header_size = [0].pack("L").bytesize
     size = raw.byteslice(0, header_size).unpack1("L")
     payload = raw.byteslice(header_size, size)
 
-    assert_equal ["tick!", { "id" => "test_id", "lease" => "lease_id" }], JSON.parse(payload)
+    assert_equal(["tick!", { "id" => "test_id" }], JSON.parse(payload))
   end
 end