You are here

FieldDataCountTest.php in Zircon Profile 8

Same filename and directory in other branches
  1. 8.0 core/modules/field/src/Tests/FieldDataCountTest.php

Namespace

Drupal\field\Tests

File

core/modules/field/src/Tests/FieldDataCountTest.php
View source
<?php

/**
 * @file
 * Contains \Drupal\field\Tests\FieldDataCountTest.
 */
namespace Drupal\field\Tests;

use Drupal\Core\Entity\Sql\SqlContentEntityStorage;
use Drupal\field\Entity\FieldConfig;
use Drupal\field\Entity\FieldStorageConfig;

/**
 * Tests counting field data records and the hasData() method on
 * FieldStorageConfig entity.
 *
 * @group field
 * @see \Drupal\Core\Entity\FieldableEntityStorageInterface::countFieldData()
 * @see \Drupal\field\Entity\FieldStorageConfig::hasData()
 */
class FieldDataCountTest extends FieldUnitTestBase {

  /**
   * @var \Drupal\Core\Entity\DynamicallyFieldableEntityStorageInterface
   */
  protected $storage;

  /**
   * @var \Drupal\Core\Entity\DynamicallyFieldableEntityStorageInterface
   */
  protected $storageRev;

  /**
   * @var \Drupal\Core\Entity\DynamicallyFieldableEntityStorageInterface
   */
  protected $storageUser;

  /**
   * {@inheritdoc}
   */
  protected function setUp() {
    parent::setUp();
    $this
      ->installEntitySchema('entity_test_rev');
    $this->storage = \Drupal::entityManager()
      ->getStorage('entity_test');
    $this->storageRev = \Drupal::entityManager()
      ->getStorage('entity_test_rev');
    $this->storageUser = \Drupal::entityManager()
      ->getStorage('user');
  }

