Add updateBatch test

Author: sclubricants

tests/system/Database/Live/UpdateTest.php

@@ -12,6 +12,7 @@
 namespace CodeIgniter\Database\Live;
 
 use CodeIgniter\Database\Exceptions\DatabaseException;
+use CodeIgniter\Database\Forge;
 use CodeIgniter\Database\RawSql;
 use CodeIgniter\Test\CIUnitTestCase;
 use CodeIgniter\Test\DatabaseTestTrait;
@@ -27,6 +28,11 @@ final class UpdateTest extends CIUnitTestCase
 {
     use DatabaseTestTrait;
 
+    /**
+     * @var Forge|mixed
+     */
+    public $forge;
+
     protected $refresh = true;
     protected $seed    = CITestSeeder::class;
 
@@ -496,4 +502,89 @@ public function testNoConstraintFound()
         $this->db->table('job')
             ->updateBatch($jobData);
     }
+
+    public function testUpdateBatchWithQuery()
+    {
+        $this->forge = Database::forge($this->DBGroup);
+
+        $this->forge->dropTable('user2', true);
+
+        $this->forge->addField([
+            'id'          => ['type' => 'INTEGER', 'constraint' => 3, 'auto_increment' => true],
+            'name'        => ['type' => 'VARCHAR', 'constraint' => 80],
+            'email'       => ['type' => 'VARCHAR', 'constraint' => 100],
+            'country'     => ['type' => 'VARCHAR', 'constraint' => 40],
+            'created_at'  => ['type' => 'DATETIME', 'null' => true],
+            'updated_at'  => ['type' => 'DATETIME', 'null' => true],
+            'deleted_at'  => ['type' => 'DATETIME', 'null' => true],
+            'last_loggin' => ['type' => 'DATETIME', 'null' => true],
+        ])->addKey('id', true)->addUniqueKey('email')->addKey('country')->createTable('user2', true);
+
+        $data = [
+            [
+                'name'    => 'Derek Jones user2',
+                'email'   => '[email protected]',
+                'country' => 'France',
+            ],
+            [
+                'name'    => 'Ahmadinejad user2',
+                'email'   => '[email protected]',
+                'country' => 'Greece',
+            ],
+            [
+                'name'    => 'Richard A Causey user2',
+                'email'   => '[email protected]',
+                'country' => 'France',
+            ],
+            [
+                'name'    => 'Chris Martin user2',
+                'email'   => '[email protected]',
+                'country' => 'Greece',
+            ],
+            [
+                'name'    => 'New User user2',
+                'email'   => '[email protected]',
+                'country' => 'US',
+            ],
+            [
+                'name'    => 'New User2 user2',
+                'email'   => '[email protected]',
+                'country' => 'US',
+            ],
+        ];
+        $this->db->table('user2')->insertBatch($data);
+
+        if ($this->db->DBDriver === 'SQLite3' && ! (version_compare($this->db->getVersion(), '3.33.0') >= 0)) {
+            $this->markTestSkipped('Only SQLite 3.33 and newer can complete this test.');
+        }
+
+        $updateFields = ['country', 'updated_at' => new RawSql('CURRENT_TIMESTAMP')];
+
+        $subQuery = $this->db->table('user2')
+            ->select('email, country')
+            ->where('country', 'France');
+
+        $affectedRows = $this->db->table('user')
+            ->updateFields($updateFields, true)
+            ->updateBatch($subQuery, 'email');
+
+        $this->assertSame(2, (int) $affectedRows);
+
+        $this->seeInDatabase('user', ['name' => 'Derek Jones', 'country' => 'France']);
+        $this->seeInDatabase('user', ['name' => 'Ahmadinejad', 'country' => 'Iran']);
+        $this->seeInDatabase('user', ['name' => 'Richard A Causey', 'country' => 'France']);
+        $this->seeInDatabase('user', ['name' => 'Chris Martin', 'country' => 'UK']);
+
+        $result = $this->db->table('user')->get()->getResultArray();
+
+        foreach ($result as $row) {
+            if ($row['email'] === '[email protected]' || $row['email'] === '[email protected]') {
+                $this->assertNotNull($row['updated_at']);
+            } else {
+                $this->assertNull($row['updated_at']);
+            }
+        }
+
+        $this->forge->dropTable('user2', true);
+    }
 }