EntityTestUpdateStorage.php

<?php

declare (strict_types=1);
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);
    }

}