Add some updateBatch() tests

These are some tests I had developed but forgot to include them earlier for updateBatch. They exercise some of the same methods used with upsert.

Author: sclubricants

tests/system/Database/Live/UpdateTest.php

@@ -324,4 +324,162 @@ public function testUpdateBatchConstraintsRawSqlAndAlias()
             'country' => 'UK',
         ]);
     }
+
+    public function testUpdateBatchUpdateFieldsAndAlias()
+    {
+        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.');
+        }
+
+        $data = [
+            [
+                'email'   => '[email protected]',
+                'name'    => 'Derek Jones Does Not Change',
+                'country' => 'Greece',
+            ],
+            [
+
+                'email'   => '[email protected]',
+                'name'    => 'Ahmadinejad No change',
+                'country' => 'Greece',
+            ],
+        ];
+
+        $rawSql = new RawSql('CURRENT_TIMESTAMP');
+
+        $updateFields = ['country', 'updated_at' => $rawSql];
+
+        $this->db->table('user')->updateFields($updateFields)->onConstraint('email')->updateBatch($data);
+
+        // check to see if update_at was updated
+        $result = $this->db->table('user')
+            ->where("email IN ('[email protected]','[email protected]')")
+            ->get()
+            ->getResultArray();
+
+        foreach ($result as $row) {
+            $this->assertNotNull($row['updated_at']);
+        }
+
+        // only country and update_at should have changed
+        $this->seeInDatabase('user', ['name' => 'Derek Jones', 'country' => 'Greece']);
+        $this->seeInDatabase('user', ['name' => 'Ahmadinejad', 'country' => 'Greece']);
+
+        // Original dataset from seeder
+        $data = [
+            [
+                'name'    => 'Derek Should Change',
+                'email'   => '[email protected]',
+                'country' => 'Greece', // will update
+            ],
+            [
+                'name'    => 'Ahmadinejad', // did't change above and will not change
+                'email'   => '[email protected]',
+                'country' => 'Iran', // will not update
+            ],
+            [
+                'name'    => 'Should Not Change',
+                'email'   => '[email protected]',
+                'country' => 'Greece', // will not update
+            ],
+            [
+                'name'    => 'Should Change',
+                'email'   => '[email protected]',
+                'country' => 'UK', // will update
+            ],
+        ];
+
+        $updateFields = ['name', 'updated_at' => new RawSql('NULL')];
+
+        $esc = $this->db->escapeChar;
+
+        // contraint is email and if the updated country = the source country
+        // setting alias allows us to reference it in RawSql
+        $this->db->table('user')
+            ->updateFields($updateFields)
+            ->onConstraint(['email', new RawSql("{$esc}db_user{$esc}.{$esc}country{$esc} = {$esc}_update{$esc}.{$esc}country{$esc}")])
+            ->setAlias('_update')
+            ->updateBatch($data);
+
+        $result = $this->db->table('user')->get()->getResultArray();
+
+        foreach ($result as $row) {
+            if ($row['email'] === '[email protected]') {
+                $this->assertNotNull($row['updated_at']);
+            } else {
+                $this->assertNull($row['updated_at']);
+            }
+        }
+
+        $this->seeInDatabase('user', ['name' => 'Derek Should Change', 'country' => 'Greece']);
+        $this->seeInDatabase('user', ['name' => 'Ahmadinejad', 'country' => 'Greece']);
+        $this->seeInDatabase('user', ['name' => 'Richard A Causey', 'country' => 'US']);
+        $this->seeInDatabase('user', ['name' => 'Should Change', 'country' => 'UK']);
+    }
+
+    public function testUpdateBatchWithoutOnConstraint()
+    {
+        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.');
+        }
+
+        $data = [
+            [
+                'name'    => 'Derek Nothing', // won't update
+                'email'   => '[email protected]',
+                'country' => 'Canada',
+            ],
+            [
+                'name'    => 'Ahmadinejad',
+                'email'   => '[email protected]',
+                'country' => 'Canada',
+            ],
+            [
+                'name'    => 'Richard A Causey',
+                'email'   => '[email protected]',
+                'country' => 'Canada',
+            ],
+            [
+                'name'    => 'Chris Martin',
+                'email'   => '[email protected]',
+                'country' => 'Canada',
+            ],
+        ];
+
+        $this->db->table('user')->updateBatch($data, 'email, name', 2);
+
+        $result = $this->db->table('user')->get()->getResultArray();
+
+        foreach ($result as $row) {
+            if ($row['email'] === '[email protected]') {
+                $this->assertSame('US', $row['country']);
+            } else {
+                $this->assertSame('Canada', $row['country']);
+            }
+        }
+    }
+
+    public function testRawSqlConstraint()
+    {
+        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.');
+        }
+
+        $data = [
+            [
+                'name'    => 'Derek Jones',
+                'email'   => '[email protected]',
+                'country' => 'Germany',
+            ],
+        ];
+
+        $builder = $this->db->table('user');
+
+        $builder->setData($data, true, 'db_myalias')
+            ->updateFields('name, country')
+            ->onConstraint(new RawSql($this->db->protectIdentifiers('user.email') . ' = ' . $this->db->protectIdentifiers('myalias.email')))
+            ->updateBatch();
+
+        $this->seeInDatabase('user', ['email' => '[email protected]', 'country' => 'Germany']);
+    }
 }