EntityReferenceFieldTest.php in Drupal 8

Namespace

Drupal\KernelTests\Core\Entity

File

core/tests/Drupal/KernelTests/Core/Entity/EntityReferenceFieldTest.php

<?php

namespace Drupal\KernelTests\Core\Entity;

use Drupal\Tests\SchemaCheckTestTrait;
use Drupal\Core\Entity\EntityInterface;
use Drupal\Core\Entity\EntityStorageException;
use Drupal\Core\Field\BaseFieldDefinition;
use Drupal\Core\Field\FieldStorageDefinitionInterface;
use Drupal\field\Entity\FieldConfig;
use Drupal\field\Entity\FieldStorageConfig;
use Drupal\Tests\field\Traits\EntityReferenceTestTrait;
use Drupal\user\Entity\Role;
use Drupal\user\Entity\User;
use Drupal\user\RoleInterface;
use Drupal\user\UserInterface;
use Drupal\entity_test\Entity\EntityTestStringId;

/**
 * Tests for the entity reference field.
 *
 * @group Entity
 */
class EntityReferenceFieldTest extends EntityKernelTestBase {
  use SchemaCheckTestTrait;
  use EntityReferenceTestTrait;

  /**
   * The entity type used in this test.
   *
   * @var string
   */
  protected $entityType = 'entity_test';

  /**
   * The entity type that is being referenced.
   *
   * @var string
   */
  protected $referencedEntityType = 'entity_test_rev';

  /**
   * The bundle used in this test.
   *
   * @var string
   */
  protected $bundle = 'entity_test';

  /**
   * The name of the field used in this test.
   *
   * @var string
   */
  protected $fieldName = 'field_test';

  /**
   * Modules to install.
   *
   * @var array
   */
  public static $modules = [
    'entity_reference_test',
    'entity_test_update',
  ];

  /**
   * {@inheritdoc}
   */
  protected function setUp() {
    parent::setUp();
    $this
      ->installEntitySchema('entity_test_rev');

    // Create a field.
    $this
      ->createEntityReferenceField($this->entityType, $this->bundle, $this->fieldName, 'Field test', $this->referencedEntityType, 'default', [
      'target_bundles' => [
        $this->bundle,
      ],
    ], FieldStorageDefinitionInterface::CARDINALITY_UNLIMITED);
  }

  /**
   * Tests reference field validation.
   */
  public function testEntityReferenceFieldValidation() {

    // Test a valid reference.
    $referenced_entity = $this->container
      ->get('entity_type.manager')
      ->getStorage($this->referencedEntityType)
      ->create([
      'type' => $this->bundle,
    ]);
    $referenced_entity
      ->save();
    $entity = $this->container
      ->get('entity_type.manager')
      ->getStorage($this->entityType)
      ->create([
      'type' => $this->bundle,
    ]);
    $entity->{$this->fieldName}->target_id = $referenced_entity
      ->id();
    $violations = $entity->{$this->fieldName}
      ->validate();
    $this
      ->assertEqual($violations
      ->count(), 0, 'Validation passes.');

    // Test an invalid reference.
    $entity->{$this->fieldName}->target_id = 9999;
    $violations = $entity->{$this->fieldName}
      ->validate();
    $this
      ->assertEqual($violations
      ->count(), 1, 'Validation throws a violation.');
    $this
      ->assertEqual($violations[0]
      ->getMessage(), t('The referenced entity (%type: %id) does not exist.', [
      '%type' => $this->referencedEntityType,
      '%id' => 9999,
    ]));

    // Test a non-referenceable bundle.
    entity_test_create_bundle('non_referenceable', NULL, $this->referencedEntityType);
    $referenced_entity = $this->entityTypeManager
      ->getStorage($this->referencedEntityType)
      ->create([
      'type' => 'non_referenceable',
    ]);
    $referenced_entity
      ->save();
    $entity->{$this->fieldName}->target_id = $referenced_entity
      ->id();
    $violations = $entity->{$this->fieldName}
      ->validate();
    $this
      ->assertEqual($violations
      ->count(), 1, 'Validation throws a violation.');
    $this
      ->assertEqual($violations[0]
      ->getMessage(), t('This entity (%type: %id) cannot be referenced.', [
      '%type' => $this->referencedEntityType,
      '%id' => $referenced_entity
        ->id(),
    ]));
  }

