function LayoutBuilderIsActiveCacheContextTest::testGetContext

@covers ::getContext @covers ::getDisplay

@dataProvider providerTestGetContext

File

core/modules/layout_builder/tests/src/Unit/LayoutBuilderIsActiveCacheContextTest.php, line 55

Class

LayoutBuilderIsActiveCacheContextTest

@coversDefaultClass <a href="/api/drupal/core%21modules%21layout_builder%21src%21Cache%21LayoutBuilderIsActiveCacheContext.php/class/LayoutBuilderIsActiveCacheContext/9" title="Determines whether Layout Builder is active for a given entity type or not." class="local">\Drupal\layout_builder\Cache\LayoutBuilderIsActiveCacheContext</a>

Namespace

Code

public function testGetContext($is_overridden, $expected) {
    $entity_display = $this->prophesize(LayoutEntityDisplayInterface::class);
    $entity_display->isOverridable()
        ->willReturn($is_overridden);
    $entity_type_id = 'a_fieldable_entity_type';
    $fieldable_entity = $this->prophesize(FieldableEntityInterface::class);
    $fieldable_entity->getEntityTypeId()
        ->willReturn($entity_type_id);
    $fieldable_entity->bundle()
        ->willReturn('the_bundle_id');
    $route_match = $this->prophesize(RouteMatchInterface::class);
    $route_match->getParameter($entity_type_id)
        ->willReturn($fieldable_entity->reveal());
    // \Drupal\Core\Entity\Entity\EntityViewDisplay::collectRenderDisplay() is a
    // static method and can not be mocked on its own. All of the expectations
    // of that method are mocked in the next code block.
    $entity_query = $this->prophesize(QueryInterface::class);
    $entity_query->condition(Argument::cetera())
        ->willReturn($entity_query);
    $entity_query->execute()
        ->willReturn([
        'a_fieldable_entity_type.the_bundle_id.full' => 'a_fieldable_entity_type.the_bundle_id.full',
    ]);
    $entity_storage = $this->prophesize(EntityStorageInterface::class);
    $entity_storage->getQuery('AND')
        ->willReturn($entity_query->reveal());
    $entity_storage->loadMultiple(Argument::type('array'))
        ->willReturn([
        'a_fieldable_entity_type.the_bundle_id.full' => $entity_display->reveal(),
    ]);
    $entity_type_manager = $this->prophesize(EntityTypeManagerInterface::class);
    $entity_type_manager->getStorage('entity_view_display')
        ->willReturn($entity_storage->reveal());
    $module_handler = $this->prophesize(ModuleHandlerInterface::class);
    $container = new ContainerBuilder();
    $container->set('entity_type.manager', $entity_type_manager->reveal());
    $container->set('module_handler', $module_handler->reveal());
    \Drupal::setContainer($container);
    $cache_context = new LayoutBuilderIsActiveCacheContext($route_match->reveal());
    $actual = $cache_context->getContext($entity_type_id);
    $this->assertSame($expected, $actual);
}