TestableSqlContentEntityStorage.php

<?php

declare (strict_types=1);
namespace Drupal\KernelTests\Core\Entity\Sql;

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

/**
 * A test subclass of SqlContentEntityStorage.
 */
class TestableSqlContentEntityStorage extends SqlContentEntityStorage {
  
  /**
   * Make some properties public to allow manual injection of dependencies.
   */
  
  /**
   * {@inheritdoc}
   */
  public $database;
  
  /**
   * {@inheritdoc}
   */
  public $entityType;
  
  /**
   * {@inheritdoc}
   */
  public $fieldStorageDefinitions;
  
  /**
   * {@inheritdoc}
   */
  protected $tableMapping;
  
  /**
   * Override the constructor to bypass the parent's constructor.
   */
  public function __construct() {
    // Do nothing to bypass parent's constructor.
  }
  
  /**
   * Sets the table mapping.
   *
   * @param \Drupal\Core\Entity\Sql\TableMappingInterface $table_mapping
   *   The dummy table mapping.
   */
  public function setTableMapping(TableMappingInterface $table_mapping) : void {
    $this->tableMapping = $table_mapping;
  }
  
  /**
   * Overrides original SqlContentEntityStorage::getTableMapping().
   *
   * {@inheritdoc}
   */
  public function getTableMapping(?array $storage_definitions = NULL) {
    return $this->tableMapping;
  }
  
  /**
   * Exposes the protected deleteFromDedicatedTables() method for testing.
   *
   * @param array $ids
   *   The array of entity IDs to delete.
   */
  public function publicDeleteFromDedicatedTables(array $ids) : void {
    $this->deleteFromDedicatedTables($ids);
  }

}