  /**
   * Tests the multiple target entities loader.
   */
  public function testReferencedEntitiesMultipleLoad() {

    // Create the parent entity.
    $entity = $this->container
      ->get('entity_type.manager')
      ->getStorage($this->entityType)
      ->create([
      'type' => $this->bundle,
    ]);

    // Create three target entities and attach them to parent field.
    $target_entities = [];
    $reference_field = [];
    for ($i = 0; $i < 3; $i++) {
      $target_entity = $this->container
        ->get('entity_type.manager')
        ->getStorage($this->referencedEntityType)
        ->create([
        'type' => $this->bundle,
      ]);
      $target_entity
        ->save();
      $target_entities[] = $target_entity;
      $reference_field[]['target_id'] = $target_entity
        ->id();
    }

    // Also attach a non-existent entity and a NULL target id.
    $reference_field[3]['target_id'] = 99999;
    $target_entities[3] = NULL;
    $reference_field[4]['target_id'] = NULL;
    $target_entities[4] = NULL;

    // Attach the first created target entity as the sixth item ($delta == 5) of
    // the parent entity field. We want to test the case when the same target
    // entity is referenced twice (or more times) in the same entity reference
    // field.
    $reference_field[5] = $reference_field[0];
    $target_entities[5] = $target_entities[0];

    // Create a new target entity that is not saved, thus testing the
    // "autocreate" feature.
    $target_entity_unsaved = $this->container
      ->get('entity_type.manager')
      ->getStorage($this->referencedEntityType)
      ->create([
      'type' => $this->bundle,
      'name' => $this
        ->randomString(),
    ]);
    $reference_field[6]['entity'] = $target_entity_unsaved;
    $target_entities[6] = $target_entity_unsaved;

    // Set the field value.
    $entity->{$this->fieldName}
      ->setValue($reference_field);

    // Load the target entities using EntityReferenceField::referencedEntities().
    $entities = $entity->{$this->fieldName}
      ->referencedEntities();

    // Test returned entities:
    // - Deltas must be preserved.
    // - Non-existent entities must not be retrieved in target entities result.
    foreach ($target_entities as $delta => $target_entity) {
      if (!empty($target_entity)) {
        if (!$target_entity
          ->isNew()) {

          // There must be an entity in the loaded set having the same id for
          // the same delta.
          $this
            ->assertEqual($target_entity
            ->id(), $entities[$delta]
            ->id());
        }
        else {

          // For entities that were not yet saved, there must an entity in the
          // loaded set having the same label for the same delta.
          $this
            ->assertEqual($target_entity
            ->label(), $entities[$delta]
            ->label());
        }
      }
      else {

        // A non-existent or NULL entity target id must not return any item in
        // the target entities set.
        $this
          ->assertFalse(isset($entities[$delta]));
      }
    }
  }

  /**
   * Tests referencing entities with string IDs.
   */
  public function testReferencedEntitiesStringId() {
    $field_name = 'entity_reference_string_id';
    $this
      ->installEntitySchema('entity_test_string_id');
    $this
      ->createEntityReferenceField($this->entityType, $this->bundle, $field_name, 'Field test', 'entity_test_string_id', 'default', [
      'target_bundles' => [
        $this->bundle,
      ],
    ], FieldStorageDefinitionInterface::CARDINALITY_UNLIMITED);

    // Create the parent entity.
    $entity = $this->container
      ->get('entity_type.manager')
      ->getStorage($this->entityType)
      ->create([
      'type' => $this->bundle,
    ]);

    // Create the default target entity.
    $target_entity = EntityTestStringId::create([
      'id' => $this
        ->randomString(),
      'type' => $this->bundle,
    ]);
    $target_entity
      ->save();

    // Set the field value.
    $entity->{$field_name}
      ->setValue([
      [
        'target_id' => $target_entity
          ->id(),
      ],
    ]);

    // Load the target entities using EntityReferenceField::referencedEntities().
    $entities = $entity->{$field_name}
      ->referencedEntities();
    $this
      ->assertEqual($entities[0]
      ->id(), $target_entity
      ->id());

    // Test that a string ID works as a default value and the field's config
    // schema is correct.
    $field = FieldConfig::loadByName($this->entityType, $this->bundle, $field_name);
    $field
      ->setDefaultValue($target_entity
      ->id());
    $field
      ->save();
    $this
      ->assertConfigSchema(\Drupal::service('config.typed'), 'field.field.' . $field
      ->id(), $field
      ->toArray());

    // Test that the default value works.
    $entity = $this->container
      ->get('entity_type.manager')
      ->getStorage($this->entityType)
      ->create([
      'type' => $this->bundle,
    ]);
    $entities = $entity->{$field_name}
      ->referencedEntities();
    $this
      ->assertEqual($entities[0]
      ->id(), $target_entity
      ->id());
  }

