View source
<?php
namespace Drupal\system\Tests\Entity;
use Drupal\Core\Entity\EntityStorageException;
use Drupal\user\UserInterface;
class EntityApiTest extends EntityUnitTestBase {
protected function setUp() {
parent::setUp();
foreach (entity_test_entity_types() as $entity_type_id) {
if ($entity_type_id != 'entity_test') {
$this
->installEntitySchema($entity_type_id);
}
}
}
public function testCRUD() {
foreach (entity_test_entity_types() as $entity_type) {
$this
->assertCRUD($entity_type, $this
->createUser());
}
}
protected function assertCRUD($entity_type, UserInterface $user1) {
$entity = entity_create($entity_type, array(
'name' => 'test',
'user_id' => $user1
->id(),
));
$entity
->save();
$entity = entity_create($entity_type, array(
'name' => 'test2',
'user_id' => $user1
->id(),
));
$entity
->save();
$entity = entity_create($entity_type, array(
'name' => 'test',
'user_id' => NULL,
));
$entity
->save();
$entities = array_values(entity_load_multiple_by_properties($entity_type, array(
'name' => 'test',
)));
$this
->assertEqual($entities[0]->name->value, 'test', format_string('%entity_type: Created and loaded entity', array(
'%entity_type' => $entity_type,
)));
$this
->assertEqual($entities[1]->name->value, 'test', format_string('%entity_type: Created and loaded entity', array(
'%entity_type' => $entity_type,
)));
$loaded_entity = entity_load($entity_type, $entity
->id());
$this
->assertEqual($loaded_entity
->id(), $entity
->id(), format_string('%entity_type: Loaded a single entity by id.', array(
'%entity_type' => $entity_type,
)));
$entities = array_values(entity_load_multiple_by_properties($entity_type, array(
'name' => 'test2',
)));
$entities[0]
->delete();
$entities = array_values(entity_load_multiple_by_properties($entity_type, array(
'name' => 'test2',
)));
$this
->assertEqual($entities, array(), format_string('%entity_type: Entity deleted.', array(
'%entity_type' => $entity_type,
)));
$entities = array_values(entity_load_multiple_by_properties($entity_type, array(
'name' => 'test',
)));
$entities[0]->name->value = 'test3';
$entities[0]
->save();
$entity = entity_load($entity_type, $entities[0]
->id());
$this
->assertEqual($entity->name->value, 'test3', format_string('%entity_type: Entity updated.', array(
'%entity_type' => $entity_type,
)));
$ids = array_keys(entity_load_multiple($entity_type));
entity_delete_multiple($entity_type, $ids);
$all = entity_load_multiple($entity_type);
$this
->assertTrue(empty($all), format_string('%entity_type: Deleted all entities.', array(
'%entity_type' => $entity_type,
)));
$definition = \Drupal::entityManager()
->getDefinition($entity_type);
$this
->assertEqual(0, db_query('SELECT COUNT(*) FROM {' . $definition
->getBaseTable() . '}')
->fetchField(), 'Base table was emptied');
if ($data_table = $definition
->getDataTable()) {
$this
->assertEqual(0, db_query('SELECT COUNT(*) FROM {' . $data_table . '}')
->fetchField(), 'Data table was emptied');
}
if ($revision_table = $definition
->getRevisionTable()) {
$this
->assertEqual(0, db_query('SELECT COUNT(*) FROM {' . $revision_table . '}')
->fetchField(), 'Data table was emptied');
}
$entities = array();
$entity = entity_create($entity_type, array(
'name' => 'test',
'user_id' => $user1
->id(),
));
$entity
->save();
$entities['test'] = $entity;
$entity = entity_create($entity_type, array(
'name' => 'test2',
'user_id' => $user1
->id(),
));
$entity
->save();
$entities['test2'] = $entity;
$controller = \Drupal::entityManager()
->getStorage($entity_type);
$controller
->delete($entities);
$all = entity_load_multiple($entity_type);
$this
->assertTrue(empty($all), format_string('%entity_type: Deleted all entities.', array(
'%entity_type' => $entity_type,
)));
$definition = \Drupal::entityManager()
->getDefinition($entity_type);
$this
->assertEqual(0, db_query('SELECT COUNT(*) FROM {' . $definition
->getBaseTable() . '}')
->fetchField(), 'Base table was emptied');
if ($data_table = $definition
->getDataTable()) {
$this
->assertEqual(0, db_query('SELECT COUNT(*) FROM {' . $data_table . '}')
->fetchField(), 'Data table was emptied');
}
if ($revision_table = $definition
->getRevisionTable()) {
$this
->assertEqual(0, db_query('SELECT COUNT(*) FROM {' . $revision_table . '}')
->fetchField(), 'Data table was emptied');
}
}
public function testEntityStorageExceptionHandling() {
$entity = entity_create('entity_test', array(
'name' => 'test',
));
try {
$GLOBALS['entity_test_throw_exception'] = TRUE;
$entity
->save();
$this
->fail('Entity presave EntityStorageException thrown but not caught.');
} catch (EntityStorageException $e) {
$this
->assertEqual($e
->getcode(), 1, 'Entity presave EntityStorageException caught.');
}
$entity = entity_create('entity_test', array(
'name' => 'test2',
));
try {
unset($GLOBALS['entity_test_throw_exception']);
$entity
->save();
$this
->pass('Exception presave not thrown and not caught.');
} catch (EntityStorageException $e) {
$this
->assertNotEqual($e
->getCode(), 1, 'Entity presave EntityStorageException caught.');
}
$entity = entity_create('entity_test', array(
'name' => 'test3',
));
$entity
->save();
try {
$GLOBALS['entity_test_throw_exception'] = TRUE;
$entity
->delete();
$this
->fail('Entity predelete EntityStorageException not thrown.');
} catch (EntityStorageException $e) {
$this
->assertEqual($e
->getCode(), 2, 'Entity predelete EntityStorageException caught.');
}
unset($GLOBALS['entity_test_throw_exception']);
$entity = entity_create('entity_test', array(
'name' => 'test4',
));
$entity
->save();
try {
$entity
->delete();
$this
->pass('Entity predelete EntityStorageException not thrown and not caught.');
} catch (EntityStorageException $e) {
$this
->assertNotEqual($e
->getCode(), 2, 'Entity predelete EntityStorageException thrown.');
}
}
}