EntityBundleConditionTest.php

<?php

namespace Drupal\KernelTests\Core\Entity;

use Drupal\entity_test\Entity\EntityTestBundle;
use Drupal\entity_test\Entity\EntityTestWithBundle;

/**
 * Tests that entity bundle conditions works properly.
 *
 * @group Entity
 */
class EntityBundleConditionTest extends EntityKernelTestBase {
  
  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $this->installEntitySchema('entity_test_with_bundle');
    // Create the entity bundles required for testing.
    $bundle = EntityTestBundle::create([
      'id' => 'page',
      'label' => 'page',
    ]);
    $bundle->save();
    $bundle = EntityTestBundle::create([
      'id' => 'article',
      'label' => 'article',
    ]);
    $bundle->save();
    $bundle = EntityTestBundle::create([
      'id' => 'test',
      'label' => 'test',
    ]);
    $bundle->save();
  }
  
  /**
   * Tests conditions.
   */
  public function testConditions() {
    $this->createUser();
    // Get some entities of various bundles to check against.
    $page = EntityTestWithBundle::create([
      'type' => 'page',
      'name' => $this->randomMachineName(),
    ]);
    $page->save();
    $article = EntityTestWithBundle::create([
      'type' => 'article',
      'name' => $this->randomMachineName(),
    ]);
    $article->save();
    $test = EntityTestWithBundle::create([
      'type' => 'test',
      'name' => $this->randomMachineName(),
    ]);
    $test->save();
    // Grab the bundle condition and configure it to check against bundle of
    // 'article' and set the context to the page type entity.
    /** @var \Drupal\Core\Entity\Plugin\Condition\EntityBundle $condition */
    $condition = $this->container
      ->get('plugin.manager.condition')
      ->createInstance('entity_bundle:entity_test_with_bundle')
      ->setConfig('bundles', [
      'article' => 'article',
    ])
      ->setContextValue('entity_test_with_bundle', $page);
    $this->assertFalse($condition->execute(), 'Page type entities fail bundle checks for articles.');
    // Check for the proper summary.
    $this->assertEquals('Test entity bundle is article', $condition->summary());
    $this->assertEquals('entity_test', $condition->getPluginDefinition()['provider']);
    // Set the bundle check to page.
    $condition->setConfig('bundles', [
      'page' => 'page',
    ]);
    $this->assertTrue($condition->execute(), 'Page type entities pass bundle checks for pages');
    // Check for the proper summary.
    $this->assertEquals('Test entity bundle is page', $condition->summary());
    // Set the bundle check to page or article.
    $condition->setConfig('bundles', [
      'page' => 'page',
      'article' => 'article',
    ]);
    $this->assertTrue($condition->execute(), 'Page type entities pass bundle checks for pages or articles');
    // Check for the proper summary.
    $this->assertEquals('Test entity bundle is page or article', $condition->summary());
    // Set the context to the article entity.
    $condition->setContextValue('entity_test_with_bundle', $article);
    $this->assertTrue($condition->execute(), 'Article type entities pass bundle checks for pages or articles');
    // Set the context to the test entity.
    $condition->setContextValue('entity_test_with_bundle', $test);
    $this->assertFalse($condition->execute(), 'Test type entities pass bundle checks for pages or articles');
    // Check a greater than 2 bundles summary scenario.
    $condition->setConfig('bundles', [
      'page' => 'page',
      'article' => 'article',
      'test' => 'test',
    ]);
    $this->assertEquals('Test entity bundle is page, article or test', $condition->summary());
  }

}