Fix random()

luminarix · Dec 25, 2024 · d76e036 · d76e036
1 parent 6f4e6d2
commit d76e036
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/EnumConcern.php b/src/EnumConcern.php
@@ -92,7 +92,7 @@ public static function getLabel(UnitEnum $case): string
     /**
      * Selects and returns a random enum case.
      */
-    public static function random(int $count = 1): Collection
+    public static function random(?int $count = null): Collection|static
     {
         return self::toCollection()->random($count);
     }

diff --git a/tests/FunctionTest.php b/tests/FunctionTest.php
@@ -61,6 +61,14 @@ function testEnumMethods($enumClass, $expectedValues, $expectedNames, $expectedA
             ->toBeIn($enumClass::cases());
     });
 
+    it("gets a random enum case for {$enumClass} with count", function () use ($enumClass) {
+        $test = $enumClass::random(2);
+
+        expect($test)
+            ->toBeInstanceOf(Collection::class)
+            ->toHaveCount(2);
+    });
+
     it("checks if a value exists in the enum for {$enumClass}", function () use ($enumClass, $expectedValues) {
         $exists = $enumClass::hasValue($expectedValues[0]);
         $notExists = $enumClass::hasValue('nonexistent');