fix bug in fetchIdentities when using first()

Author: b1tc0re

src/Models/UserModel.php

@@ -142,6 +142,10 @@ protected function fetchIdentities(array $data): array
 
         $mappedUsers = $this->assignIdentities($data, $identities);
 
+        if ($data['method'] === 'first') {
+            $data['id'] = $data['data']->id;
+        }
+
         $data['data'] = $data['singleton'] ? $mappedUsers[$data['id']] : $mappedUsers;
 
         return $data;

tests/Unit/UserModelTest.php

@@ -278,4 +278,49 @@ public function testSaveArrayNoDataToUpdate(): void
 
         $users->save(['id' => $user->id]);
     }
+
+    public function testGetFirstIdentity(): void
+    {
+        $errors = [];
+        $users  = $this->createUserModel();
+        $user   = $this->createNewUser();
+        $users->save($user);
+
+        // Custom error handler
+        set_error_handler(static function ($severity, $message, $file, $line) use (&$errors) {
+            $errors[] = [
+                'message'  => $message,
+                'severity' => $severity,
+                'file'     => $file,
+                'line'     => $line,
+            ];
+
+            // Return true so that the default handler is not executed
+            return true;
+        });
+
+        try {
+            $user = $users->withIdentities()->first();
+            $this->assertSame(
+                $user->getIdentity('all')->secret,
+                'foo@bar.com',
+                'Verify first() method retrieves user with identity',
+            );
+        } finally {
+            // restore handler
+            restore_error_handler();
+        }
+
+        if ($errors !== []) {
+            $errorMessages = array_map(static fn ($error) => sprintf(
+                '[%s] %s in %s:%s',
+                $error['severity'],
+                $error['message'],
+                $error['file'],
+                $error['line'],
+            ), $errors);
+
+            $this->fail("Errors found:\n" . implode("\n", $errorMessages));
+        }
+    }
 }