View source
<?php
namespace Drupal\Tests\dynamic_entity_reference\Kernel;
use Drupal\Core\Field\FieldStorageDefinitionInterface;
use Drupal\entity_test\Entity\EntityTestBundle;
use Drupal\field\Entity\FieldConfig;
use Drupal\field\Entity\FieldStorageConfig;
use Drupal\KernelTests\Core\Entity\EntityKernelTestBase;
use Drupal\Tests\SchemaCheckTestTrait;
class DynamicEntityReferenceFieldTest extends EntityKernelTestBase {
use SchemaCheckTestTrait;
protected $entityType = 'entity_test';
protected $referencedEntityType = 'entity_test_with_bundle';
protected $bundle = 'entity_test';
protected $fieldName = 'field_test';
protected static $modules = [
'dynamic_entity_reference',
];
protected function setUp() : void {
parent::setUp();
$this
->installEntitySchema('entity_test_with_bundle');
FieldStorageConfig::create([
'field_name' => $this->fieldName,
'type' => 'dynamic_entity_reference',
'entity_type' => $this->entityType,
'cardinality' => FieldStorageDefinitionInterface::CARDINALITY_UNLIMITED,
'settings' => [
'exclude_entity_types' => FALSE,
'entity_type_ids' => [
$this->referencedEntityType,
],
],
])
->save();
FieldConfig::create([
'field_name' => $this->fieldName,
'entity_type' => $this->entityType,
'bundle' => $this->bundle,
'label' => 'Field test',
'settings' => [
$this->referencedEntityType => [
'handler' => 'default:' . $this->referencedEntityType,
'handler_settings' => [
'target_bundles' => NULL,
],
],
],
])
->save();
}
public function testEntityReferenceFieldValidation() {
$entity_type_manager = \Drupal::entityTypeManager();
$referenced_entity = $entity_type_manager
->getStorage($this->referencedEntityType)
->create([
'type' => $this->bundle,
]);
$referenced_entity
->save();
$entity = $entity_type_manager
->getStorage($this->entityType)
->create([
'type' => $this->bundle,
]);
$entity->{$this->fieldName}->target_type = $referenced_entity
->getEntityTypeId();
$entity->{$this->fieldName}->target_id = $referenced_entity
->id();
$violations = $entity->{$this->fieldName}
->validate();
$this
->assertEquals($violations
->count(), 0, 'Validation passes.');
$entity = $entity_type_manager
->getStorage($this->entityType)
->create([
'type' => $this->bundle,
]);
$entity->{$this->fieldName}->entity = $referenced_entity;
$violations = $entity->{$this->fieldName}
->validate();
$this
->assertEquals($violations
->count(), 0, 'Validation passes.');
$entity = $entity_type_manager
->getStorage($this->entityType)
->create([
'type' => $this->bundle,
]);
$entity->{$this->fieldName}->target_type = $referenced_entity
->getEntityTypeId();
$entity->{$this->fieldName}->target_id = 9999;
$violations = $entity->{$this->fieldName}
->validate();
$this
->assertEquals($violations
->count(), 1, 'Validation throws a violation.');
$this
->assertEquals($violations[0]
->getMessage(), t('The referenced entity (%type: %id) does not exist.', [
'%type' => $this->referencedEntityType,
'%id' => 9999,
]));
$entity = $entity_type_manager
->getStorage($this->entityType)
->create([
'type' => $this->bundle,
]);
$entity->{$this->fieldName}->target_type = $entity
->getEntityTypeId();
$entity->{$this->fieldName}->target_id = $referenced_entity
->id();
$violations = $entity->{$this->fieldName}
->validate();
$this
->assertEquals($violations
->count(), 1, 'Validation throws a violation.');
$this
->assertEquals($violations[0]
->getMessage(), t('The referenced entity type (%type) is not allowed for this field.', [
'%type' => $this->entityType,
]));
$entity = $entity_type_manager
->getStorage($this->entityType)
->create([
'type' => $this->bundle,
]);
$entity->{$this->fieldName}->entity = $entity;
$violations = $entity->{$this->fieldName}
->validate();
$this
->assertEquals($violations
->count(), 1, 'Validation throws a violation.');
$this
->assertEquals($violations[0]
->getMessage(), t('The referenced entity type (%type) is not allowed for this field.', [
'%type' => $entity
->getEntityTypeId(),
]));
$field_config = $this->container
->get('entity_type.manager')
->getStorage('field_config')
->load($this->entityType . '.' . $this->bundle . '.' . $this->fieldName);
$settings = [
'handler' => 'default:' . $this->referencedEntityType,
'handler_settings' => [
'target_bundles' => [],
],
];
$field_config
->setSetting('entity_test_with_bundle', $settings);
$field_config
->save();
$entity = $entity_type_manager
->getStorage($this->entityType)
->create([
'type' => $this->bundle,
]);
$entity->{$this->fieldName}->target_type = $referenced_entity
->getEntityTypeId();
$entity->{$this->fieldName}->target_id = $referenced_entity
->id();
$violations = $entity->{$this->fieldName}
->validate();
$this
->assertEquals($violations
->count(), 1, 'Validation throws a violation.');
$this
->assertEquals($violations[0]
->getMessage(), t('No bundle is allowed for (%type)', [
'%type' => $this->referencedEntityType,
]));
$bundle = EntityTestBundle::create([
'id' => 'newbundle',
'label' => 'New Bundle',
'revision' => FALSE,
]);
$bundle
->save();
$field_config = $this->container
->get('entity_type.manager')
->getStorage('field_config')
->load($this->entityType . '.' . $this->bundle . '.' . $this->fieldName);
$settings = [
'handler' => 'default:' . $this->referencedEntityType,
'handler_settings' => [
'target_bundles' => [
'newbundle',
],
],
];
$field_config
->setSetting('entity_test_with_bundle', $settings);
$field_config
->save();
$entity = $entity_type_manager
->getStorage($this->entityType)
->create([
'type' => $this->bundle,
]);
$entity->{$this->fieldName}->target_type = $referenced_entity
->getEntityTypeId();
$entity->{$this->fieldName}->target_id = $referenced_entity
->id();
$violations = $entity->{$this->fieldName}
->validate();
$this
->assertEquals($violations
->count(), 1, 'Validation throws a violation.');
$this
->assertEquals($violations[0]
->getMessage(), t('Referenced entity %label does not belong to one of the supported bundles (%bundles).', [
'%label' => $referenced_entity
->label(),
'%bundles' => 'newbundle',
]));
}
public function testReferencedEntitiesMultipleLoad() {
$entity_type_manager = \Drupal::entityTypeManager();
$entity = $entity_type_manager
->getStorage($this->entityType)
->create([
'type' => $this->bundle,
]);
$target_entities = [];
$reference_field = [];
for ($i = 0; $i < 3; $i++) {
$target_entity = $entity_type_manager
->getStorage($this->referencedEntityType)
->create([
'type' => $this->bundle,
]);
$target_entity
->save();
$target_entities[] = $target_entity;
$reference_field[] = [
'target_id' => $target_entity
->id(),
'target_type' => $this->referencedEntityType,
];
}
$reference_field[3]['target_id'] = 99999;
$reference_field[3]['target_type'] = $this->referencedEntityType;
$target_entities[3] = NULL;
$reference_field[4]['target_id'] = NULL;
$reference_field[4]['target_type'] = $this->referencedEntityType;
$target_entities[4] = NULL;
$reference_field[5]['target_id'] = 99999;
$reference_field[5]['target_type'] = $this->entityType;
$target_entities[5] = NULL;
$reference_field[6]['target_id'] = NULL;
$reference_field[6]['target_type'] = NULL;
$target_entities[6] = NULL;
$reference_field[7] = $reference_field[0];
$target_entities[7] = $target_entities[0];
$target_entity_unsaved = $entity_type_manager
->getStorage($this->referencedEntityType)
->create([
'type' => $this->bundle,
'name' => $this
->randomString(),
]);
$reference_field[8]['entity'] = $target_entity_unsaved;
$target_entities[8] = $target_entity_unsaved;
$entity->{$this->fieldName}
->setValue($reference_field);
$entities = $entity->{$this->fieldName}
->referencedEntities();
foreach ($target_entities as $delta => $target_entity) {
if (!empty($target_entity)) {
if (!$target_entity
->isNew()) {
$this
->assertEquals($target_entity
->id(), $entities[$delta]
->id());
}
else {
$this
->assertEquals($target_entity
->label(), $entities[$delta]
->label());
}
}
else {
$this
->assertFalse(isset($entities[$delta]));
}
}
}
}