test: add test for Model and Entity with cast primaryKey

Author: kenjis

tests/_support/Entity/Cast/CastUUID.php

@@ -0,0 +1,43 @@
+<?php
+
+/**
+ * This file is part of CodeIgniter 4 framework.
+ *
+ * (c) CodeIgniter Foundation <[email protected]>
+ *
+ * For the full copyright and license information, please view
+ * the LICENSE file that was distributed with this source code.
+ */
+
+namespace Tests\Support\Entity\Cast;
+
+use CodeIgniter\Entity\Cast\BaseCast;
+
+class CastUUID extends BaseCast
+{
+    /**
+     * Get
+     *
+     * @param string $binary Binary UUID
+     *
+     * @return string String UUID
+     */
+    public static function get($binary, array $params = []): string
+    {
+        $string = unpack('h*', $binary);
+
+        return preg_replace('/([0-9a-f]{8})([0-9a-f]{4})([0-9a-f]{4})([0-9a-f]{4})([0-9a-f]{12})/', '$1-$2-$3-$4-$5', $string[1]);
+    }
+
+    /**
+     * Set
+     *
+     * @param string $string String UUID
+     *
+     * @return string Binary UUID
+     */
+    public static function set($string, array $params = []): string
+    {
+        return pack('h*', str_replace('-', '', $string));
+    }
+}

tests/_support/Entity/UUID.php

@@ -0,0 +1,25 @@
+<?php
+
+/**
+ * This file is part of CodeIgniter 4 framework.
+ *
+ * (c) CodeIgniter Foundation <[email protected]>
+ *
+ * For the full copyright and license information, please view
+ * the LICENSE file that was distributed with this source code.
+ */
+
+namespace Tests\Support\Entity;
+
+use CodeIgniter\Entity\Entity;
+use Tests\Support\Entity\Cast\CastUUID;
+
+class UUID extends Entity
+{
+    protected $casts = [
+        'id' => 'uuid',
+    ];
+    protected $castHandlers = [
+        'uuid' => CastUUID::class,
+    ];
+}

tests/_support/Models/UUIDPkeyModel.php

@@ -0,0 +1,25 @@
+<?php
+
+/**
+ * This file is part of CodeIgniter 4 framework.
+ *
+ * (c) CodeIgniter Foundation <[email protected]>
+ *
+ * For the full copyright and license information, please view
+ * the LICENSE file that was distributed with this source code.
+ */
+
+namespace Tests\Support\Models;
+
+use CodeIgniter\Model;
+use Tests\Support\Entity\UUID;
+
+class UUIDPkeyModel extends Model
+{
+    protected $table            = 'uuid';
+    protected $useAutoIncrement = false;
+    protected $returnType       = UUID::class;
+    protected $allowedFields    = [
+        'value',
+    ];
+}

tests/system/Models/UpdateModelTest.php

@@ -14,12 +14,15 @@
 use CodeIgniter\Database\Exceptions\DatabaseException;
 use CodeIgniter\Database\Exceptions\DataException;
 use CodeIgniter\Entity\Entity;
+use Config\Database;
 use InvalidArgumentException;
 use stdClass;
+use Tests\Support\Entity\UUID;
 use Tests\Support\Models\EventModel;
 use Tests\Support\Models\JobModel;
 use Tests\Support\Models\SecondaryModel;
 use Tests\Support\Models\UserModel;
+use Tests\Support\Models\UUIDPkeyModel;
 use Tests\Support\Models\ValidModel;
 use Tests\Support\Models\WithoutAutoIncrementModel;
 
@@ -375,6 +378,39 @@ public function testUpdateWithEntityNoAllowedFields(): void
         $this->model->update($id, $entity);
     }
 
+    public function testUpdateEntityWithPrimaryKeyCast(): void
+    {
+        $this->createUuidTable();
+
+        $this->createModel(UUIDPkeyModel::class);
+
+        $entity        = new UUID();
+        $entity->id    = '550e8400-e29b-41d4-a716-446655440000';
+        $entity->value = 'test1';
+
+        $id     = $this->model->insert($entity);
+        $entity = $this->model->find($id);
+
+        $entity->value = 'id';
+        $result        = $this->model->save($entity);
+
+        $this->assertTrue($result);
+
+        $entity = $this->model->find($id);
+
+        $this->assertSame('id', $entity->value);
+    }
+
+    private function createUuidTable(): void
+    {
+        $forge = Database::forge($this->DBGroup);
+        $forge->dropTable('uuid');
+        $forge->addField([
+            'id'    => ['type' => 'BINARY', 'constraint' => 16],
+            'value' => ['type' => 'VARCHAR', 'constraint' => 400, 'null' => true],
+        ])->addKey('id', true)->createTable('uuid', true);
+    }
+
     public function testUseAutoIncrementSetToFalseUpdate(): void
     {
         $key = 'key';