EntityConverterTest.php

<?php

namespace Drupal\Tests\Core\ParamConverter;

use Drupal\Component\Plugin\Exception\PluginNotFoundException;
use Drupal\Core\Entity\ContentEntityInterface;
use Drupal\Core\Entity\ContentEntityStorageInterface;
use Drupal\Core\Entity\ContentEntityTypeInterface;
use Drupal\Core\Entity\EntityTypeManagerInterface;
use Drupal\Core\Entity\EntityRepositoryInterface;
use Drupal\Core\ParamConverter\EntityConverter;
use Drupal\Core\ParamConverter\ParamNotConvertedException;
use Drupal\Core\Plugin\Context\Context;
use Drupal\Core\Plugin\Context\ContextDefinition;
use Drupal\Core\Plugin\Context\ContextRepositoryInterface;
use Drupal\Core\TypedData\DataDefinition;
use Drupal\Core\TypedData\TypedDataInterface;
use Drupal\Core\TypedData\TypedDataManagerInterface;
use Drupal\Tests\UnitTestCase;
use Symfony\Component\DependencyInjection\ContainerInterface;
use Symfony\Component\Routing\Route;

/**
 * @coversDefaultClass \Drupal\Core\ParamConverter\EntityConverter
 * @group ParamConverter
 * @group Entity
 */
class EntityConverterTest extends UnitTestCase {
    
