Issue #3190541 by tim.plunkett, godotislate, clayfreeman: LayoutBuilderIsActiveCacheContext not providing correct context value
parent
66449600c6
commit
3ea2660dd3
|
@ -4,15 +4,17 @@ namespace Drupal\layout_builder\Cache;
|
|||
|
||||
use Drupal\Core\Cache\CacheableMetadata;
|
||||
use Drupal\Core\Cache\Context\CalculatedCacheContextInterface;
|
||||
use Drupal\Core\Entity\Entity\EntityViewDisplay;
|
||||
use Drupal\Core\Entity\FieldableEntityInterface;
|
||||
use Drupal\Core\Routing\RouteMatchInterface;
|
||||
use Drupal\layout_builder\OverridesSectionStorageInterface;
|
||||
use Drupal\layout_builder\Entity\LayoutEntityDisplayInterface;
|
||||
|
||||
/**
|
||||
* Determines whether Layout Builder is active for a given entity type or not.
|
||||
*
|
||||
* Cache context ID: 'layout_builder_is_active:%entity_type_id', e.g.
|
||||
* 'layout_builder_is_active:node' (to vary by whether the Node entity type has
|
||||
* Layout Builder enabled).
|
||||
* 'layout_builder_is_active:node' (to vary by whether custom layout overrides
|
||||
* are allowed for the Node entity specified by the route parameter).
|
||||
*
|
||||
* @internal
|
||||
* Tagged services are internal.
|
||||
|
@ -81,8 +83,14 @@ class LayoutBuilderIsActiveCacheContext implements CalculatedCacheContextInterfa
|
|||
*/
|
||||
protected function getDisplay($entity_type_id) {
|
||||
if ($entity = $this->routeMatch->getParameter($entity_type_id)) {
|
||||
if ($entity instanceof OverridesSectionStorageInterface) {
|
||||
return $entity->getDefaultSectionStorage();
|
||||
if ($entity instanceof FieldableEntityInterface) {
|
||||
// @todo Expand to work for all view modes in
|
||||
// https://www.drupal.org/node/2907413.
|
||||
$view_mode = 'full';
|
||||
$display = EntityViewDisplay::collectRenderDisplay($entity, $view_mode);
|
||||
if ($display instanceof LayoutEntityDisplayInterface) {
|
||||
return $display;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -0,0 +1,110 @@
|
|||
<?php
|
||||
|
||||
namespace Drupal\Tests\layout_builder\Unit;
|
||||
|
||||
use Drupal\Core\DependencyInjection\ContainerBuilder;
|
||||
use Drupal\Core\Entity\EntityStorageInterface;
|
||||
use Drupal\Core\Entity\EntityTypeManagerInterface;
|
||||
use Drupal\Core\Entity\FieldableEntityInterface;
|
||||
use Drupal\Core\Entity\Query\QueryInterface;
|
||||
use Drupal\Core\Extension\ModuleHandlerInterface;
|
||||
use Drupal\Core\Routing\RouteMatchInterface;
|
||||
use Drupal\layout_builder\Cache\LayoutBuilderIsActiveCacheContext;
|
||||
use Drupal\layout_builder\Entity\LayoutEntityDisplayInterface;
|
||||
use Drupal\Tests\UnitTestCase;
|
||||
use Prophecy\Argument;
|
||||
|
||||
/**
|
||||
* @coversDefaultClass \Drupal\layout_builder\Cache\LayoutBuilderIsActiveCacheContext
|
||||
*
|
||||
* @group layout_builder
|
||||
*/
|
||||
class LayoutBuilderIsActiveCacheContextTest extends UnitTestCase {
|
||||
|
||||
/**
|
||||
* @covers ::getContext
|
||||
*/
|
||||
public function testGetContextMissingEntityTypeId() {
|
||||
$route_match = $this->prophesize(RouteMatchInterface::class);
|
||||
$cache_context = new LayoutBuilderIsActiveCacheContext($route_match->reveal());
|
||||
$this->expectException(\LogicException::class);
|
||||
$this->expectExceptionMessage('Missing entity type ID');
|
||||
$cache_context->getContext();
|
||||
}
|
||||
|
||||
/**
|
||||
* @covers ::getContext
|
||||
* @covers ::getDisplay
|
||||
*/
|
||||
public function testGetContextNonFieldableEntity() {
|
||||
$route_match = $this->prophesize(RouteMatchInterface::class);
|
||||
$route_match->getParameter('not_a_fieldable_entity')->willReturn('something that is not a fieldable entity');
|
||||
|
||||
$cache_context = new LayoutBuilderIsActiveCacheContext($route_match->reveal());
|
||||
$expected = '0';
|
||||
$actual = $cache_context->getContext('not_a_fieldable_entity');
|
||||
$this->assertSame($expected, $actual);
|
||||
}
|
||||
|
||||
/**
|
||||
* @covers ::getContext
|
||||
* @covers ::getDisplay
|
||||
*
|
||||
* @dataProvider providerTestGetContext
|
||||
*/
|
||||
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);
|
||||
}
|
||||
|
||||
/**
|
||||
* Provides test data for ::testGetContext().
|
||||
*/
|
||||
public function providerTestGetContext() {
|
||||
$data = [];
|
||||
$data['overridden'] = [
|
||||
TRUE,
|
||||
'1',
|
||||
];
|
||||
$data['not overridden'] = [
|
||||
FALSE,
|
||||
'0',
|
||||
];
|
||||
return $data;
|
||||
}
|
||||
|
||||
}
|
Loading…
Reference in New Issue