EntityTypeInfoTest.php

<?php

namespace Drupal\Tests\content_moderation\Kernel;

use Drupal\content_moderation\Entity\Handler\ModerationHandler;
use Drupal\content_moderation\EntityTypeInfo;
use Drupal\entity_test\Entity\EntityTestBundle;
use Drupal\KernelTests\KernelTestBase;
use Drupal\Tests\content_moderation\Traits\ContentModerationTestTrait;

/**
 * @coversDefaultClass \Drupal\content_moderation\EntityTypeInfo
 *
 * @group content_moderation
 */
class EntityTypeInfoTest extends KernelTestBase {
    use ContentModerationTestTrait;
    
    /**
     * Modules to enable.
     *
     * @var array
     */
    protected static $modules = [
        'content_moderation',
        'workflows',
        'entity_test',
    ];
    
    /**
     * The entity type manager.
     *
     * @var \Drupal\Core\Entity\EntityTypeManagerInterface
     */
    protected $entityTypeManager;
    
    /**
     * The entity type info class.
     *
     * @var \Drupal\content_moderation\EntityTypeInfo
     */
    protected $entityTypeInfo;
    
    /**
     * The entity field manager.
     *
     * @var \Drupal\Core\Entity\EntityFieldManagerInterface
     */
    protected $entityFieldManager;
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->entityTypeInfo = $this->container
            ->get('class_resolver')
            ->getInstanceFromDefinition(EntityTypeInfo::class);
        $this->entityTypeManager = $this->container
            ->get('entity_type.manager');
        $this->entityFieldManager = $this->container
            ->get('entity_field.manager');
        $this->installConfig([
            'content_moderation',
        ]);
    }
    
    /**
     * @covers ::entityBaseFieldInfo
     */
    public function testEntityBaseFieldInfo() {
        $definition = $this->entityTypeManager
            ->getDefinition('entity_test');
        $definition->setHandlerClass('moderation', ModerationHandler::class);
        $this->enableModeration('entity_test', 'entity_test');
        $base_fields = $this->entityTypeInfo
            ->entityBaseFieldInfo($definition);
        $this->assertFalse($base_fields['moderation_state']->isReadOnly());
        $this->assertTrue($base_fields['moderation_state']->isComputed());
        $this->assertTrue($base_fields['moderation_state']->isTranslatable());
    }
    
    /**
     * Tests the correct entity types have moderation added.
     *
     * @covers ::entityTypeAlter
     *
     * @dataProvider providerTestEntityTypeAlter
     */
    public function testEntityTypeAlter($entity_type_id, $moderatable) {
        $entity_types = $this->entityTypeManager
            ->getDefinitions();
        $this->assertSame($moderatable, $entity_types[$entity_type_id]->hasHandlerClass('moderation'));
    }
    
    /**
     * Provides test data for testEntityTypeAlter().
     *
     * @return array
     *   An array of test cases, where each test case is an array with the
     *   following values:
     *   - An entity type ID.
     *   - Whether the entity type is moderatable or not.
     */
    public function providerTestEntityTypeAlter() {
        $tests = [];
        $tests['non_internal_non_revisionable'] = [
            'entity_test',
            FALSE,
        ];
        $tests['non_internal_revisionable'] = [
            'entity_test_rev',
            TRUE,
        ];
        $tests['internal_non_revisionable'] = [
            'entity_test_no_label',
            FALSE,
        ];
        $tests['internal_revisionable'] = [
            'content_moderation_state',
            FALSE,
        ];
        return $tests;
    }
    
    /**
     * @covers ::entityBaseFieldInfo
     */
    public function testBaseFieldOnlyAddedToModeratedEntityTypes() {
        $definition = $this->entityTypeManager
            ->getDefinition('entity_test_with_bundle');
        EntityTestBundle::create([
            'id' => 'moderated',
        ])->save();
        EntityTestBundle::create([
            'id' => 'unmoderated',
        ])->save();
        $base_fields = $this->entityTypeInfo
            ->entityBaseFieldInfo($definition);
        $this->assertFalse(isset($base_fields['moderation_state']));
        $this->enableModeration('entity_test_with_bundle', 'moderated');
        $base_fields = $this->entityTypeInfo
            ->entityBaseFieldInfo($definition);
        $this->assertTrue(isset($base_fields['moderation_state']));
    }
    
    /**
     * Tests entity base field provider.
     */
    public function testEntityBaseFieldProvider() {
        $this->enableModeration('entity_test_mulrev', 'entity_test_mulrev');
        $this->container
            ->get('state')
            ->set('entity_test.field_test_item', TRUE);
        $field_definitions = $this->entityFieldManager
            ->getFieldDefinitions('entity_test_mulrev', 'entity_test_mulrev');
        $this->assertEquals('entity_test', $field_definitions['field_test_item']->getProvider());
    }
    
    /**
     * Add moderation to an entity type and bundle.
     */
    protected function enableModeration($entity_type_id, $bundle) {
        $workflow = $this->createEditorialWorkflow();
        $workflow->getTypePlugin()
            ->addEntityTypeAndBundle($entity_type_id, $bundle);
        $workflow->save();
    }

}