CacheCollectorTest.php

<?php

namespace Drupal\Tests\Core\Cache;

use Drupal\Core\Cache\Cache;
use Drupal\Tests\UnitTestCase;

/**
 * @coversDefaultClass \Drupal\Core\Cache\CacheCollector
 * @group Cache
 */
class CacheCollectorTest extends UnitTestCase {
    
    /**
     * The cache backend that should be used.
     *
     * @var \Drupal\Core\Cache\CacheBackendInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $cacheBackend;
    
    /**
     * The cache tags invalidator.
     *
     * @var \Drupal\Core\Cache\CacheTagsInvalidatorInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $cacheTagsInvalidator;
    
    /**
     * The lock backend that should be used.
     *
     * @var \PHPUnit\Framework\MockObject\MockObject
     */
    protected $lock;
    
    /**
     * The cache id used for the test.
     *
     * @var string
     */
    protected $cid;
    
    /**
     * Cache collector implementation to test.
     *
     * @var \Drupal\Tests\Core\Cache\CacheCollectorHelper
     */
    protected $collector;
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        $this->cacheBackend = $this->createMock('Drupal\\Core\\Cache\\CacheBackendInterface');
        $this->cacheTagsInvalidator = $this->createMock('Drupal\\Core\\Cache\\CacheTagsInvalidatorInterface');
        $this->lock = $this->createMock('Drupal\\Core\\Lock\\LockBackendInterface');
        $this->cid = $this->randomMachineName();
        $this->collector = new CacheCollectorHelper($this->cid, $this->cacheBackend, $this->lock);
        $this->getContainerWithCacheTagsInvalidator($this->cacheTagsInvalidator);
    }
    
    /**
     * Tests the resolve cache miss function.
     */
    public function testResolveCacheMiss() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        $this->collector
            ->setCacheMissData($key, $value);
        $this->assertEquals($value, $this->collector
            ->get($key));
    }
    
    /**
     * Tests setting and getting values when the cache is empty.
     */
    public function testSetAndGet() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        $this->assertNull($this->collector
            ->get($key));
        $this->collector
            ->set($key, $value);
        $this->assertTrue($this->collector
            ->has($key));
        $this->assertEquals($value, $this->collector
            ->get($key));
    }
    
    /**
     * Makes sure that NULL is a valid value and is collected.
     */
    public function testSetAndGetNull() {
        $key = $this->randomMachineName();
        $value = NULL;
        $this->cacheBackend
            ->expects($this->once())
            ->method('invalidate')
            ->with($this->cid);
        $this->collector
            ->set($key, $value);
        $this->assertTrue($this->collector
            ->has($key));
        $this->assertEquals($value, $this->collector
            ->get($key));
        // Ensure that getting a value that isn't set does not mark it as
        // existent.
        $non_existing_key = $this->randomMachineName(7);
        $this->collector
            ->get($non_existing_key);
        $this->assertFalse($this->collector
            ->has($non_existing_key));
    }
    
    /**
     * Tests returning value from the collected cache.
     */
    public function testGetFromCache() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        $cache = (object) [
            'data' => [
                $key => $value,
            ],
            'created' => (int) $_SERVER['REQUEST_TIME'],
        ];
        $this->cacheBackend
            ->expects($this->once())
            ->method('get')
            ->with($this->cid)
            ->willReturn($cache);
        $this->assertTrue($this->collector
            ->has($key));
        $this->assertEquals($value, $this->collector
            ->get($key));
        $this->assertEquals(0, $this->collector
            ->getCacheMisses());
    }
    
    /**
     * Tests setting and deleting values.
     */
    public function testDelete() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        $this->assertNull($this->collector
            ->get($key));
        $this->collector
            ->set($key, $value);
        $this->assertTrue($this->collector
            ->has($key));
        $this->assertEquals($value, $this->collector
            ->get($key));
        $this->cacheBackend
            ->expects($this->once())
            ->method('invalidate')
            ->with($this->cid);
        $this->collector
            ->delete($key);
        $this->assertFalse($this->collector
            ->has($key));
        $this->assertEquals(NULL, $this->collector
            ->get($key));
    }
    
    /**
     * Tests updating the cache when no changes were made.
     */
    public function testUpdateCacheNoChanges() {
        $this->lock
            ->expects($this->never())
            ->method('acquire');
        $this->cacheBackend
            ->expects($this->never())
            ->method('set');
        // Destruct the object to trigger the update data process.
        $this->collector
            ->destruct();
    }
    
    /**
     * Tests updating the cache after a set.
     */
    public function testUpdateCache() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        $this->collector
            ->setCacheMissData($key, $value);
        $this->collector
            ->get($key);
        // Set up mock objects for the expected calls, first a lock acquire, then
        // cache get to look for conflicting cache entries, then a cache set and
        // finally the lock is released again.
        $this->lock
            ->expects($this->once())
            ->method('acquire')
            ->with($this->cid . ':Drupal\\Core\\Cache\\CacheCollector')
            ->willReturn(TRUE);
        $this->cacheBackend
            ->expects($this->once())
            ->method('get')
            ->with($this->cid, FALSE);
        $this->cacheBackend
            ->expects($this->once())
            ->method('set')
            ->with($this->cid, [
            $key => $value,
        ], Cache::PERMANENT, []);
        $this->lock
            ->expects($this->once())
            ->method('release')
            ->with($this->cid . ':Drupal\\Core\\Cache\\CacheCollector');
        // Destruct the object to trigger the update data process.
        $this->collector
            ->destruct();
    }
    
    /**
     * Tests updating the cache when the lock acquire fails.
     */
    public function testUpdateCacheLockFail() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        $this->collector
            ->setCacheMissData($key, $value);
        $this->collector
            ->get($key);
        // The lock acquire returns false, so the method should abort.
        $this->lock
            ->expects($this->once())
            ->method('acquire')
            ->with($this->cid . ':Drupal\\Core\\Cache\\CacheCollector')
            ->willReturn(FALSE);
        $this->cacheBackend
            ->expects($this->never())
            ->method('set');
        // Destruct the object to trigger the update data process.
        $this->collector
            ->destruct();
    }
    
    /**
     * Tests updating the cache when there is a conflict after cache invalidation.
     */
    public function testUpdateCacheInvalidatedConflict() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        // Set up mock cache get with conflicting entries.
        $this->cacheBackend
            ->expects($this->exactly(2))
            ->method('get')
            ->with($this->cid)
            ->willReturnOnConsecutiveCalls((object) [
            'data' => [
                $key => $value,
            ],
            'created' => (int) $_SERVER['REQUEST_TIME'],
        ], (object) [
            'data' => [
                $key => $value,
            ],
            'created' => (int) $_SERVER['REQUEST_TIME'] + 1,
        ]);
        $this->cacheBackend
            ->expects($this->once())
            ->method('invalidate')
            ->with($this->cid);
        $this->collector
            ->set($key, 'new value');
        // Set up mock objects for the expected calls, first a lock acquire, then
        // when cache get finds conflicting entries it deletes the cache and aborts.
        $this->lock
            ->expects($this->once())
            ->method('acquire')
            ->with($this->cid . ':Drupal\\Core\\Cache\\CacheCollector')
            ->willReturn(TRUE);
        $this->cacheBackend
            ->expects($this->once())
            ->method('delete')
            ->with($this->cid);
        $this->lock
            ->expects($this->once())
            ->method('release')
            ->with($this->cid . ':Drupal\\Core\\Cache\\CacheCollector');
        // Destruct the object to trigger the update data process.
        $this->collector
            ->destruct();
    }
    
    /**
     * Tests updating the cache when a different request.
     */
    public function testUpdateCacheMerge() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        $this->collector
            ->setCacheMissData($key, $value);
        $this->collector
            ->get($key);
        // Set up mock objects for the expected calls, first a lock acquire, then
        // cache get to look for existing cache entries, which does find
        // and then it merges them.
        $this->lock
            ->expects($this->once())
            ->method('acquire')
            ->with($this->cid . ':Drupal\\Core\\Cache\\CacheCollector')
            ->willReturn(TRUE);
        $cache = (object) [
            'data' => [
                'other key' => 'other value',
            ],
            'created' => (int) $_SERVER['REQUEST_TIME'] + 1,
        ];
        $this->cacheBackend
            ->expects($this->once())
            ->method('get')
            ->with($this->cid)
            ->willReturn($cache);
        $this->cacheBackend
            ->expects($this->once())
            ->method('set')
            ->with($this->cid, [
            'other key' => 'other value',
            $key => $value,
        ], Cache::PERMANENT, []);
        $this->lock
            ->expects($this->once())
            ->method('release')
            ->with($this->cid . ':Drupal\\Core\\Cache\\CacheCollector');
        // Destruct the object to trigger the update data process.
        $this->collector
            ->destruct();
    }
    
    /**
     * Tests updating the cache after a delete.
     */
    public function testUpdateCacheDelete() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        $cache = (object) [
            'data' => [
                $key => $value,
            ],
            'created' => (int) $_SERVER['REQUEST_TIME'],
        ];
        // Set up mock expectation, on the second call the with the second argument
        // set to TRUE because we triggered a cache invalidation.
        $this->cacheBackend
            ->expects($this->exactly(2))
            ->method('get')
            ->withConsecutive([
            $this->cid,
        ], [
            $this->cid,
            TRUE,
        ])
            ->willReturn($cache);
        $this->collector
            ->delete($key);
        // Set up mock objects for the expected calls, first a lock acquire, then
        // a cache set and finally the lock is released again.
        $this->lock
            ->expects($this->once())
            ->method('acquire')
            ->with($this->cid . ':Drupal\\Core\\Cache\\CacheCollector')
            ->willReturn(TRUE);
        $this->cacheBackend
            ->expects($this->once())
            ->method('set')
            ->with($this->cid, [], Cache::PERMANENT, []);
        $this->lock
            ->expects($this->once())
            ->method('release')
            ->with($this->cid . ':Drupal\\Core\\Cache\\CacheCollector');
        // Destruct the object to trigger the update data process.
        $this->collector
            ->destruct();
    }
    
    /**
     * Tests a reset of the cache collector.
     */
    public function testUpdateCacheReset() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        // Set the data and request it.
        $this->collector
            ->setCacheMissData($key, $value);
        $this->assertEquals($value, $this->collector
            ->get($key));
        $this->assertEquals($value, $this->collector
            ->get($key));
        // Should have been added to the storage and only be requested once.
        $this->assertEquals(1, $this->collector
            ->getCacheMisses());
        // Reset the collected cache, should call it again.
        $this->collector
            ->reset();
        $this->assertEquals($value, $this->collector
            ->get($key));
        $this->assertEquals(2, $this->collector
            ->getCacheMisses());
    }
    
    /**
     * Tests a clear of the cache collector.
     */
    public function testUpdateCacheClear() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        // Set the data and request it.
        $this->collector
            ->setCacheMissData($key, $value);
        $this->assertEquals($value, $this->collector
            ->get($key));
        $this->assertEquals($value, $this->collector
            ->get($key));
        // Should have been added to the storage and only be requested once.
        $this->assertEquals(1, $this->collector
            ->getCacheMisses());
        // Clear the collected cache, should call it again.
        $this->cacheBackend
            ->expects($this->once())
            ->method('delete')
            ->with($this->cid);
        $this->cacheTagsInvalidator
            ->expects($this->never())
            ->method('invalidateTags');
        $this->collector
            ->clear();
        $this->assertEquals($value, $this->collector
            ->get($key));
        $this->assertEquals(2, $this->collector
            ->getCacheMisses());
    }
    
    /**
     * Tests a clear of the cache collector using tags.
     */
    public function testUpdateCacheClearTags() {
        $key = $this->randomMachineName();
        $value = $this->randomMachineName();
        $tags = [
            $this->randomMachineName(),
        ];
        $this->collector = new CacheCollectorHelper($this->cid, $this->cacheBackend, $this->lock, $tags);
        // Set the data and request it.
        $this->collector
            ->setCacheMissData($key, $value);
        $this->assertEquals($value, $this->collector
            ->get($key));
        $this->assertEquals($value, $this->collector
            ->get($key));
        // Should have been added to the storage and only be requested once.
        $this->assertEquals(1, $this->collector
            ->getCacheMisses());
        // Clear the collected cache using the tags, should call it again.
        $this->cacheBackend
            ->expects($this->never())
            ->method('delete');
        $this->cacheTagsInvalidator
            ->expects($this->once())
            ->method('invalidateTags')
            ->with($tags);
        $this->collector
            ->clear();
        $this->assertEquals($value, $this->collector
            ->get($key));
        $this->assertEquals(2, $this->collector
            ->getCacheMisses());
    }

}