InlineBlockUsageTest.php

<?php

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

use Drupal\Core\Database\Connection;
use Drupal\entity_test\Entity\EntityTest;
use Drupal\KernelTests\KernelTestBase;
use Drupal\layout_builder\InlineBlockUsageInterface;

/**
 * Class for testing the InlineBlockUsage service.
 *
 * @coversDefaultClass \Drupal\layout_builder\InlineBlockUsage
 *
 * @group layout_builder
 */
class InlineBlockUsageTest extends KernelTestBase {
  
  /**
   * {@inheritdoc}
   */
  protected static $modules = [
    'layout_discovery',
    'layout_builder',
    'entity_test',
    'user',
  ];
  
  /**
   * The database connection.
   */
  protected Connection $database;
  
  /**
   * The inline block usage service.
   */
  protected InlineBlockUsageInterface $inlineBlockUsage;
  
  /**
   * The entity for testing.
   */
  protected EntityTest $entity;
  protected function setUp() : void {
    parent::setUp();
    $this->database = $this->container
      ->get('database');
    $this->inlineBlockUsage = $this->container
      ->get('inline_block.usage');
    $this->installSchema('layout_builder', [
      'inline_block_usage',
    ]);
    entity_test_create_bundle('bundle_with_extra_fields');
    $this->installEntitySchema('entity_test');
    $this->entity = EntityTest::create();
    $this->entity
      ->save();
  }
  
  /**
   * Covers ::addUsage.
   */
  public function testAddUsage() : void {
    $this->inlineBlockUsage
      ->addUsage('1', $this->entity);
    $results = $this->database
      ->select('inline_block_usage')
      ->fields('inline_block_usage')
      ->condition('block_content_id', 1)
      ->condition('layout_entity_id', $this->entity
      ->id())
      ->condition('layout_entity_type', $this->entity
      ->getEntityTypeId())
      ->execute()
      ->fetchAll();
    $this->assertCount(1, $results);
  }
  
  /**
   * Covers ::getUnused.
   */
  public function testGetUnused() : void {
    // Add a valid usage.
    $this->inlineBlockUsage
      ->addUsage('1', $this->entity);
    $this->assertEmpty($this->inlineBlockUsage
      ->getUnused());
    // Add an invalid usage.
    $this->database
      ->merge('inline_block_usage')
      ->keys([
      'block_content_id' => 2,
      'layout_entity_id' => NULL,
      'layout_entity_type' => NULL,
    ])
      ->execute();
    $this->assertCount(1, $this->inlineBlockUsage
      ->getUnused());
  }
  
  /**
   * Covers ::removeByLayoutEntity.
   */
  public function testRemoveByLayoutEntity() : void {
    $this->inlineBlockUsage
      ->addUsage('1', $this->entity);
    $this->inlineBlockUsage
      ->removeByLayoutEntity($this->entity);
    $results = $this->database
      ->select('inline_block_usage')
      ->fields('inline_block_usage')
      ->condition('block_content_id', '1')
      ->isNull('layout_entity_id')
      ->isNull('layout_entity_type')
      ->execute()
      ->fetchAll();
    $this->assertCount(1, $results);
  }
  
  /**
   * Covers ::deleteUsage.
   */
  public function testDeleteUsage() : void {
    $this->inlineBlockUsage
      ->addUsage('1', $this->entity);
    $this->inlineBlockUsage
      ->deleteUsage([
      '1',
    ]);
    $results = $this->database
      ->select('inline_block_usage')
      ->fields('inline_block_usage')
      ->condition('block_content_id', 1)
      ->condition('layout_entity_id', $this->entity
      ->id())
      ->condition('layout_entity_type', $this->entity
      ->getEntityTypeId())
      ->execute()
      ->fetchAll();
    $this->assertEmpty($results);
  }
  
  /**
   * Covers ::getUsage.
   */
  public function testGetUsage() : void {
    $this->inlineBlockUsage
      ->addUsage('1', $this->entity);
    $result = $this->inlineBlockUsage
      ->getUsage('1');
    $this->assertEquals($this->entity
      ->id(), $result->layout_entity_id);
    $this->assertEquals($this->entity
      ->getEntityTypeId(), $result->layout_entity_type);
  }

}