MergeTest.php in Zircon Profile 8.0
Same filename and directory in other branches
Contains \Drupal\system\Tests\Database\MergeTest.
Namespace
Drupal\system\Tests\DatabaseFile
core/modules/system/src/Tests/Database/MergeTest.phpView source
<?php
/**
* @file
* Contains \Drupal\system\Tests\Database\MergeTest.
*/
namespace Drupal\system\Tests\Database;
use Drupal\Core\Database\Query\Merge;
use Drupal\Core\Database\Query\InvalidMergeQueryException;
/**
* Tests the MERGE query builder.
*
* @group Database
*/
class MergeTest extends DatabaseTestBase {
/**
* Confirms that we can merge-insert a record successfully.
*/
function testMergeInsert() {
$num_records_before = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$result = db_merge('test_people')
->key('job', 'Presenter')
->fields(array(
'age' => 31,
'name' => 'Tiffany',
))
->execute();
$this
->assertEqual($result, Merge::STATUS_INSERT, 'Insert status returned.');
$num_records_after = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$this
->assertEqual($num_records_before + 1, $num_records_after, 'Merge inserted properly.');
$person = db_query('SELECT * FROM {test_people} WHERE job = :job', array(
':job' => 'Presenter',
))
->fetch();
$this
->assertEqual($person->name, 'Tiffany', 'Name set correctly.');
$this
->assertEqual($person->age, 31, 'Age set correctly.');
$this
->assertEqual($person->job, 'Presenter', 'Job set correctly.');
}
/**
* Confirms that we can merge-update a record successfully.
*/
function testMergeUpdate() {
$num_records_before = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$result = db_merge('test_people')
->key('job', 'Speaker')
->fields(array(
'age' => 31,
'name' => 'Tiffany',
))
->execute();
$this
->assertEqual($result, Merge::STATUS_UPDATE, 'Update status returned.');
$num_records_after = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$this
->assertEqual($num_records_before, $num_records_after, 'Merge updated properly.');
$person = db_query('SELECT * FROM {test_people} WHERE job = :job', array(
':job' => 'Speaker',
))
->fetch();
$this
->assertEqual($person->name, 'Tiffany', 'Name set correctly.');
$this
->assertEqual($person->age, 31, 'Age set correctly.');
$this
->assertEqual($person->job, 'Speaker', 'Job set correctly.');
}
/**
* Confirms that we can merge-update a record successfully.
*
* This test varies from the previous test because it manually defines which
* fields are inserted, and which fields are updated.
*/
function testMergeUpdateExcept() {
$num_records_before = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
db_merge('test_people')
->key('job', 'Speaker')
->insertFields(array(
'age' => 31,
))
->updateFields(array(
'name' => 'Tiffany',
))
->execute();
$num_records_after = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$this
->assertEqual($num_records_before, $num_records_after, 'Merge updated properly.');
$person = db_query('SELECT * FROM {test_people} WHERE job = :job', array(
':job' => 'Speaker',
))
->fetch();
$this
->assertEqual($person->name, 'Tiffany', 'Name set correctly.');
$this
->assertEqual($person->age, 30, 'Age skipped correctly.');
$this
->assertEqual($person->job, 'Speaker', 'Job set correctly.');
}
/**
* Confirms that we can merge-update a record, with alternate replacement.
*/
function testMergeUpdateExplicit() {
$num_records_before = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
db_merge('test_people')
->key('job', 'Speaker')
->insertFields(array(
'age' => 31,
'name' => 'Tiffany',
))
->updateFields(array(
'name' => 'Joe',
))
->execute();
$num_records_after = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$this
->assertEqual($num_records_before, $num_records_after, 'Merge updated properly.');
$person = db_query('SELECT * FROM {test_people} WHERE job = :job', array(
':job' => 'Speaker',
))
->fetch();
$this
->assertEqual($person->name, 'Joe', 'Name set correctly.');
$this
->assertEqual($person->age, 30, 'Age skipped correctly.');
$this
->assertEqual($person->job, 'Speaker', 'Job set correctly.');
}
/**
* Confirms that we can merge-update a record successfully, with expressions.
*/
function testMergeUpdateExpression() {
$num_records_before = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$age_before = db_query('SELECT age FROM {test_people} WHERE job = :job', array(
':job' => 'Speaker',
))
->fetchField();
// This is a very contrived example, as I have no idea why you'd want to
// change age this way, but that's beside the point.
// Note that we are also double-setting age here, once as a literal and
// once as an expression. This test will only pass if the expression wins,
// which is what is supposed to happen.
db_merge('test_people')
->key('job', 'Speaker')
->fields(array(
'name' => 'Tiffany',
))
->insertFields(array(
'age' => 31,
))
->expression('age', 'age + :age', array(
':age' => 4,
))
->execute();
$num_records_after = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$this
->assertEqual($num_records_before, $num_records_after, 'Merge updated properly.');
$person = db_query('SELECT * FROM {test_people} WHERE job = :job', array(
':job' => 'Speaker',
))
->fetch();
$this
->assertEqual($person->name, 'Tiffany', 'Name set correctly.');
$this
->assertEqual($person->age, $age_before + 4, 'Age updated correctly.');
$this
->assertEqual($person->job, 'Speaker', 'Job set correctly.');
}
/**
* Tests that we can merge-insert without any update fields.
*/
function testMergeInsertWithoutUpdate() {
$num_records_before = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
db_merge('test_people')
->key('job', 'Presenter')
->execute();
$num_records_after = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$this
->assertEqual($num_records_before + 1, $num_records_after, 'Merge inserted properly.');
$person = db_query('SELECT * FROM {test_people} WHERE job = :job', array(
':job' => 'Presenter',
))
->fetch();
$this
->assertEqual($person->name, '', 'Name set correctly.');
$this
->assertEqual($person->age, 0, 'Age set correctly.');
$this
->assertEqual($person->job, 'Presenter', 'Job set correctly.');
}
/**
* Confirms that we can merge-update without any update fields.
*/
function testMergeUpdateWithoutUpdate() {
$num_records_before = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
db_merge('test_people')
->key('job', 'Speaker')
->execute();
$num_records_after = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$this
->assertEqual($num_records_before, $num_records_after, 'Merge skipped properly.');
$person = db_query('SELECT * FROM {test_people} WHERE job = :job', array(
':job' => 'Speaker',
))
->fetch();
$this
->assertEqual($person->name, 'Meredith', 'Name skipped correctly.');
$this
->assertEqual($person->age, 30, 'Age skipped correctly.');
$this
->assertEqual($person->job, 'Speaker', 'Job skipped correctly.');
db_merge('test_people')
->key('job', 'Speaker')
->insertFields(array(
'age' => 31,
))
->execute();
$num_records_after = db_query('SELECT COUNT(*) FROM {test_people}')
->fetchField();
$this
->assertEqual($num_records_before, $num_records_after, 'Merge skipped properly.');
$person = db_query('SELECT * FROM {test_people} WHERE job = :job', array(
':job' => 'Speaker',
))
->fetch();
$this
->assertEqual($person->name, 'Meredith', 'Name skipped correctly.');
$this
->assertEqual($person->age, 30, 'Age skipped correctly.');
$this
->assertEqual($person->job, 'Speaker', 'Job skipped correctly.');
}
/**
* Tests that an invalid merge query throws an exception.
*/
function testInvalidMerge() {
try {
// This query will fail because there is no key field specified.
// Normally it would throw an exception but we are suppressing it with
// the throw_exception option.
$options['throw_exception'] = FALSE;
db_merge('test_people', $options)
->fields(array(
'age' => 31,
'name' => 'Tiffany',
))
->execute();
$this
->pass('$options[\'throw_exception\'] is FALSE, no InvalidMergeQueryException thrown.');
} catch (InvalidMergeQueryException $e) {
$this
->fail('$options[\'throw_exception\'] is FALSE, but InvalidMergeQueryException thrown for invalid query.');
return;
}
try {
// This query will fail because there is no key field specified.
db_merge('test_people')
->fields(array(
'age' => 31,
'name' => 'Tiffany',
))
->execute();
} catch (InvalidMergeQueryException $e) {
$this
->pass('InvalidMergeQueryException thrown for invalid query.');
return;
}
$this
->fail('No InvalidMergeQueryException thrown');
}
}