function EntityTypeRepositoryTest::testGetEntityTypeFromClass

Tests get entity type from class.

File

core/tests/Drupal/Tests/Core/Entity/EntityTypeRepositoryTest.php, line 125

Class

EntityTypeRepositoryTest

Tests Drupal\Core\Entity\EntityTypeRepository.

Namespace

Code

public function testGetEntityTypeFromClass() : void {
  $apple = $this->prophesize(EntityTypeInterface::class);
  $banana = $this->prophesize(EntityTypeInterface::class);
  $this->setUpEntityTypeDefinitions([
    'apple' => $apple,
    'banana' => $banana,
  ]);
  $apple->getOriginalClass()
    ->willReturn('\\Drupal\\apple\\Entity\\Apple');
  $apple->getDecoratedClasses()
    ->willReturn([
    '\\Drupal\\apple\\Entity\\Apple',
  ]);
  $banana->getOriginalClass()
    ->willReturn('\\Drupal\\banana\\Entity\\Banana');
  $banana->getDecoratedClasses()
    ->willReturn([
    '\\Drupal\\banana\\Entity\\Banana',
  ]);
  $banana->getClass()
    ->willReturn('\\Drupal\\mango\\Entity\\Mango');
  $banana->id()
    ->willReturn('banana')
    ->shouldBeCalledTimes(2);
  $entity_type_id = $this->entityTypeRepository
    ->getEntityTypeFromClass('\\Drupal\\banana\\Entity\\Banana');
  $this->assertSame('banana', $entity_type_id);
  $entity_type_id = $this->entityTypeRepository
    ->getEntityTypeFromClass('\\Drupal\\mango\\Entity\\Mango');
  $this->assertSame('banana', $entity_type_id);
}