From 901317e4a352e3df011c7bad23bfb5c7e06ce499 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tina=20M=C3=BCller?= <tina.mueller@suse.com>
Date: Mon, 13 Nov 2023 19:51:14 +0100
Subject: [PATCH] Make openqa-cli retry also on connection errors

There may be temporary network problems or an overloaded webui. In those
cases we also want to retry.

Issue: https://progress.opensuse.org/issues/138545
---
 lib/OpenQA/Command.pm | 2 +-
 t/43-cli-api.t        | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/lib/OpenQA/Command.pm b/lib/OpenQA/Command.pm
index 2cfe2a302eb..de977df7c70 100644
--- a/lib/OpenQA/Command.pm
+++ b/lib/OpenQA/Command.pm
@@ -124,7 +124,7 @@ sub retry_tx ($self, $client, $tx, $retries = undef, $delay = undef) {
     for (;; --$retries) {
         $tx = $client->start($tx);
         my $res_code = $tx->res->code // 0;
-        return $self->handle_result($tx) unless $res_code =~ /50[23]/ && $retries > 0;
+        return $self->handle_result($tx) unless $res_code =~ /^(50[23]|0)$/ && $retries > 0;
         print encode('UTF-8',
             "Request failed, hit error $res_code, retrying up to $retries more times after waiting …\n");
         sleep $delay;
diff --git a/t/43-cli-api.t b/t/43-cli-api.t
index 6b2ad11c82e..f3f59766a8b 100644
--- a/t/43-cli-api.t
+++ b/t/43-cli-api.t
@@ -451,6 +451,11 @@ EOF
 
     ($stdout, $stderr, @result) = capture sub { $api->run('--retries', '1', @params) };
     like $stdout, qr/failed.*retrying/, 'requests are retried on error if requested';
+
+    @params = ('--host', 'http://localhost:123456', '--retries', 1, 'api', 'test');
+    ($stdout, $stderr, @result) = capture sub { $api->run(@params) };
+    like $stderr, qr/Connection refused/, 'aborts on connection refused';
+    like $stdout, qr/failed.*retrying/, 'requests are retried on error if requested';
 };
 
 subtest 'Pretty print JSON' => sub {