ContentEntityStorageBaseTest.php

<?php

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

use Drupal\KernelTests\KernelTestBase;

/**
 * @coversDefaultClass \Drupal\Core\Entity\ContentEntityStorageBase
 *
 * @group Entity
 */
class ContentEntityStorageBaseTest extends KernelTestBase {
    
    /**
     * {@inheritdoc}
     */
    protected static $modules = [
        'entity_test',
        'user',
    ];
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->installEntitySchema('entity_test');
        $this->installEntitySchema('user');
    }
    
    /**
     * @covers ::create
     *
     * @dataProvider providerTestCreate
     */
    public function testCreate(string|array $bundle) : void {
        $storage = $this->container
            ->get('entity_type.manager')
            ->getStorage('entity_test');
        $entity = $storage->create([
            'type' => $bundle,
        ]);
        $this->assertEquals('test_bundle', $entity->bundle());
    }
    
    /**
     * Provides test data for testCreate().
     */
    public static function providerTestCreate() : \Generator {
        (yield 'scalar' => [
            'bundle' => 'test_bundle',
        ]);
        (yield 'array keyed by delta' => [
            'bundle' => [
                0 => [
                    'value' => 'test_bundle',
                ],
            ],
        ]);
        (yield 'array keyed by main property name' => [
            'bundle' => [
                'value' => 'test_bundle',
            ],
        ]);
    }
    
    /**
     * @covers ::create
     */
    public function testReCreate() : void {
        $storage = $this->container
            ->get('entity_type.manager')
            ->getStorage('entity_test');
        $values = $storage->create([
            'type' => 'test_bundle',
        ])
            ->toArray();
        $entity = $storage->create($values);
        $this->assertEquals('test_bundle', $entity->bundle());
    }

}