LayoutTempstoreRepositoryTest.php

<?php

declare (strict_types=1);
namespace Drupal\Tests\layout_builder\Unit;

use Drupal\Core\TempStore\SharedTempStore;
use Drupal\Core\TempStore\SharedTempStoreFactory;
use Drupal\layout_builder\LayoutTempstoreRepository;
use Drupal\layout_builder\SectionStorageInterface;
use Drupal\Tests\UnitTestCase;

/**
 * @coversDefaultClass \Drupal\layout_builder\LayoutTempstoreRepository
 * @group layout_builder
 */
class LayoutTempstoreRepositoryTest extends UnitTestCase {
  
  /**
   * @covers ::get
   * @covers ::has
   */
  public function testGetEmptyTempstore() : void {
    $section_storage = $this->prophesize(SectionStorageInterface::class);
    $section_storage->getStorageType()
      ->willReturn('my_storage_type');
    $section_storage->getStorageId()
      ->willReturn('my_storage_id');
    $tempstore = $this->prophesize(SharedTempStore::class);
    $tempstore->get('my_storage_id')
      ->shouldBeCalled();
    $tempstore_factory = $this->prophesize(SharedTempStoreFactory::class);
    $tempstore_factory->get('layout_builder.section_storage.my_storage_type')
      ->willReturn($tempstore->reveal());
    $repository = new LayoutTempstoreRepository($tempstore_factory->reveal());
    $this->assertFalse($repository->has($section_storage->reveal()));
    $result = $repository->get($section_storage->reveal());
    $this->assertSame($section_storage->reveal(), $result);
  }
  
  /**
   * @covers ::get
   * @covers ::has
   */
  public function testGetLoadedTempstore() : void {
    $section_storage = $this->prophesize(SectionStorageInterface::class);
    $section_storage->getStorageType()
      ->willReturn('my_storage_type');
    $section_storage->getStorageId()
      ->willReturn('my_storage_id');
    $tempstore_section_storage = $this->prophesize(SectionStorageInterface::class);
    $tempstore = $this->prophesize(SharedTempStore::class);
    $tempstore->get('my_storage_id')
      ->willReturn([
      'section_storage' => $tempstore_section_storage->reveal(),
    ]);
    $tempstore_factory = $this->prophesize(SharedTempStoreFactory::class);
    $tempstore_factory->get('layout_builder.section_storage.my_storage_type')
      ->willReturn($tempstore->reveal());
    $repository = new LayoutTempstoreRepository($tempstore_factory->reveal());
    $this->assertTrue($repository->has($section_storage->reveal()));
    $result = $repository->get($section_storage->reveal());
    $this->assertSame($tempstore_section_storage->reveal(), $result);
    $this->assertNotSame($section_storage->reveal(), $result);
  }
  
  /**
   * @covers ::get
   */
  public function testGetInvalidEntry() : void {
    $section_storage = $this->prophesize(SectionStorageInterface::class);
    $section_storage->getStorageType()
      ->willReturn('my_storage_type');
    $section_storage->getStorageId()
      ->willReturn('my_storage_id');
    $tempstore = $this->prophesize(SharedTempStore::class);
    $tempstore->get('my_storage_id')
      ->willReturn([
      'section_storage' => 'this_is_not_an_entity',
    ]);
    $tempstore_factory = $this->prophesize(SharedTempStoreFactory::class);
    $tempstore_factory->get('layout_builder.section_storage.my_storage_type')
      ->willReturn($tempstore->reveal());
    $repository = new LayoutTempstoreRepository($tempstore_factory->reveal());
    $this->expectException(\UnexpectedValueException::class);
    $this->expectExceptionMessage('The entry with storage type "my_storage_type" and ID "my_storage_id" is invalid');
    $repository->get($section_storage->reveal());
  }

}