  /**
   * Tests entityCount() and hadData() methods.
   */
  public function testEntityCountAndHasData() {

    // Create a field with a cardinality of 2 to show that we are counting
    // entities and not rows in a table.

    /** @var \Drupal\field\Entity\FieldStorageConfig $field_storage */
    $field_storage = entity_create('field_storage_config', array(
      'field_name' => 'field_int',
      'entity_type' => 'entity_test',
      'type' => 'integer',
      'cardinality' => 2,
    ));
    $field_storage
      ->save();
    entity_create('field_config', array(
      'field_storage' => $field_storage,
      'bundle' => 'entity_test',
    ))
      ->save();
    $this
      ->assertIdentical($field_storage
      ->hasdata(), FALSE, 'There are no entities with field data.');
    $this
      ->assertIdentical($this->storage
      ->countFieldData($field_storage), 0, 'There are 0 entities with field data.');

    // Create 1 entity without the field.
    $entity = entity_create('entity_test');
    $entity->name->value = $this
      ->randomMachineName();
    $entity
      ->save();
    $this
      ->assertIdentical($field_storage
      ->hasdata(), FALSE, 'There are no entities with field data.');
    $this
      ->assertIdentical($this->storage
      ->countFieldData($field_storage), 0, 'There are 0 entities with field data.');

    // Create 12 entities to ensure that the purging works as expected.
    for ($i = 0; $i < 12; $i++) {
      $entity = entity_create('entity_test');
      $entity->field_int[] = mt_rand(1, 99);
      $entity->field_int[] = mt_rand(1, 99);
      $entity->name[] = $this
        ->randomMachineName();
      $entity
        ->save();
    }
    $storage = \Drupal::entityManager()
      ->getStorage('entity_test');
    if ($storage instanceof SqlContentEntityStorage) {

      // Count the actual number of rows in the field table.
      $table_mapping = $storage
        ->getTableMapping();
      $field_table_name = $table_mapping
        ->getDedicatedDataTableName($field_storage);
      $result = db_select($field_table_name, 't')
        ->fields('t')
        ->countQuery()
        ->execute()
        ->fetchField();
      $this
        ->assertEqual($result, 24, 'The field table has 24 rows.');
    }
    $this
      ->assertIdentical($field_storage
      ->hasdata(), TRUE, 'There are entities with field data.');
    $this
      ->assertEqual($this->storage
      ->countFieldData($field_storage), 12, 'There are 12 entities with field data.');

    // Ensure the methods work on deleted fields.
    $field_storage
      ->delete();
    $this
      ->assertIdentical($field_storage
      ->hasdata(), TRUE, 'There are entities with deleted field data.');
    $this
      ->assertEqual($this->storage
      ->countFieldData($field_storage), 12, 'There are 12 entities with deleted field data.');
    field_purge_batch(6);
    $this
      ->assertIdentical($field_storage
      ->hasdata(), TRUE, 'There are entities with deleted field data.');
    $this
      ->assertEqual($this->storage
      ->countFieldData($field_storage), 6, 'There are 6 entities with deleted field data.');
    $entity_type = 'entity_test_rev';
    $this
      ->createFieldWithStorage('_2', $entity_type);
    $entity_init = entity_create($entity_type, array(
      'type' => $entity_type,
    ));
    $cardinality = $this->fieldTestData->field_storage_2
      ->getCardinality();
    $this
      ->assertIdentical($this->fieldTestData->field_storage_2
      ->hasData(), FALSE, 'There are no entities with field data.');
    $this
      ->assertIdentical($this->storageRev
      ->countFieldData($this->fieldTestData->field_storage_2), 0, 'There are 0 entities with field data.');

    // Create 1 entity with the field.
    $entity = clone $entity_init;
    $values = $this
      ->_generateTestFieldValues($this->fieldTestData->field_storage_2
      ->getCardinality());
    $entity->{$this->fieldTestData->field_name_2} = $values;
    $entity
      ->setNewRevision();
    $entity
      ->save();
    $first_revision = $entity
      ->getRevisionId();
    $this
      ->assertIdentical($this->fieldTestData->field_storage_2
      ->hasData(), TRUE, 'There are entities with field data.');
    $this
      ->assertIdentical($this->storageRev
      ->countFieldData($this->fieldTestData->field_storage_2), 1, 'There is 1 entity with field data.');
    $entity->{$this->fieldTestData->field_name_2} = array();
    $entity
      ->setNewRevision();
    $entity
      ->save();
    $this
      ->assertIdentical($this->fieldTestData->field_storage_2
      ->hasData(), TRUE, 'There are entities with field data.');
    $storage = $this->container
      ->get('entity.manager')
      ->getStorage($entity_type);
    $entity = $storage
      ->loadRevision($first_revision);
    $this
      ->assertEqual(count($entity->{$this->fieldTestData->field_name_2}), $cardinality, format_string('Revision %revision_id: expected number of values.', array(
      '%revision_id' => $first_revision,
    )));
  }

  /**
   * Verify that we can count a table that contains an entry with index 0.
   */
  public function testCountWithIndex0() {

    // Create a field that will require dedicated storage.

    /** @var \Drupal\field\Entity\FieldStorageConfig $field_storage */
    $field_storage = FieldStorageConfig::create(array(
      'field_name' => 'field_int',
      'entity_type' => 'user',
      'type' => 'integer',
      'cardinality' => 2,
    ));
    $field_storage
      ->save();
    FieldConfig::create(array(
      'field_storage' => $field_storage,
      'bundle' => 'user',
    ))
      ->save();

    // Create an entry for the anonymous user, who has user ID 0.
    $user = $this->storageUser
      ->create(array(
      'uid' => 0,
      'name' => 'anonymous',
      'mail' => NULL,
      'status' => FALSE,
      'field_int' => 42,
    ));
    $user
      ->save();

    // Test shared table storage.
    $storage = $user
      ->getFieldDefinition('name')
      ->getFieldStorageDefinition();
    $this
      ->assertIdentical(TRUE, $this->storageUser
      ->countFieldData($storage, TRUE));

    // Test dedicated table storage.
    $storage = $user
      ->getFieldDefinition('field_int')
      ->getFieldStorageDefinition();
    $this
      ->assertIdentical(TRUE, $this->storageUser
      ->countFieldData($storage, TRUE));
  }

}

Classes

Namesort descending Description
FieldDataCountTest Tests counting field data records and the hasData() method on FieldStorageConfig entity.