View source  
  <?php
namespace Drupal\Tests\multiversion\Functional;
class EntityQueryTest extends MultiversionFunctionalTestBase {
  
  protected $entityTypes = [
    'entity_test' => [],
    'entity_test_rev' => [],
    'entity_test_mul' => [],
    'entity_test_mulrev' => [],
    'node' => [
      'type' => 'article',
      'title' => 'New article',
    ],
    'taxonomy_term' => [
      'name' => 'A term',
      'vid' => 123,
    ],
    'comment' => [
      'entity_type' => 'node',
      'field_name' => 'comment',
      'subject' => 'How much wood would a woodchuck chuck',
      'mail' => 'someone@example.com',
    ],
    'file' => [
      'uid' => 1,
      'filename' => 'multiversion.txt',
      'uri' => 'public://multiversion.txt',
      'filemime' => 'text/plain',
      'status' => 1,
    ],
    'shortcut' => [
      'shortcut_set' => 'default',
      'title' => 'Llama',
      'weight' => 0,
      'link' => [
        [
          'uri' => 'internal:/admin',
        ],
      ],
    ],
  ];
  
  protected $factory;
  
  protected function setUp() {
    parent::setUp();
    $this->factory = \Drupal::service('entity.query');
  }
  public function testQuery() {
    foreach ($this->entityTypes as $entity_type_id => $info) {
      if ($entity_type_id == 'file') {
        file_put_contents($info['uri'], 'Hello world!');
        $this
          ->assertTrue($info['uri'], t('The test file has been created.'));
      }
      $entity_type = $this->entityTypeManager
        ->getDefinition($entity_type_id);
      $entity = $this->entityTypeManager
        ->getStorage($entity_type_id)
        ->create($info);
      $entity
        ->save();
      
      $expected_results = [
        '1',
      ];
      $results = $this->factory
        ->get($entity_type_id)
        ->execute();
      $this
        ->assertIdentical(array_values($results), $expected_results, "Query without isNotDeleted on existing {$entity_type_id} returned expected result.");
      $results = $this->factory
        ->get($entity_type_id)
        ->isNotDeleted()
        ->execute();
      $this
        ->assertIdentical(array_values($results), $expected_results, "Query with isNotDeleted on existing {$entity_type_id} returned expected result.");
      $results = $this->factory
        ->get($entity_type_id)
        ->isDeleted()
        ->execute();
      $this
        ->assertIdentical($results, [], "Query with isDeleted on existing {$entity_type_id} returned expected result.");
      
      $revision = 1;
      $results = $this->factory
        ->get($entity_type_id)
        ->condition($entity_type
        ->getKey('revision'), $revision)
        ->execute();
      $this
        ->assertIdentical(count($results), 1, "Revision query on existing {$entity_type_id} returned expected result.");
      
      $entity
        ->delete();
      
      $expected_results = [];
      $results = $this->factory
        ->get($entity_type_id)
        ->execute();
      $this
        ->assertIdentical(array_values($results), $expected_results, "Query without isNotDeleted on deleted {$entity_type_id} returned expected result.");
      $results = $this->factory
        ->get($entity_type_id)
        ->isNotDeleted()
        ->execute();
      $this
        ->assertIdentical(array_values($results), $expected_results, "Query with isNotDeleted on deleted {$entity_type_id} returned expected result.");
      $expected_results = [
        '1',
      ];
      $results = $this->factory
        ->get($entity_type_id)
        ->isDeleted()
        ->execute();
      $this
        ->assertIdentical(array_values($results), $expected_results, "Query with isDeleted on deleted {$entity_type_id} returned expected result.");
      $results = $this->factory
        ->get($entity_type_id)
        ->condition($entity_type
        ->getKey('revision'), 2)
        ->execute();
      $this
        ->assertIdentical(count($results), 1, "Revision query on deleted {$entity_type_id} returned expected result.");
    }
  }
}