function EntityResolverManagerTest::setupEntityTypes

Creates the entity type manager mock returning entity type objects.

File

core/tests/Drupal/Tests/Core/Entity/EntityResolverManagerTest.php, line 443

Class

EntityResolverManagerTest

@coversDefaultClass <a href="/api/drupal/core%21lib%21Drupal%21Core%21Entity%21EntityResolverManager.php/class/EntityResolverManager/9" title="Sets the entity route parameter converter options automatically." class="local">\Drupal\Core\Entity\EntityResolverManager</a> @group Entity

Namespace

Code

protected function setupEntityTypes() {
    $definition = $this->createMock('Drupal\\Core\\Entity\\EntityTypeInterface');
    $definition->expects($this->any())
        ->method('getClass')
        ->willReturn('Drupal\\Tests\\Core\\Entity\\SimpleTestEntity');
    $definition->expects($this->any())
        ->method('isRevisionable')
        ->willReturn(FALSE);
    $revisionable_definition = $this->createMock('Drupal\\Core\\Entity\\EntityTypeInterface');
    $revisionable_definition->expects($this->any())
        ->method('getClass')
        ->willReturn('Drupal\\Tests\\Core\\Entity\\SimpleTestEntity');
    $revisionable_definition->expects($this->any())
        ->method('isRevisionable')
        ->willReturn(TRUE);
    $this->entityTypeManager
        ->expects($this->any())
        ->method('getDefinitions')
        ->willReturn([
        'entity_test' => $definition,
        'entity_test_rev' => $revisionable_definition,
    ]);
    $this->entityTypeManager
        ->expects($this->any())
        ->method('getDefinition')
        ->willReturnCallback(function ($entity_type) use ($definition, $revisionable_definition) {
        if ($entity_type == 'entity_test') {
            return $definition;
        }
        elseif ($entity_type === 'entity_test_rev') {
            return $revisionable_definition;
        }
        else {
            return NULL;
        }
    });
}