yaffs.net Git - yaffs-website/blob - web/core/tests/Drupal/KernelTests/Core/Database/UpsertTest.php

   1 <?php
   2
   3 namespace Drupal\KernelTests\Core\Database;
   4
   5 use Drupal\Core\Database\Database;
   6
   7 /**
   8  * Tests the Upsert query builder.
   9  *
  10  * @group Database
  11  */
  12 class UpsertTest extends DatabaseTestBase {
  13
  14   /**
  15    * Confirms that we can upsert (update-or-insert) records successfully.
  16    */
  17   public function testUpsert() {
  18     $connection = Database::getConnection();
  19     $num_records_before = $connection->query('SELECT COUNT(*) FROM {test_people}')->fetchField();
  20
  21     $upsert = $connection->upsert('test_people')
  22       ->key('job')
  23       ->fields(['job', 'age', 'name']);
  24
  25     // Add a new row.
  26     $upsert->values([
  27       'job' => 'Presenter',
  28       'age' => 31,
  29       'name' => 'Tiffany',
  30     ]);
  31
  32     // Update an existing row.
  33     $upsert->values([
  34       'job' => 'Speaker',
  35       // The initial age was 30.
  36       'age' => 32,
  37       'name' => 'Meredith',
  38     ]);
  39
  40     $upsert->execute();
  41
  42     $num_records_after = $connection->query('SELECT COUNT(*) FROM {test_people}')->fetchField();
  43     $this->assertEqual($num_records_before + 1, $num_records_after, 'Rows were inserted and updated properly.');
  44
  45     $person = $connection->query('SELECT * FROM {test_people} WHERE job = :job', [':job' => 'Presenter'])->fetch();
  46     $this->assertEqual($person->job, 'Presenter', 'Job set correctly.');
  47     $this->assertEqual($person->age, 31, 'Age set correctly.');
  48     $this->assertEqual($person->name, 'Tiffany', 'Name set correctly.');
  49
  50     $person = $connection->query('SELECT * FROM {test_people} WHERE job = :job', [':job' => 'Speaker'])->fetch();
  51     $this->assertEqual($person->job, 'Speaker', 'Job was not changed.');
  52     $this->assertEqual($person->age, 32, 'Age updated correctly.');
  53     $this->assertEqual($person->name, 'Meredith', 'Name was not changed.');
  54   }
  55
  56 }