You are here

EntityBundleFieldTest.php in Drupal 10

File

core/tests/Drupal/KernelTests/Core/Entity/EntityBundleFieldTest.php
View source
<?php

namespace Drupal\KernelTests\Core\Entity;


/**
 * Tests adding a custom bundle field.
 *
 * @group Entity
 */
class EntityBundleFieldTest extends EntityKernelTestBase {

  /**
   * Modules to enable.
   *
   * @var array
   */
  protected static $modules = [
    'entity_schema_test',
    'entity_test_update',
  ];

  /**
   * The module handler.
   *
   * @var \Drupal\Core\Extension\ModuleHandlerInterface
   */
  protected $moduleHandler;

  /**
   * The database connection used.
   *
   * @var \Drupal\Core\Database\Connection
   */
  protected $database;

  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $this
      ->installSchema('user', [
      'users_data',
    ]);
    $this
      ->installEntitySchema('entity_test_update');
    $this->moduleHandler = $this->container
      ->get('module_handler');
    $this->database = $this->container
      ->get('database');
  }

  /**
   * Tests making use of a custom bundle field.
   */
  public function testCustomBundleFieldUsage() {
    entity_test_create_bundle('custom', NULL, 'entity_test_update');

    // Check that an entity with bundle entity_test does not have the custom
    // field.
    $storage = $this->entityTypeManager
      ->getStorage('entity_test_update');
    $entity = $storage
      ->create([
      'type' => 'entity_test_update',
    ]);
    $this
      ->assertFalse($entity
      ->hasField('custom_bundle_field'));

    // Check that the custom bundle has the defined custom field and check
    // saving and deleting of custom field data.
    $entity = $storage
      ->create([
      'type' => 'custom',
    ]);
    $this
      ->assertTrue($entity
      ->hasField('custom_bundle_field'));

    // Ensure that the field exists in the field map.
    $field_map = \Drupal::service('entity_field.manager')
      ->getFieldMap();
    $this
      ->assertEquals([
      'type' => 'string',
      'bundles' => [
        'custom' => 'custom',
      ],
    ], $field_map['entity_test_update']['custom_bundle_field']);
    $entity->custom_bundle_field->value = 'swanky';
    $entity
      ->save();
    $storage
      ->resetCache();
    $entity = $storage
      ->load($entity
      ->id());
    $this
      ->assertEquals('swanky', $entity->custom_bundle_field->value, 'Entity was saved correctly');
    $entity->custom_bundle_field->value = 'cozy';
    $entity
      ->save();
    $storage
      ->resetCache();
    $entity = $storage
      ->load($entity
      ->id());
    $this
      ->assertEquals('cozy', $entity->custom_bundle_field->value, 'Entity was updated correctly.');
    $entity
      ->delete();

    /** @var \Drupal\Core\Entity\Sql\DefaultTableMapping $table_mapping */
    $table_mapping = $storage
      ->getTableMapping();
    $table = $table_mapping
      ->getDedicatedDataTableName($entity
      ->getFieldDefinition('custom_bundle_field')
      ->getFieldStorageDefinition());
    $result = $this->database
      ->select($table, 'f')
      ->fields('f')
      ->condition('f.entity_id', $entity
      ->id())
      ->execute();
    $this
      ->assertFalse($result
      ->fetchAssoc(), 'Field data has been deleted');

    // Create another entity to test that values are marked as deleted when a
    // bundle is deleted.
    $entity = $storage
      ->create([
      'type' => 'custom',
      'custom_bundle_field' => 'new',
    ]);
    $entity
      ->save();
    entity_test_delete_bundle('custom', 'entity_test_update');
    $table = $table_mapping
      ->getDedicatedDataTableName($entity
      ->getFieldDefinition('custom_bundle_field')
      ->getFieldStorageDefinition(), TRUE);
    $result = $this->database
      ->select($table, 'f')
      ->condition('f.entity_id', $entity
      ->id())
      ->condition('deleted', 1)
      ->countQuery()
      ->execute();
    $this
      ->assertEquals(1, $result
      ->fetchField(), 'Field data has been deleted');

    // Ensure that the field no longer exists in the field map.
    $field_map = \Drupal::service('entity_field.manager')
      ->getFieldMap();
    $this
      ->assertFalse(isset($field_map['entity_test_update']['custom_bundle_field']));

    // Purge field data, and check that the storage definition has been
    // completely removed once the data is purged.
    field_purge_batch(10);
    $this
      ->assertFalse($this->database
      ->schema()
      ->tableExists($table), 'Custom field table was deleted');
  }

}

Classes

Namesort descending Description
EntityBundleFieldTest Tests adding a custom bundle field.