  /**
   * Tests all the possible ways to autocreate an entity via the API.
   */
  public function testAutocreateApi() {
    $entity = $this->entityTypeManager
      ->getStorage($this->entityType)
      ->create([
      'name' => $this
        ->randomString(),
    ]);

    // Test content entity autocreation.
    $this
      ->assertUserAutocreate($entity, function (EntityInterface $entity, UserInterface $user) {
      $entity
        ->set('user_id', $user);
    });
    $this
      ->assertUserAutocreate($entity, function (EntityInterface $entity, UserInterface $user) {
      $entity
        ->set('user_id', $user, FALSE);
    });
    $this
      ->assertUserAutocreate($entity, function (EntityInterface $entity, UserInterface $user) {
      $entity->user_id
        ->setValue($user);
    });
    $this
      ->assertUserAutocreate($entity, function (EntityInterface $entity, UserInterface $user) {
      $entity->user_id[0]
        ->get('entity')
        ->setValue($user);
    });
    $this
      ->assertUserAutocreate($entity, function (EntityInterface $entity, UserInterface $user) {
      $entity->user_id
        ->setValue([
        'entity' => $user,
        'target_id' => NULL,
      ]);
    });
    try {
      $message = 'Setting both the entity and an invalid target_id property fails.';
      $this
        ->assertUserAutocreate($entity, function (EntityInterface $entity, UserInterface $user) {
        $user
          ->save();
        $entity->user_id
          ->setValue([
          'entity' => $user,
          'target_id' => $this
            ->generateRandomEntityId(),
        ]);
      });
      $this
        ->fail($message);
    } catch (\InvalidArgumentException $e) {

      // Expected exception; just continue testing.
    }
    $this
      ->assertUserAutocreate($entity, function (EntityInterface $entity, UserInterface $user) {
      $entity->user_id = $user;
    });
    $this
      ->assertUserAutocreate($entity, function (EntityInterface $entity, UserInterface $user) {
      $entity->user_id->entity = $user;
    });

    // Test config entity autocreation.
    $this
      ->assertUserRoleAutocreate($entity, function (EntityInterface $entity, RoleInterface $role) {
      $entity
        ->set('user_role', $role);
    });
    $this
      ->assertUserRoleAutocreate($entity, function (EntityInterface $entity, RoleInterface $role) {
      $entity
        ->set('user_role', $role, FALSE);
    });
    $this
      ->assertUserRoleAutocreate($entity, function (EntityInterface $entity, RoleInterface $role) {
      $entity->user_role
        ->setValue($role);
    });
    $this
      ->assertUserRoleAutocreate($entity, function (EntityInterface $entity, RoleInterface $role) {
      $entity->user_role[0]
        ->get('entity')
        ->setValue($role);
    });
    $this
      ->assertUserRoleAutocreate($entity, function (EntityInterface $entity, RoleInterface $role) {
      $entity->user_role
        ->setValue([
        'entity' => $role,
        'target_id' => NULL,
      ]);
    });
    try {
      $message = 'Setting both the entity and an invalid target_id property fails.';
      $this
        ->assertUserRoleAutocreate($entity, function (EntityInterface $entity, RoleInterface $role) {
        $role
          ->save();
        $entity->user_role
          ->setValue([
          'entity' => $role,
          'target_id' => $this
            ->generateRandomEntityId(TRUE),
        ]);
      });
      $this
        ->fail($message);
    } catch (\InvalidArgumentException $e) {

      // Expected exception; just continue testing.
    }
    $this
      ->assertUserRoleAutocreate($entity, function (EntityInterface $entity, RoleInterface $role) {
      $entity->user_role = $role;
    });
    $this
      ->assertUserRoleAutocreate($entity, function (EntityInterface $entity, RoleInterface $role) {
      $entity->user_role->entity = $role;
    });

    // Test target entity saving after setting it as new.
    $storage = $this->entityTypeManager
      ->getStorage('user');
    $user_id = $this
      ->generateRandomEntityId();
    $user = $storage
      ->create([
      'uid' => $user_id,
      'name' => $this
        ->randomString(),
    ]);
    $entity->user_id = $user;
    $user
      ->save();
    $entity
      ->save();
    $this
      ->assertEqual($entity->user_id->target_id, $user
      ->id());
  }

  /**
   * Asserts that the setter callback performs autocreation for users.
   *
   * @param \Drupal\Core\Entity\EntityInterface $entity
   *   The referencing entity.
   * @param $setter_callback
   *   A callback setting the target entity on the referencing entity.
   *
   * @return bool
   *   TRUE if the user was autocreated, FALSE otherwise.
   */
  protected function assertUserAutocreate(EntityInterface $entity, $setter_callback) {
    $storage = $this->entityTypeManager
      ->getStorage('user');
    $user_id = $this
      ->generateRandomEntityId();
    $user = $storage
      ->create([
      'uid' => $user_id,
      'name' => $this
        ->randomString(),
    ]);
    $setter_callback($entity, $user);
    $entity
      ->save();
    $storage
      ->resetCache();
    $user = User::load($user_id);
    return $this
      ->assertEqual($entity->user_id->target_id, $user
      ->id());
  }

