wip

spatie · Nov 30, 2021 · 71e4bf2 · 71e4bf2
1 parent 8b15d4f
commit 71e4bf2
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/ResultStores/StoredCheckResults/StoredCheckResults.php b/src/ResultStores/StoredCheckResults/StoredCheckResults.php
@@ -6,6 +6,7 @@
 use DateTimeInterface;
 use Illuminate\Support\Collection;
 use Spatie\Health\Enums\Status;
+use function Pest\Laravel\instance;
 
 class StoredCheckResults
 {
@@ -60,10 +61,19 @@ public function containsFailingCheck(): bool
         return ! $this->allChecksOk();
     }
 
-    public function containsCheckWithStatus(Status $status): bool
+    /**
+     * @param array<int, Status>|Status $statuses
+     *
+     * @return bool
+     */
+    public function containsCheckWithStatus(array|Status $statuses): bool
     {
+        if ($statuses instanceof Status) {
+            $statuses = array($statuses);
+        }
+
         return $this->storedCheckResults->contains(
-            fn (StoredCheckResult $line) => $line->status === $status->value
+            fn (StoredCheckResult $line) => in_array($line->status, $statuses)
         );
     }
 

diff --git a/tests/ResultStores/StoredCheckResultsTest.php b/tests/ResultStores/StoredCheckResultsTest.php
@@ -15,6 +15,9 @@
     expect($storedCheckResults->containsCheckWithStatus(Status::warning()))->toBeTrue();
     expect($storedCheckResults->containsCheckWithStatus(Status::failed()))->toBeFalse();
     expect($storedCheckResults->containsCheckWithStatus(Status::crashed()))->toBeFalse();
+
+    expect($storedCheckResults->containsCheckWithStatus([Status::warning(), Status::failed()]))->toBeTrue();
+    expect($storedCheckResults->containsCheckWithStatus([Status::crashed(), Status::failed()]))->toBeFalse();
 });
 
 function makeStoredCheckResultWithStatus(Status $status): StoredCheckResult