return first element from random element

src/Extensions/Extension.php

@@ -42,6 +42,7 @@ protected function pickArrayRandomElements(array $array, int $elements = 1)
 
     protected function pickArrayRandomElement(array $array)
     {
-        return $this->pickArrayRandomElements($array)[0];
+        $elements = $this->pickArrayRandomElements($array);
+        return reset($elements);
     }
 }

tests/Unit/Extensions/ExtensionTest.php

@@ -2,7 +2,10 @@
 
 namespace Extensions;
 
+use Random\Randomizer;
+use ReflectionClass;
 use Xefi\Faker\Container\Container;
+use Xefi\Faker\Extensions\Extension;
 use Xefi\Faker\Tests\Unit\TestCase;
 
 class ExtensionTest extends TestCase
@@ -26,4 +29,34 @@ public function testExtensionReturnOne(): void
             $container->returnOne()
         );
     }
+
+    public function testExtensionPickArrayRandomElements(): void
+    {
+        $elements = range(1, 100);
+
+        $method = (new ReflectionClass(Extension::class))->getMethod('pickArrayRandomElements');
+        $result = $method->invoke(new Extension(new Randomizer), $elements, 5);
+
+        $this->assertCount(
+            5,
+            $result
+        );
+
+        foreach ($result as $item) {
+            $this->assertGreaterThanOrEqual(1, $item);
+            $this->assertLessThanOrEqual(100, $item);
+        }
+    }
+
+    public function testExtensionPickArrayRandomElement(): void
+    {
+        $elements = range(1, 100);
+
+        $method = (new ReflectionClass(Extension::class))->getMethod('pickArrayRandomElement');
+        $result = $method->invoke(new Extension(new Randomizer), $elements);
+
+        $this->assertIsNotArray($result);
+        $this->assertGreaterThanOrEqual(1, $result);
+        $this->assertLessThanOrEqual(100, $result);
+    }
 }