X-Git-Url: http://www.aleph1.co.uk/gitweb/?p=yaffs-website;a=blobdiff_plain;f=web%2Fcore%2Fmodules%2Flayout_builder%2Ftests%2Fsrc%2FUnit%2FLayoutBuilderRouteEnhancerTest.php;fp=web%2Fcore%2Fmodules%2Flayout_builder%2Ftests%2Fsrc%2FUnit%2FLayoutBuilderRouteEnhancerTest.php;h=62c4924d5b011c0133d68f0ac21074937fc00634;hp=0000000000000000000000000000000000000000;hb=af6d1fb995500ae68849458ee10d66abbdcfb252;hpb=680c79a86e3ed402f263faeac92e89fb6d9edcc0

diff --git a/web/core/modules/layout_builder/tests/src/Unit/LayoutBuilderRouteEnhancerTest.php b/web/core/modules/layout_builder/tests/src/Unit/LayoutBuilderRouteEnhancerTest.php
new file mode 100644
index 000000000..62c4924d5
--- /dev/null
+++ b/web/core/modules/layout_builder/tests/src/Unit/LayoutBuilderRouteEnhancerTest.php
@@ -0,0 +1,40 @@
+<?php
+
+namespace Drupal\Tests\layout_builder\Unit;
+
+use Drupal\layout_builder\Routing\LayoutBuilderRouteEnhancer;
+use Drupal\Tests\UnitTestCase;
+use Symfony\Cmf\Component\Routing\RouteObjectInterface;
+use Symfony\Component\HttpFoundation\Request;
+use Symfony\Component\Routing\Route;
+
+/**
+ * @coversDefaultClass \Drupal\layout_builder\Routing\LayoutBuilderRouteEnhancer
+ * @group layout_builder
+ */
+class LayoutBuilderRouteEnhancerTest extends UnitTestCase {
+
+  /**
+   * @covers ::enhance
+   */
+  public function testEnhanceValidDefaults() {
+    $route = new Route('/the/path', [], [], ['_layout_builder' => TRUE]);
+    $route_enhancer = new LayoutBuilderRouteEnhancer();
+    $defaults = [
+      RouteObjectInterface::ROUTE_OBJECT => $route,
+    ];
+    // Ensure that the 'section_storage' key now contains the value stored for a
+    // given entity type.
+    $expected = [
+      RouteObjectInterface::ROUTE_OBJECT => $route,
+      'is_rebuilding' => TRUE,
+    ];
+    $result = $route_enhancer->enhance($defaults, new Request(['layout_is_rebuilding' => TRUE]));
+    $this->assertEquals($expected, $result);
+
+    $expected['is_rebuilding'] = FALSE;
+    $result = $route_enhancer->enhance($defaults, new Request());
+    $this->assertEquals($expected, $result);
+  }
+
+}