add tests

yiisoft · Jan 29, 2025 · b405f17 · b405f17
1 parent 4883903
commit b405f17
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/tests/Db/Helper/DbArrayHelperTest.php b/tests/Db/Helper/DbArrayHelperTest.php
@@ -67,4 +67,38 @@ public function testIndexWithNonExistingIndexBy(): void
 
         DbArrayHelper::index($rows, 'non-existing-key', ['key']);
     }
+
+    public function testIndexWithArrangeBy(): void
+    {
+        $rows = [
+            ['key' => 'value1'],
+            ['key' => 'value2'],
+        ];
+
+        set_error_handler(static function (int $errno, string $errstr) {
+            restore_error_handler();
+            throw new \Exception('E_WARNING: ' . $errstr, $errno);
+        }, E_WARNING);
+
+        $this->expectExceptionMessage('E_WARNING: Undefined array key "non-existing-key"');
+
+        DbArrayHelper::index($rows, null, ['non-existing-key']);
+    }
+
+    public function testIndexWithClosureIndexByAndArrangeBy(): void
+    {
+        $rows = [
+            ['key' => 'value1'],
+            ['key' => 'value2'],
+        ];
+
+        $this->assertSame([
+            'value1' => [
+                'value1' => ['key' => 'value1'],
+            ],
+            'value2' => [
+                'value2' => ['key' => 'value2'],
+            ],
+        ], DbArrayHelper::index($rows, fn ($row) => $row['key'], ['key']));
+    }
 }