TypedDataDefinitionEntityBundleTest.php

<?php

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

use Drupal\Core\Entity\EntityStorageInterface;
use Drupal\Core\Entity\EntityTypeManagerInterface;
use Drupal\Core\TypedData\TypedDataManagerInterface;
use Drupal\KernelTests\KernelTestBase;

/**
 * Tests data type metadata for entity bundles.
 *
 * @group TypedData
 */
class TypedDataDefinitionEntityBundleTest extends KernelTestBase {
    
    /**
     * The typed data manager to use.
     *
     * @var \Drupal\Core\TypedData\TypedDataManager
     */
    protected TypedDataManagerInterface $typedDataManager;
    
    /**
     * The storage for the 'entity_test_bundle' entity type.
     *
     * @var \Drupal\Core\Entity\EntityStorageInterface
     */
    protected EntityStorageInterface $entityTestBundleStorage;
    
    /**
     * {@inheritdoc}
     */
    protected static $modules = [
        'entity_test',
        'user',
    ];
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->typedDataManager = $this->container
            ->get('typed_data_manager');
        $entityTypeManager = $this->container
            ->get('entity_type.manager');
        assert($entityTypeManager instanceof EntityTypeManagerInterface);
        $this->entityTestBundleStorage = $entityTypeManager->getStorage('entity_test_bundle');
    }
    
    /**
     * Tests that entity bundle definitions are derived correctly.
     *
     * @throws \Drupal\Core\Entity\EntityStorageException
     */
    public function testEntityBundleDefinitions() : void {
        // Start without any bundles and no bundle-specific data type.
        $dataTypes = array_keys($this->typedDataManager
            ->getDefinitions());
        $this->assertContains('entity:entity_test_with_bundle', $dataTypes);
        $this->assertNotContains('entity:entity_test_with_bundle:test', $dataTypes);
        // Add a bundle and make sure the bundle-specific data type is registered.
        $bundle = $this->entityTestBundleStorage
            ->create([
            'id' => 'test',
        ]);
        $bundle->save();
        $dataTypes = array_keys($this->typedDataManager
            ->getDefinitions());
        $this->assertContains('entity:entity_test_with_bundle', $dataTypes);
        $this->assertContains('entity:entity_test_with_bundle:test', $dataTypes);
        // Delete the bundle and make sure the bundle-specific data type is no
        // longer returned.
        $bundle->delete();
        $dataTypes = array_keys($this->typedDataManager
            ->getDefinitions());
        $this->assertContains('entity:entity_test_with_bundle', $dataTypes);
        $this->assertNotContains('entity:entity_test_with_bundle:test', $dataTypes);
    }

}