    /**
     * The mocked entity type manager.
     *
     * @var \Drupal\Core\Entity\EntityTypeManagerInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $entityTypeManager;
    
    /**
     * The mocked entities repository.
     *
     * @var \Drupal\Core\Entity\EntityRepositoryInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $entityRepository;
    
    /**
     * The tested entity converter.
     *
     * @var \Drupal\Core\ParamConverter\EntityConverter
     */
    protected $entityConverter;
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->entityTypeManager = $this->createMock(EntityTypeManagerInterface::class);
        $this->entityRepository = $this->createMock(EntityRepositoryInterface::class);
        $this->entityConverter = new EntityConverter($this->entityTypeManager, $this->entityRepository);
    }
    
    /**
     * Sets up mock services and class instances.
     *
     * @param object[] $service_map
     *   An associative array of service instances keyed by service name.
     */
    protected function setUpMocks($service_map = []) {
        $entity = $this->createMock(ContentEntityInterface::class);
        $entity->expects($this->any())
            ->method('getEntityTypeId')
            ->willReturn('entity_test');
        $entity->expects($this->any())
            ->method('id')
            ->willReturn('id');
        $entity->expects($this->any())
            ->method('isTranslatable')
            ->willReturn(FALSE);
        $entity->expects($this->any())
            ->method('getLoadedRevisionId')
            ->willReturn('revision_id');
        $storage = $this->createMock(ContentEntityStorageInterface::class);
        $storage->expects($this->any())
            ->method('load')
            ->with('id')
            ->willReturn($entity);
        $storage->expects($this->any())
            ->method('getLatestRevisionId')
            ->with('id')
            ->willReturn('revision_id');
        $this->entityTypeManager
            ->expects($this->any())
            ->method('getStorage')
            ->with('entity_test')
            ->willReturn($storage);
        $entity_type = $this->createMock(ContentEntityTypeInterface::class);
        $entity_type->expects($this->any())
            ->method('isRevisionable')
            ->willReturn(TRUE);
        $this->entityTypeManager
            ->expects($this->any())
            ->method('getDefinition')
            ->with('entity_test')
            ->willReturn($entity_type);
        $context_repository = $this->createMock(ContextRepositoryInterface::class);
        $context_repository->expects($this->any())
            ->method('getAvailableContexts')
            ->willReturn([]);
        $context_definition = $this->createMock(DataDefinition::class);
        foreach ([
            'setLabel',
            'setDescription',
            'setRequired',
            'setConstraints',
        ] as $method) {
            $context_definition->expects($this->any())
                ->method($method)
                ->willReturn($context_definition);
        }
        $context_definition->expects($this->any())
            ->method('getConstraints')
            ->willReturn([]);
        $typed_data_manager = $this->createMock(TypedDataManagerInterface::class);
        $typed_data_manager->expects($this->any())
            ->method('create')
            ->willReturn($this->createMock(TypedDataInterface::class));
        $typed_data_manager->expects($this->any())
            ->method('createDataDefinition')
            ->willReturn($context_definition);
        $service_map += [
            'context.repository' => $context_repository,
            'typed_data_manager' => $typed_data_manager,
        ];
        
        /** @var \Symfony\Component\DependencyInjection\ContainerInterface|\PHPUnit\Framework\MockObject\MockObject $container */
        $container = $this->createMock(ContainerInterface::class);
        $return_map = [];
        foreach ($service_map as $name => $service) {
            $return_map[] = [
                $name,
                1,
                $service,
            ];
        }
        $container->expects($this->any())
            ->method('get')
            ->willReturnMap($return_map);
        \Drupal::setContainer($container);
    }
    
    /**
     * Tests the applies() method.
     *
     * @dataProvider providerTestApplies
     *
     * @covers ::applies
     */
    public function testApplies(array $definition, $name, Route $route, $applies) {
        $this->entityTypeManager
            ->expects($this->any())
            ->method('hasDefinition')
            ->willReturnCallback(function ($entity_type) {
            return 'entity_test' == $entity_type;
        });
        $this->assertEquals($applies, $this->entityConverter
            ->applies($definition, $name, $route));
    }
    
    /**
     * Provides test data for testApplies()
     */
    public function providerTestApplies() {
        $data = [];
        $data[] = [
            [
                'type' => 'entity:foo',
            ],
            'foo',
            new Route('/test/{foo}/bar'),
            FALSE,
        ];
        $data[] = [
            [
                'type' => 'entity:entity_test',
            ],
            'foo',
            new Route('/test/{foo}/bar'),
            TRUE,
        ];
        $data[] = [
            [
                'type' => 'entity:entity_test',
            ],
            'entity_test',
            new Route('/test/{entity_test}/bar'),
            TRUE,
        ];
        $data[] = [
            [
                'type' => 'entity:{entity_test}',
            ],
            'entity_test',
            new Route('/test/{entity_test}/bar'),
            FALSE,
        ];
        $data[] = [
            [
                'type' => 'entity:{entity_type}',
            ],
            'entity_test',
            new Route('/test/{entity_type}/{entity_test}/bar'),
            TRUE,
        ];
        $data[] = [
            [
                'type' => 'foo',
            ],
            'entity_test',
            new Route('/test/{entity_type}/{entity_test}/bar'),
            FALSE,
        ];
        return $data;
    }
    
    /**
     * Tests the convert() method.
     *
     * @dataProvider providerTestConvert
     *
     * @covers ::convert
     */
    public function testConvert($value, array $definition, array $defaults, $expected_result) {
        $this->setUpMocks();
        $this->entityRepository
            ->expects($this->any())
            ->method('getCanonical')
            ->willReturnCallback(function ($entity_type_id, $entity_id) {
            return $entity_type_id === 'entity_test' && $entity_id === 'valid_id' ? (object) [
                'id' => 'valid_id',
            ] : NULL;
        });
        $this->assertEquals($expected_result, $this->entityConverter
            ->convert($value, $definition, 'foo', $defaults));
    }
    
    /**
     * Provides test data for testConvert.
     */
    public function providerTestConvert() {
        $data = [];
        // Existing entity type.
        $data[] = [
            'valid_id',
            [
                'type' => 'entity:entity_test',
            ],
            [
                'foo' => 'valid_id',
            ],
            (object) [
                'id' => 'valid_id',
            ],
        ];
        // Invalid ID.
        $data[] = [
            'invalid_id',
            [
                'type' => 'entity:entity_test',
            ],
            [
                'foo' => 'invalid_id',
            ],
            NULL,
        ];
        // Entity type placeholder.
        $data[] = [
            'valid_id',
            [
                'type' => 'entity:{entity_type}',
            ],
            [
                'foo' => 'valid_id',
                'entity_type' => 'entity_test',
            ],
            (object) [
                'id' => 'valid_id',
            ],
        ];
        return $data;
    }
    
    /**
     * Tests the convert() method with an invalid entity type.
     */
    public function testConvertWithInvalidEntityType() {
        $this->setUpMocks();
        $contexts = [
            EntityRepositoryInterface::CONTEXT_ID_LEGACY_CONTEXT_OPERATION => new Context(new ContextDefinition('string'), 'entity_upcast'),
        ];
        $plugin_id = 'invalid_id';
        $this->entityRepository
            ->expects($this->once())
            ->method('getCanonical')
            ->with($plugin_id, 'id', $contexts)
            ->willThrowException(new PluginNotFoundException($plugin_id));
        $this->expectException(PluginNotFoundException::class);
        $this->entityConverter
            ->convert('id', [
            'type' => 'entity:' . $plugin_id,
        ], 'foo', [
            'foo' => 'id',
        ]);
    }
    
    /**
     * Tests the convert() method with an invalid dynamic entity type.
     */
    public function testConvertWithInvalidDynamicEntityType() {
        $this->expectException(ParamNotConvertedException::class);
        $this->expectExceptionMessage('The "foo" parameter was not converted because the "invalid_id" parameter is missing.');
        $this->entityConverter
            ->convert('id', [
            'type' => 'entity:{invalid_id}',
        ], 'foo', [
            'foo' => 'id',
        ]);
    }

}