You are here

public function UncacheableQueryAccessTest::testEntityQuery in Entity API 8

Tests EntityQuery filtering.

File

tests/src/Kernel/QueryAccess/UncacheableQueryAccessTest.php, line 102

Class

UncacheableQueryAccessTest
Test uncacheable query access filtering for EntityQuery and Views.

Namespace

Drupal\Tests\entity\Kernel\QueryAccess

Code

public function testEntityQuery() {

  // Admin permission, full access.
  $admin_user = $this
    ->createUser([], [
    'administer entity_test_enhanced_with_owner',
  ]);
  $this->container
    ->get('current_user')
    ->setAccount($admin_user);
  $result = $this->storage
    ->getQuery()
    ->sort('id')
    ->execute();
  $this
    ->assertEquals([
    $this->entities[0]
      ->id(),
    $this->entities[1]
      ->id(),
    $this->entities[2]
      ->id(),
  ], array_values($result));

  // No view permissions, no access.
  $user = $this
    ->createUser([], [
    'access content',
  ]);
  $this->container
    ->get('current_user')
    ->setAccount($user);
  $result = $this->storage
    ->getQuery()
    ->execute();
  $this
    ->assertEmpty($result);

  // View own (published-only).
  $user = $this
    ->createUser([], [
    'view own entity_test_enhanced_with_owner',
  ]);
  $this->container
    ->get('current_user')
    ->setAccount($user);
  $this->entities[0]
    ->set('user_id', $user
    ->id());
  $this->entities[0]
    ->save();
  $this->entities[1]
    ->set('user_id', $user
    ->id());
  $this->entities[1]
    ->save();
  $result = $this->storage
    ->getQuery()
    ->sort('id')
    ->execute();
  $this
    ->assertEquals([
    $this->entities[1]
      ->id(),
  ], array_values($result));

  // View any (published-only).
  $user = $this
    ->createUser([], [
    'view any entity_test_enhanced_with_owner',
  ]);
  $this->container
    ->get('current_user')
    ->setAccount($user);
  $result = $this->storage
    ->getQuery()
    ->sort('id')
    ->execute();
  $this
    ->assertEquals([
    $this->entities[1]
      ->id(),
    $this->entities[2]
      ->id(),
  ], array_values($result));

  // View own unpublished.
  $user = $this
    ->createUser([], [
    'view own unpublished entity_test_enhanced_with_owner',
  ]);
  $this->container
    ->get('current_user')
    ->setAccount($user);
  $this->entities[0]
    ->set('user_id', $user
    ->id());
  $this->entities[0]
    ->save();
  $this->entities[1]
    ->set('user_id', $user
    ->id());
  $this->entities[1]
    ->save();
  $result = $this->storage
    ->getQuery()
    ->sort('id')
    ->execute();
  $this
    ->assertEquals([
    $this->entities[0]
      ->id(),
  ], array_values($result));

  // View own unpublished + view any (published-only).
  $user = $this
    ->createUser([], [
    'view own unpublished entity_test_enhanced_with_owner',
    'view any entity_test_enhanced_with_owner',
  ]);
  $this->container
    ->get('current_user')
    ->setAccount($user);
  $this->entities[0]
    ->set('user_id', $user
    ->id());
  $this->entities[0]
    ->save();
  $result = $this->storage
    ->getQuery()
    ->sort('id')
    ->execute();
  $this
    ->assertEquals([
    $this->entities[0]
      ->id(),
    $this->entities[1]
      ->id(),
    $this->entities[2]
      ->id(),
  ], array_values($result));

  // View own $first_bundle + View any $second_bundle.
  $user = $this
    ->createUser([], [
    'view own first entity_test_enhanced_with_owner',
    'view any second entity_test_enhanced_with_owner',
  ]);
  $this->container
    ->get('current_user')
    ->setAccount($user);
  $this->entities[1]
    ->set('user_id', $user
    ->id());
  $this->entities[1]
    ->save();
  $result = $this->storage
    ->getQuery()
    ->sort('id')
    ->execute();
  $this
    ->assertEquals([
    $this->entities[1]
      ->id(),
    $this->entities[2]
      ->id(),
  ], array_values($result));
}