EntityBaseTest.php

<?php

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

use Drupal\Core\Entity\Plugin\DataType\ConfigEntityAdapter;
use Drupal\Core\Entity\Plugin\DataType\EntityAdapter;
use Drupal\entity_test\Entity\EntityTestBundle;
use Drupal\entity_test\Entity\EntityTestWithBundle;

/**
 * Tests the functionality provided by \Drupal\Core\Entity\EntityBase.
 *
 * @coversDefaultClass \Drupal\Core\Entity\EntityBase
 * @group Entity
 */
class EntityBaseTest extends EntityKernelTestBase {
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->installEntitySchema('entity_test_with_bundle');
    }
    
    /**
     * Tests that the correct entity adapter is returned.
     *
     * @covers ::getTypedData
     */
    public function testGetTypedData() : void {
        $bundle = EntityTestBundle::create([
            'id' => $this->randomMachineName(),
        ]);
        $bundle->save();
        $entity = EntityTestWithBundle::create([
            'type' => $bundle->id(),
            'name' => $this->randomString(),
        ]);
        $entity->save();
        $this->assertInstanceOf(ConfigEntityAdapter::class, $bundle->getTypedData());
        $this->assertInstanceOf(EntityAdapter::class, $entity->getTypedData());
    }

}