<?php

/**

 * @file

 * Unit test file for the entity API.

 */

/**

 * Tests EntityFieldQuery.

 */

class EntityFieldQueryTestCase extends DrupalWebTestCase {

  public static function getInfo() {

    return array(

      'name' => 'Entity query',

      'description' => 'Test the EntityFieldQuery class.',

      'group' => 'Entity API',

    );

  }

  function setUp() {

    parent::setUp(array('node', 'field_test', 'entity_query_access_test', 'node_access_test'));

    field_test_create_bundle('bundle1');

    field_test_create_bundle('bundle2');

    field_test_create_bundle('test_bundle');

    field_test_create_bundle('test_entity_bundle');

    $instances = array();

    $this->fields = array();

    $this->field_names[0] = $field_name = drupal_strtolower($this->randomName() . '_field_name');

    $field = array('field_name' => $field_name, 'type' => 'test_field', 'cardinality' => 4);

    $field = field_create_field($field);

    $this->fields[0] = $field;

    $instance = array(

      'field_name' => $field_name,

      'entity_type' => '',

      'bundle' => '',

      'label' => $this->randomName() . '_label',

      'description' => $this->randomName() . '_description',

      'weight' => mt_rand(0, 127),

      'settings' => array(

        'test_instance_setting' => $this->randomName(),

      ),

      'widget' => array(

        'type' => 'test_field_widget',

        'label' => 'Test Field',

        'settings' => array(

          'test_widget_setting' => $this->randomName(),

        )

      )

    );

    $instances[0] = $instance;

    // Add an instance to that bundle.

    $instances[0]['bundle'] = 'bundle1';

    $instances[0]['entity_type'] = 'test_entity_bundle_key';

    field_create_instance($instances[0]);

    $instances[0]['bundle'] = 'bundle2';

    field_create_instance($instances[0]);

    $instances[0]['bundle'] = $instances[0]['entity_type'] = 'test_entity_bundle';

    field_create_instance($instances[0]);

    $this->field_names[1] = $field_name = drupal_strtolower($this->randomName() . '_field_name');

    $field = array('field_name' => $field_name, 'type' => 'shape', 'cardinality' => 4);

    $field = field_create_field($field);

    $this->fields[1] = $field;

    $instance = array(

      'field_name' => $field_name,

      'entity_type' => '',

      'bundle' => '',

      'label' => $this->randomName() . '_label',

      'description' => $this->randomName() . '_description',

      'weight' => mt_rand(0, 127),

      'settings' => array(

        'test_instance_setting' => $this->randomName(),

      ),

      'widget' => array(

        'type' => 'test_field_widget',

        'label' => 'Test Field',

        'settings' => array(

          'test_widget_setting' => $this->randomName(),

        )

      )

    );

    $instances[1] = $instance;

    // Add a field instance to the bundles.

    $instances[1]['bundle'] = 'bundle1';

    $instances[1]['entity_type'] = 'test_entity_bundle_key';

    field_create_instance($instances[1]);

    $instances[1]['bundle'] = $instances[1]['entity_type'] = 'test_entity_bundle';

    field_create_instance($instances[1]);

    $this->instances = $instances;

    // Write entity base table if there is one.

    $entities = array();

    // Create entities which have a 'bundle key' defined.

    for ($i = 1; $i < 7; $i++) {

      $entity = new stdClass();

      $entity->ftid = $i;

      $entity->fttype = ($i < 5) ? 'bundle1' : 'bundle2';

      $entity->{$this->field_names[0]}[LANGUAGE_NONE][0]['value'] = $i;

      drupal_write_record('test_entity_bundle_key', $entity);

      field_attach_insert('test_entity_bundle_key', $entity);

    }

    $entity = new stdClass();

    $entity->ftid = 5;

    $entity->fttype = 'test_entity_bundle';

    $entity->{$this->field_names[1]}[LANGUAGE_NONE][0]['shape'] = 'square';

    $entity->{$this->field_names[1]}[LANGUAGE_NONE][0]['color'] = 'red';

    $entity->{$this->field_names[1]}[LANGUAGE_NONE][1]['shape'] = 'circle';

    $entity->{$this->field_names[1]}[LANGUAGE_NONE][1]['color'] = 'blue';

    drupal_write_record('test_entity_bundle', $entity);

    field_attach_insert('test_entity_bundle', $entity);

    $instances[2] = $instance;

    $instances[2]['bundle'] = 'test_bundle';

    $instances[2]['field_name'] = $this->field_names[0];

    $instances[2]['entity_type'] = 'test_entity';

    field_create_instance($instances[2]);

    // Create entities with support for revisions.

    for ($i = 1; $i < 5; $i++) {

      $entity = new stdClass();

      $entity->ftid = $i;

      $entity->ftvid = $i;

      $entity->fttype = 'test_bundle';

      $entity->{$this->field_names[0]}[LANGUAGE_NONE][0]['value'] = $i;

      drupal_write_record('test_entity', $entity);

      field_attach_insert('test_entity', $entity);

      drupal_write_record('test_entity_revision', $entity);

    }

    // Add two revisions to an entity.

    for ($i = 100; $i < 102; $i++) {

      $entity = new stdClass();

      $entity->ftid = 4;

      $entity->ftvid = $i;

      $entity->fttype = 'test_bundle';

      $entity->{$this->field_names[0]}[LANGUAGE_NONE][0]['value'] = $i;

      drupal_write_record('test_entity', $entity, 'ftid');

      drupal_write_record('test_entity_revision', $entity);

      db_update('test_entity')

       ->fields(array('ftvid' => $entity->ftvid))

       ->condition('ftid', $entity->ftid)

       ->execute();

      field_attach_update('test_entity', $entity);

    }

  }

