test: fix specs for run command

shakacode · Aug 21, 2024 · fcbfc73 · fcbfc73
1 parent d9f4ebb
commit fcbfc73
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/command/run.rb b/lib/command/run.rb
@@ -243,7 +243,7 @@ def update_runner_workload # rubocop:disable Metrics/CyclomaticComplexity, Metri
 
       step("Updating runner workload '#{runner_workload}'") do
         # Update runner workload
-        @expected_deployed_version = cp.cron_workload_deployed_version(runner_workload) + 1
+        @expected_deployed_version = (cp.cron_workload_deployed_version(runner_workload) || 0) + 1
         cp.apply_hash("kind" => "workload", "name" => runner_workload, "spec" => spec)
       end
     end
@@ -256,7 +256,7 @@ def wait_for_runner_workload_deploy
 
     def wait_for_runner_workload_update
       step("Waiting for runner workload '#{runner_workload}' to be updated", retry_on_failure: true) do
-        cp.cron_workload_deployed_version(runner_workload) >= expected_deployed_version
+        (cp.cron_workload_deployed_version(runner_workload) || 0) >= expected_deployed_version
       end
     end