yaffs.net Git - yaffs-website/blob - web/core/tests/Drupal/KernelTests/Core/Entity/EntityViewHookTest.php

   1 <?php
   2
   3 namespace Drupal\KernelTests\Core\Entity;
   4
   5 use Drupal\entity_test\Entity\EntityTest;
   6
   7 /**
   8  * Test view/render hooks for entities.
   9  *
  10  * @todo Add tests for the following hooks. https://www.drupal.org/node/2755353
  11  * hook_entity_view_display_alter()
  12  * hook_entity_prepare_view()
  13  * hook_ENTITY_TYPE_view()
  14  * hook_entity_view()
  15  * hook_ENTITY_TYPE_view_alter()
  16  * hook_entity_view_alter()
  17  *
  18  * @group Entity
  19  */
  20 class EntityViewHookTest extends EntityKernelTestBase {
  21
  22   /**
  23    * Test hook_entity_display_build_alter().
  24    */
  25   public function testHookEntityDisplayBuildAlter() {
  26     entity_test_create_bundle('display_build_alter_bundle');
  27     /** @var \Drupal\Core\Render\RendererInterface $renderer */
  28     $renderer = $this->container->get('renderer');
  29
  30     $entity_ids = [];
  31     // Create some entities to test.
  32     for ($i = 0; $i < 5; $i++) {
  33       $entity = EntityTest::create([
  34         'name' => $this->randomMachineName(),
  35         'type' => 'display_build_alter_bundle',
  36       ]);
  37       $entity->save();
  38       $entity_ids[] = $entity->id();
  39     }
  40
  41     /** @var \Drupal\entity_test\EntityTestViewBuilder $view_builder */
  42     $view_builder = $this->container->get('entity_type.manager')->getViewBuilder('entity_test');
  43
  44     /** @var \Drupal\Core\Entity\EntityStorageInterface $storage */
  45     $storage = $this->container->get('entity_type.manager')->getStorage('entity_test');
  46     $storage->resetCache();
  47     $entities = $storage->loadMultiple($entity_ids);
  48
  49     $build = $view_builder->viewMultiple($entities);
  50
  51     $output = $renderer->renderRoot($build);
  52     $this->setRawContent($output->__toString());
  53     // Confirm that the content added in
  54     // entity_test_entity_display_build_alter() appears multiple times, not
  55     // just for the final entity.
  56     foreach ($entity_ids as $id) {
  57       $this->assertText('Content added in hook_entity_display_build_alter for entity id ' . $id);
  58     }
  59   }
  60
  61 }