diff --git a/core/modules/layout_builder/tests/src/Functional/LayoutSectionTest.php b/core/modules/layout_builder/tests/src/Functional/LayoutSectionTest.php
index 90ad29886a..4b41c9d40b 100644
--- a/core/modules/layout_builder/tests/src/Functional/LayoutSectionTest.php
+++ b/core/modules/layout_builder/tests/src/Functional/LayoutSectionTest.php
@@ -55,7 +55,7 @@ protected function setUp() {
    */
   public function providerTestLayoutSectionFormatter() {
     $data = [];
-    $data['block_with_context'] = [
+    $data['block_with_global_context'] = [
       [
         [
           'section' => new Section('layout_onecol', [], [
@@ -80,6 +80,31 @@ public function providerTestLayoutSectionFormatter() {
       'user:2',
       'UNCACHEABLE',
     ];
+    $data['block_with_entity_context'] = [
+      [
+        [
+          'section' => new Section('layout_onecol', [], [
+            'baz' => new SectionComponent('baz', 'content', [
+              'id' => 'field_block:node:body',
+              'context_mapping' => [
+                'entity' => 'layout_builder.entity',
+              ],
+            ]),
+          ]),
+        ],
+      ],
+      [
+        '.layout--onecol',
+        '.field--name-body',
+      ],
+      [
+        'Body',
+        'The node body',
+      ],
+      '',
+      '',
+      'MISS',
+    ];
     $data['single_section_single_block'] = [
       [
         [