function EntityViewBuilderTest::testHookEntityTypeViewModeAlter

Tests a view mode alter on an entity.

File

core/tests/Drupal/KernelTests/Core/Entity/EntityViewBuilderTest.php, line 316

Class

EntityViewBuilderTest

Tests the entity view builder.

Namespace

Code

public function testHookEntityTypeViewModeAlter() : void {
  $entity_ids = [];
  // Create some entities to test.
  for ($i = 0; $i < 5; $i++) {
    $entity = $this->createTestEntity('entity_test');
    $entity->save();
    $entity_ids[] = $entity->id();
  }
  /** @var \Drupal\entity_test\EntityTestViewBuilder $view_builder */
  $view_builder = $this->container
    ->get('entity_type.manager')
    ->getViewBuilder('entity_test');
  /** @var \Drupal\Core\Entity\EntityStorageInterface $storage */
  $storage = $this->container
    ->get('entity_type.manager')
    ->getStorage('entity_test');
  $storage->resetCache();
  $entities = $storage->loadMultiple($entity_ids);
  $build = $view_builder->viewMultiple($entities, 'entity_test.vm_alter_test');
  foreach ($build as $key => $entity_build) {
    if (!is_numeric($key)) {
      continue;
    }
    $this->assertArrayHasKey('#view_mode', $entity_build);
    $this->assertEquals('entity_test.vm_alter_full', $entity_build['#view_mode']);
  }
}