/**
* Tests the getViewBuilder() method.
*
* @covers ::getViewBuilder
*/
public function testGetViewBuilder()
{
$class = $this->getTestHandlerClass();
$entity = $this->getMock('Drupal\\Core\\Entity\\EntityTypeInterface');
$entity->expects($this->once())->method('getHandlerClass')->with('view_builder')->will($this->returnValue($class));
$this->setUpEntityManager(array('test_entity_type' => $entity));
$this->assertInstanceOf($class, $this->entityManager->getViewBuilder('test_entity_type'));
}
/**
* Tests the getViewBuilder() method.
*
* @covers ::getViewBuilder
*/
public function testGetViewBuilder()
{
$class = $this->getTestHandlerClass();
$entity = $this->prophesize(EntityTypeInterface::class);
$entity->getHandlerClass('view_builder')->willReturn($class);
$this->setUpEntityManager(array('test_entity_type' => $entity));
$this->assertInstanceOf($class, $this->entityManager->getViewBuilder('test_entity_type'));
}
