public function EntityUnitTest::testLoadMultiple in Drupal 9

Same name and namespace in other branches

1. 8 core/tests/Drupal/Tests/Core/Entity/EntityUnitTest.php \Drupal\Tests\Core\Entity\EntityUnitTest::testLoadMultiple()

@covers ::loadMultiple

Tests Entity::loadMultiple() when called statically on a subclass of Entity.

File

core/tests/Drupal/Tests/Core/Entity/EntityUnitTest.php, line 282

Class

EntityUnitTest

@coversDefaultClass \Drupal\Core\Entity\EntityBase @group Entity @group Access

Namespace

Drupal\Tests\Core\Entity

Code

public function testLoadMultiple() {
  $this
    ->setupTestLoad();
  $class_name = get_class($this->entity);
  $entity_type_repository = $this
    ->getMockForAbstractClass(EntityTypeRepositoryInterface::class);
  $entity_type_repository
    ->expects($this
    ->once())
    ->method('getEntityTypeFromClass')
    ->with($class_name)
    ->willReturn($this->entityTypeId);
  $storage = $this
    ->createMock(EntityStorageInterface::class);
  $storage
    ->expects($this
    ->once())
    ->method('loadMultiple')
    ->with([
    1,
  ])
    ->will($this
    ->returnValue([
    1 => $this->entity,
  ]));
  $this->entityTypeManager
    ->expects($this
    ->once())
    ->method('getStorage')
    ->with($this->entityTypeId)
    ->will($this
    ->returnValue($storage));
  \Drupal::getContainer()
    ->set('entity_type.repository', $entity_type_repository);

  // Call Entity::loadMultiple statically and check that it returns the mock
  // entity.
  $this
    ->assertSame([
    1 => $this->entity,
  ], $class_name::loadMultiple([
    1,
  ]));
}