From 9d46ad259c058451b59eae6da3ae39332a70c755 Mon Sep 17 00:00:00 2001
From: tychovrahe <brunam@seznam.cz>
Date: Fri, 21 Mar 2025 14:58:47 +0100
Subject: [PATCH] fix(core/prodtest): remove final wait from report commands

[no changelog]
---
 core/embed/projects/prodtest/cmd/prodtest_pmic.c | 15 ++++++++-------
 core/embed/projects/prodtest/cmd/prodtest_wpc.c  | 15 ++++++++-------
 2 files changed, 16 insertions(+), 14 deletions(-)

diff --git a/core/embed/projects/prodtest/cmd/prodtest_pmic.c b/core/embed/projects/prodtest/cmd/prodtest_pmic.c
index f2786eb868..f9c99fdf84 100644
--- a/core/embed/projects/prodtest/cmd/prodtest_pmic.c
+++ b/core/embed/projects/prodtest/cmd/prodtest_pmic.c
@@ -187,13 +187,14 @@ static void prodtest_pmic_report(cli_t* cli) {
         (int)report.die_temp, (int)abs(report.die_temp * 1000) % 1000,
         report.ibat_meas_status, report.buck_status, state);
 
-    do {
-      if (cli_aborted(cli)) {
-        return;
-      }
-    } while (!ticks_expired(ticks + period));
-
-    ticks += period;
+    if (count > 0) {
+      do {
+        if (cli_aborted(cli)) {
+          return;
+        }
+      } while (!ticks_expired(ticks + period));
+      ticks += period;
+    }
   }
 
   cli_ok(cli, "");
diff --git a/core/embed/projects/prodtest/cmd/prodtest_wpc.c b/core/embed/projects/prodtest/cmd/prodtest_wpc.c
index 18ac242cfa..5bd8797eea 100644
--- a/core/embed/projects/prodtest/cmd/prodtest_wpc.c
+++ b/core/embed/projects/prodtest/cmd/prodtest_wpc.c
@@ -151,13 +151,14 @@ static void prodtest_wpc_report(cli_t* cli) {
                  report.opfreq, (int)report.ntc,
                  (int)abs(report.ntc * 1000) % 1000);
 
-    do {
-      if (cli_aborted(cli)) {
-        return;
-      }
-    } while (!ticks_expired(ticks + period));
-
-    ticks += period;
+    if (count > 0) {
+      do {
+        if (cli_aborted(cli)) {
+          return;
+        }
+      } while (!ticks_expired(ticks + period));
+      ticks += period;
+    }
   }
 
   cli_ok(cli, "");