EntityTestUpdateStorage.php in Drupal 9

Namespace

Drupal\entity_test_update

File

core/modules/system/tests/modules/entity_test_update/src/EntityTestUpdateStorage.php

<?php

namespace Drupal\entity_test_update;

use Drupal\Core\Entity\ContentEntityInterface;
use Drupal\Core\Entity\Sql\SqlContentEntityStorage;

/**
 * Helper class for entity update testing.
 *
 * @see \Drupal\KernelTests\Core\Entity\FieldableEntityDefinitionUpdateTest::testFieldableEntityTypeUpdatesErrorHandling()
 */
class EntityTestUpdateStorage extends SqlContentEntityStorage {

  /**
   * {@inheritdoc}
   */
  protected function saveToDedicatedTables(ContentEntityInterface $entity, $update = TRUE, $names = []) {

    // Simulate an error during the 'restore' process of a test entity.
    if (\Drupal::state()
      ->get('entity_test_update.throw_exception', FALSE)) {
      throw new \Exception('Peekaboo!');
    }
    parent::saveToDedicatedTables($entity, $update, $names);
  }

}

Classes

Name	Description
EntityTestUpdateStorage	Helper class for entity update testing.