function LayoutEntityHelperTraitTest::testGetEntitySections

@covers ::getEntitySections

File

core/modules/layout_builder/tests/src/Kernel/LayoutEntityHelperTraitTest.php, line 224

Class

LayoutEntityHelperTraitTest

@coversDefaultClass \Drupal\layout_builder\LayoutEntityHelperTrait

Namespace

Code

public function testGetEntitySections() : void {
    $entity = EntityTest::create([
        'name' => 'updated',
    ]);
    $section_storage_manager = $this->prophesize(SectionStorageManagerInterface::class);
    $section_storage_manager->load('')
        ->willReturn(NULL);
    $section_storage = $this->prophesize(SectionStorageInterface::class);
    $sections = [
        new Section('layout_onecol'),
    ];
    $this->assertCount(1, $sections);
    $section_storage->getSections()
        ->willReturn($sections);
    $section_storage->count()
        ->willReturn(1);
    $section_storage_manager->findByContext(Argument::cetera())
        ->willReturn($section_storage->reveal());
    $this->container
        ->set('plugin.manager.layout_builder.section_storage', $section_storage_manager->reveal());
    $class = new TestLayoutEntityHelperTrait();
    // Ensure that if the entity has a section storage the sections will be
    // returned.
    $this->assertSame($sections, $class->getEntitySections($entity));
    $section_storage_manager->findByContext(Argument::cetera())
        ->willReturn(NULL);
    $this->container
        ->set('plugin.manager.layout_builder.section_storage', $section_storage_manager->reveal());
    // Ensure that if the entity has no section storage an empty array will be
    // returned.
    $this->assertSame([], $class->getEntitySections($entity));
}