test: CodeGenerator.php

a1383n · Jan 15, 2024 · 994b52a · 994b52a
1 parent 0ee282d
commit 994b52a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/Unit/ExampleTest.php b/tests/Unit/ExampleTest.php
@@ -14,8 +14,8 @@
         ->sequence(
             fn ($code, $length) => $code->toHaveLength((int)$length->value),
             fn ($code) => match ($type) {
-                OneTimePasswordCodeType::DIGIT => $code->toBeDigits(),
-                OneTimePasswordCodeType::ALPHA => $code->toBeAlpha(),
+                OneTimePasswordCodeType::DIGIT => is_string($code) ? expect($code)->toBeDigits() : $code->toBeDigits(),
+                OneTimePasswordCodeType::ALPHA => is_string($code) ? expect($code)->toBeAlpha() : $code->toBeAlpha(),
             }
         );
 })