ContentEntityStorageBaseTest.php

<?php

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

use Drupal\Core\Entity\ContentEntityStorageBase;
use Drupal\KernelTests\KernelTestBase;
use PHPUnit\Framework\Attributes\CoversClass;
use PHPUnit\Framework\Attributes\DataProvider;
use PHPUnit\Framework\Attributes\Group;

/**
 * Tests Drupal\Core\Entity\ContentEntityStorageBase.
 */
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');
  }
  
  /**
   * Tests create.
   *
   * @legacy-covers ::create
   */
  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',
      ],
    ];
  }
  
  /**
   * Tests re create.
   *
   * @legacy-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());
  }

}