  /**

   * Tests EntityFieldQuery.

   */

  function testEntityFieldQuery() {

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle')

      ->entityCondition('entity_id', '5');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle', 5),

    ), 'Test query on an entity type with a generated bundle.');

    // Test entity_type condition.

    $query = new EntityFieldQuery();

    $query->entityCondition('entity_type', 'test_entity_bundle_key');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 6),

    ), 'Test entity entity_type condition.');

    // Test entity_id condition.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->entityCondition('entity_id', '3');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 3),

    ), 'Test entity entity_id condition.');

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->propertyCondition('ftid', '3');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 3),

    ), 'Test entity entity_id condition and entity_id property condition.');

    // Test bundle condition.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->entityCondition('bundle', 'bundle1');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 4),

    ), 'Test entity bundle condition: bundle1.');

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->entityCondition('bundle', 'bundle2');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 6),

    ), 'Test entity bundle condition: bundle2.');

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->propertyCondition('fttype', 'bundle2');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 6),

    ), 'Test entity bundle condition and bundle property condition.');

    // Test revision_id condition.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity')

      ->entityCondition('revision_id', '3');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity', 3),

    ), 'Test entity revision_id condition.');

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity')

      ->propertyCondition('ftvid', '3');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity', 3),

    ), 'Test entity revision_id condition and revision_id property condition.');

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity')

      ->fieldCondition($this->fields[0], 'value', 100, '>=')

      ->age(FIELD_LOAD_REVISION);

    $this->assertEntityFieldQuery($query, array(

        array('test_entity', 100),

        array('test_entity', 101),

    ), 'Test revision age.');

    // Test that fields attached to the non-revision supporting entity

    // 'test_entity_bundle_key' are reachable in FIELD_LOAD_REVISION.

    $query = new EntityFieldQuery();

    $query

      ->fieldCondition($this->fields[0], 'value', 100, '<')

      ->age(FIELD_LOAD_REVISION);

    $this->assertEntityFieldQuery($query, array(

        array('test_entity_bundle_key', 1),

        array('test_entity_bundle_key', 2),

        array('test_entity_bundle_key', 3),

        array('test_entity_bundle_key', 4),

        array('test_entity_bundle_key', 5),

        array('test_entity_bundle_key', 6),

        array('test_entity', 1),

        array('test_entity', 2),

        array('test_entity', 3),

        array('test_entity', 4),

    ), 'Test that fields are reachable from FIELD_LOAD_REVISION even for non-revision entities.');

    // Test entity sort by entity_id.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->entityOrderBy('entity_id', 'ASC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 6),

    ), 'Test sort entity entity_id in ascending order.', TRUE);

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->entityOrderBy('entity_id', 'DESC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 6),

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 1),

    ), 'Test sort entity entity_id in descending order.', TRUE);

    // Test entity sort by entity_id, with a field condition.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->fieldCondition($this->fields[0], 'value', 0, '>')

      ->entityOrderBy('entity_id', 'ASC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 6),

    ), 'Test sort entity entity_id in ascending order, with a field condition.', TRUE);

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->fieldCondition($this->fields[0], 'value', 0, '>')

      ->propertyOrderBy('ftid', 'DESC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 6),

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 1),

    ), 'Test sort entity entity_id property in descending order, with a field condition.', TRUE);

    // Test property sort by entity id.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->propertyOrderBy('ftid', 'ASC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 6),

    ), 'Test sort entity entity_id property in ascending order.', TRUE);

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->propertyOrderBy('ftid', 'DESC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 6),

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 1),

    ), 'Test sort entity entity_id property in descending order.', TRUE);

    // Test property sort by entity id, with a field condition.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->fieldCondition($this->fields[0], 'value', 0, '>')

      ->propertyOrderBy('ftid', 'ASC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 6),

    ), 'Test sort entity entity_id property in ascending order, with a field condition.', TRUE);

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->fieldCondition($this->fields[0], 'value', 0, '>')

      ->propertyOrderBy('ftid', 'DESC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 6),

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 1),

    ), 'Test sort entity entity_id property in descending order, with a field condition.', TRUE);

    // Test entity sort by bundle.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->entityOrderBy('bundle', 'ASC')

      ->propertyOrderBy('ftid', 'DESC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 6),

      array('test_entity_bundle_key', 5),

    ), 'Test sort entity bundle in ascending order, property in descending order.', TRUE);

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->entityOrderBy('bundle', 'DESC')

      ->propertyOrderBy('ftid', 'ASC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 6),

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 4),

    ), 'Test sort entity bundle in descending order, property in ascending order.', TRUE);

    // Test entity sort by bundle, with a field condition.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->fieldCondition($this->fields[0], 'value', 0, '>')

      ->entityOrderBy('bundle', 'ASC')

      ->propertyOrderBy('ftid', 'DESC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 6),

      array('test_entity_bundle_key', 5),

    ), 'Test sort entity bundle in ascending order, property in descending order, with a field condition.', TRUE);

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->fieldCondition($this->fields[0], 'value', 0, '>')

      ->entityOrderBy('bundle', 'DESC')

      ->propertyOrderBy('ftid', 'ASC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 6),

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 4),

    ), 'Test sort entity bundle in descending order, property in ascending order, with a field condition.', TRUE);

    // Test entity sort by bundle, field.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->entityOrderBy('bundle', 'ASC')

      ->fieldOrderBy($this->fields[0], 'value', 'DESC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 4),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 6),

      array('test_entity_bundle_key', 5),

    ), 'Test sort entity bundle in ascending order, field in descending order.', TRUE);

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity_bundle_key')

      ->entityOrderBy('bundle', 'DESC')

      ->fieldOrderBy($this->fields[0], 'value', 'ASC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity_bundle_key', 5),

      array('test_entity_bundle_key', 6),

      array('test_entity_bundle_key', 1),

      array('test_entity_bundle_key', 2),

      array('test_entity_bundle_key', 3),

      array('test_entity_bundle_key', 4),

    ), 'Test sort entity bundle in descending order, field in ascending order.', TRUE);

    // Test entity sort by revision_id.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity')

      ->entityOrderBy('revision_id', 'ASC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity', 1),

      array('test_entity', 2),

      array('test_entity', 3),

      array('test_entity', 4),

    ), 'Test sort entity revision_id in ascending order.', TRUE);

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity')

      ->entityOrderBy('revision_id', 'DESC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity', 4),

      array('test_entity', 3),

      array('test_entity', 2),

      array('test_entity', 1),

    ), 'Test sort entity revision_id in descending order.', TRUE);

    // Test entity sort by revision_id, with a field condition.

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity')

      ->fieldCondition($this->fields[0], 'value', 0, '>')

      ->entityOrderBy('revision_id', 'ASC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity', 1),

      array('test_entity', 2),

      array('test_entity', 3),

      array('test_entity', 4),

    ), 'Test sort entity revision_id in ascending order, with a field condition.', TRUE);

    $query = new EntityFieldQuery();

    $query

      ->entityCondition('entity_type', 'test_entity')

      ->fieldCondition($this->fields[0], 'value', 0, '>')

      ->entityOrderBy('revision_id', 'DESC');

    $this->assertEntityFieldQuery($query, array(

      array('test_entity', 4),

      array('test_entity', 3),

      array('test_entity', 2),

      array('test_entity', 1),

    ), 'Test sort entity revision_id in descending order, with a field condition.', TRUE);

    // Test property sort by revision_id.

    $query = new EntityFieldQuery();