  /**
   * Asserts that the setter callback performs autocreation for user roles.
   *
   * @param \Drupal\Core\Entity\EntityInterface $entity
   *   The referencing entity.
   * @param $setter_callback
   *   A callback setting the target entity on the referencing entity.
   *
   * @return bool
   *   TRUE if the user was autocreated, FALSE otherwise.
   */
  protected function assertUserRoleAutocreate(EntityInterface $entity, $setter_callback) {
    $storage = $this->entityTypeManager
      ->getStorage('user_role');
    $role_id = $this
      ->generateRandomEntityId(TRUE);
    $role = $storage
      ->create([
      'id' => $role_id,
      'label' => $this
        ->randomString(),
    ]);
    $setter_callback($entity, $role);
    $entity
      ->save();
    $storage
      ->resetCache();
    $role = Role::load($role_id);
    return $this
      ->assertEqual($entity->user_role->target_id, $role
      ->id());
  }

  /**
   * Tests that the target entity is not unnecessarily loaded.
   */
  public function testTargetEntityNoLoad() {

    // Setup a test entity type with an entity reference field to itself. We use
    // a special storage class throwing exceptions when a load operation is
    // triggered to be able to detect them.
    $entity_type = clone $this->entityTypeManager
      ->getDefinition('entity_test_update');
    $entity_type
      ->setHandlerClass('storage', '\\Drupal\\entity_test\\EntityTestNoLoadStorage');
    $this->state
      ->set('entity_test_update.entity_type', $entity_type);
    $definitions = [
      'target_reference' => BaseFieldDefinition::create('entity_reference')
        ->setSetting('target_type', $entity_type
        ->id())
        ->setSetting('handler', 'default'),
    ];
    $this->state
      ->set('entity_test_update.additional_base_field_definitions', $definitions);
    $this->entityTypeManager
      ->clearCachedDefinitions();
    $this
      ->installEntitySchema($entity_type
      ->id());

    // Create the target entity.
    $storage = $this->entityTypeManager
      ->getStorage($entity_type
      ->id());
    $target_id = $this
      ->generateRandomEntityId();
    $target = $storage
      ->create([
      'id' => $target_id,
      'name' => $this
        ->randomString(),
    ]);
    $target
      ->save();
    $this
      ->assertEqual($target_id, $target
      ->id(), 'The target entity has a random identifier.');

    // Check that populating the reference with an existing target id does not
    // trigger a load operation.
    $message = 'The target entity was not loaded.';
    try {
      $entity = $this->entityTypeManager
        ->getStorage($entity_type
        ->id())
        ->create([
        'name' => $this
          ->randomString(),
      ]);
      $entity->target_reference = $target_id;
    } catch (EntityStorageException $e) {
      $this
        ->fail($message);
    }

    // Check that the storage actually triggers the expected exception when
    // trying to load the target entity.
    $message = 'An exception is thrown when trying to load the target entity';
    try {
      $storage
        ->load($target_id);
      $this
        ->fail($message);
    } catch (EntityStorageException $e) {

      // Expected exception; just continue testing.
    }
  }

  /**
   * Tests the dependencies entity reference fields are created with.
   */
  public function testEntityReferenceFieldDependencies() {
    $field_name = 'user_reference_field';
    $entity_type = 'entity_test';
    $field_storage = FieldStorageConfig::create([
      'field_name' => $field_name,
      'type' => 'entity_reference',
      'entity_type' => $entity_type,
      'settings' => [
        'target_type' => 'user',
      ],
    ]);
    $field_storage
      ->save();
    $this
      ->assertEqual([
      'module' => [
        'entity_test',
        'user',
      ],
    ], $field_storage
      ->getDependencies());
    $field = FieldConfig::create([
      'field_name' => $field_name,
      'entity_type' => $entity_type,
      'bundle' => 'entity_test',
      'label' => $field_name,
      'settings' => [
        'handler' => 'default',
      ],
    ]);
    $field
      ->save();
    $this
      ->assertEqual([
      'config' => [
        'field.storage.entity_test.user_reference_field',
      ],
      'module' => [
        'entity_test',
      ],
    ], $field
      ->getDependencies());
  }

}

Classes

Name	Description
EntityReferenceFieldTest	Tests for the entity reference field.