BundleConstraintValidatorTest.php in Drupal 8        
                          
                  
                        
  
  
  
  
File
  core/tests/Drupal/KernelTests/Core/Entity/BundleConstraintValidatorTest.php
  
    View source  
  <?php
namespace Drupal\KernelTests\Core\Entity;
use Drupal\Core\TypedData\DataDefinition;
use Drupal\KernelTests\KernelTestBase;
class BundleConstraintValidatorTest extends KernelTestBase {
  
  protected $typedData;
  public static $modules = [
    'node',
    'field',
    'text',
    'user',
  ];
  protected function setUp() {
    parent::setUp();
    $this
      ->installEntitySchema('user');
    $this->typedData = $this->container
      ->get('typed_data_manager');
  }
  
  public function testValidation() {
    
    $this
      ->assertValidation([
      'foo',
      'bar',
    ]);
    
    $this
      ->assertValidation('foo');
  }
  
  protected function assertValidation($bundle) {
    
    $definition = DataDefinition::create('entity_reference')
      ->addConstraint('Bundle', $bundle);
    
    $node = $this->container
      ->get('entity_type.manager')
      ->getStorage('node')
      ->create([
      'type' => 'foo',
    ]);
    $typed_data = $this->typedData
      ->create($definition, $node);
    $violations = $typed_data
      ->validate();
    $this
      ->assertEqual($violations
      ->count(), 0, 'Validation passed for correct value.');
    
    $page_node = $this->container
      ->get('entity_type.manager')
      ->getStorage('node')
      ->create([
      'type' => 'baz',
    ]);
    $typed_data = $this->typedData
      ->create($definition, $page_node);
    $violations = $typed_data
      ->validate();
    $this
      ->assertEqual($violations
      ->count(), 1, 'Validation failed for incorrect value.');
    
    $violation = $violations[0];
    $this
      ->assertEqual($violation
      ->getMessage(), t('The entity must be of bundle %bundle.', [
      '%bundle' => implode(', ', (array) $bundle),
    ]), 'The message for invalid value is correct.');
    $this
      ->assertEqual($violation
      ->getRoot(), $typed_data, 'Violation root is correct.');
    $this
      ->assertEqual($violation
      ->getInvalidValue(), $page_node, 'The invalid value is set correctly in the violation.');
  }
}