From c664094f35e7c4eca9ef2abde133927af2a4e035 Mon Sep 17 00:00:00 2001
From: Nuno Maduro <enunomaduro@gmail.com>
Date: Tue, 21 Feb 2023 22:04:30 +0000
Subject: [PATCH] fix: `--retry` option update

---
 overrides/Runner/ResultCache/DefaultResultCache.php | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/overrides/Runner/ResultCache/DefaultResultCache.php b/overrides/Runner/ResultCache/DefaultResultCache.php
index f02a06c7..d3438e00 100644
--- a/overrides/Runner/ResultCache/DefaultResultCache.php
+++ b/overrides/Runner/ResultCache/DefaultResultCache.php
@@ -79,6 +79,11 @@ final class DefaultResultCache implements ResultCache
      */
     private array $defects = [];
 
+    /**
+     * @psalm-var array<string, TestStatus>
+     */
+    private array $currentDefects = [];
+
     /**
      * @psalm-var array<string, float>
      */
@@ -96,6 +101,7 @@ final class DefaultResultCache implements ResultCache
     public function setStatus(string $id, TestStatus $status): void
     {
         if ($status->isFailure() || $status->isError()) {
+            $this->currentDefects[$id] = $status;
             $this->defects[$id] = $status;
         }
     }
@@ -107,6 +113,10 @@ final class DefaultResultCache implements ResultCache
 
     public function setTime(string $id, float $time): void
     {
+        if (! isset($this->currentDefects[$id])) {
+            unset($this->defects[$id]);
+        }
+
         $this->times[$id] = $time;
     }