class TestPerComponentEntityDisplay

class TestPerComponentEntityDisplay extends ComponentEntityDisplayBase {
  const MODE_NAME = 'view_mode';
  
  /**
   * The arguments of getEntity.
   *
   * @var string[]
   */
  protected $testValues;
  
  /**
   * The test entity.
   *
   * @var \PHPUnit\Framework\MockObject\MockObject
   */
  protected $entity;
  public function __construct($entity) {
    $this->entity = $entity;
  }
  
  /**
   * Gets the test entity.
   */
  protected function getEntity($entity_type, $bundle, $view_mode) {
    $this->testValues = func_get_args();
    return $this->entity;
  }
  
  /**
   * Gets the test values.
   */
  public function getTestValues() {
    return $this->testValues;
  }

}