public function TempStoreDatabaseTest::testUserTempStore in Zircon Profile 8

Same name and namespace in other branches

1. 8.0 core/modules/user/src/Tests/TempStoreDatabaseTest.php \Drupal\user\Tests\TempStoreDatabaseTest::testUserTempStore()

Tests the UserTempStore API.

File

core/modules/user/src/Tests/TempStoreDatabaseTest.php, line 76

Contains \Drupal\user\Tests\TempStoreDatabaseTest.

Class

TempStoreDatabaseTest

Tests the temporary object storage system.

Namespace

Drupal\user\Tests

Code

public function testUserTempStore() {

  // Create a key/value collection.
  $factory = new SharedTempStoreFactory(new KeyValueExpirableFactory(\Drupal::getContainer()), new DatabaseLockBackend(Database::getConnection()), $this->container
    ->get('request_stack'));
  $collection = $this
    ->randomMachineName();

  // Create two mock users.
  for ($i = 0; $i <= 1; $i++) {
    $users[$i] = mt_rand(500, 5000000);

    // Storing the SharedTempStore objects in a class member variable causes a
    // fatal exception, because in that situation garbage collection is not
    // triggered until the test class itself is destructed, after tearDown()
    // has deleted the database tables. Store the objects locally instead.
    $stores[$i] = $factory
      ->get($collection, $users[$i]);
  }
  $key = $this
    ->randomMachineName();

  // Test that setIfNotExists() succeeds only the first time.
  for ($i = 0; $i <= 1; $i++) {

    // setIfNotExists() should be TRUE the first time (when $i is 0) and
    // FALSE the second time (when $i is 1).
    $this
      ->assertEqual(!$i, $stores[0]
      ->setIfNotExists($key, $this->objects[$i]));
    $metadata = $stores[0]
      ->getMetadata($key);
    $this
      ->assertEqual($users[0], $metadata->owner);
    $this
      ->assertIdenticalObject($this->objects[0], $stores[0]
      ->get($key));

    // Another user should get the same result.
    $metadata = $stores[1]
      ->getMetadata($key);
    $this
      ->assertEqual($users[0], $metadata->owner);
    $this
      ->assertIdenticalObject($this->objects[0], $stores[1]
      ->get($key));
  }

  // Remove the item and try to set it again.
  $stores[0]
    ->delete($key);
  $stores[0]
    ->setIfNotExists($key, $this->objects[1]);

  // This time it should succeed.
  $this
    ->assertIdenticalObject($this->objects[1], $stores[0]
    ->get($key));

  // This user can update the object.
  $stores[0]
    ->set($key, $this->objects[2]);
  $this
    ->assertIdenticalObject($this->objects[2], $stores[0]
    ->get($key));

  // The object is the same when another user loads it.
  $this
    ->assertIdenticalObject($this->objects[2], $stores[1]
    ->get($key));

  // This user should be allowed to get, update, delete.
  $this
    ->assertTrue($stores[0]
    ->getIfOwner($key) instanceof \stdClass);
  $this
    ->assertTrue($stores[0]
    ->setIfOwner($key, $this->objects[1]));
  $this
    ->assertTrue($stores[0]
    ->deleteIfOwner($key));

  // Another user can update the object and become the owner.
  $stores[1]
    ->set($key, $this->objects[3]);
  $this
    ->assertIdenticalObject($this->objects[3], $stores[0]
    ->get($key));
  $this
    ->assertIdenticalObject($this->objects[3], $stores[1]
    ->get($key));
  $metadata = $stores[1]
    ->getMetadata($key);
  $this
    ->assertEqual($users[1], $metadata->owner);

  // The first user should be informed that the second now owns the data.
  $metadata = $stores[0]
    ->getMetadata($key);
  $this
    ->assertEqual($users[1], $metadata->owner);

  // The first user should no longer be allowed to get, update, delete.
  $this
    ->assertNull($stores[0]
    ->getIfOwner($key));
  $this
    ->assertFalse($stores[0]
    ->setIfOwner($key, $this->objects[1]));
  $this
    ->assertFalse($stores[0]
    ->deleteIfOwner($key));

  // Now manually expire the item (this is not exposed by the API) and then
  // assert it is no longer accessible.
  db_update('key_value_expire')
    ->fields(array(
    'expire' => REQUEST_TIME - 1,
  ))
    ->condition('collection', "user.shared_tempstore.{$collection}")
    ->condition('name', $key)
    ->execute();
  $this
    ->assertFalse($stores[0]
    ->get($key));
  $this
    ->assertFalse($stores[1]
    ->get($key));
}