function EntityUnitTest::testCacheContexts

@covers ::getCacheContexts @covers ::addCacheContexts

File

core/tests/Drupal/Tests/Core/Entity/EntityUnitTest.php, line 601

Class

EntityUnitTest

@coversDefaultClass \Drupal\Core\Entity\Entity @group Entity @group Access

Namespace

Code

public function testCacheContexts() {
    $cache_contexts_manager = $this->getMockBuilder('Drupal\\Core\\Cache\\Context\\CacheContextsManager')
        ->disableOriginalConstructor()
        ->getMock();
    $cache_contexts_manager->method('assertValidTokens')
        ->willReturn(TRUE);
    $container = new ContainerBuilder();
    $container->set('cache_contexts_manager', $cache_contexts_manager);
    \Drupal::setContainer($container);
    // There are no cache contexts by default.
    $this->assertEquals([], $this->entity
        ->getCacheContexts());
    // Add an additional cache context.
    $this->entity
        ->addCacheContexts([
        'user',
    ]);
    $this->assertEquals([
        'user',
    ], $this->entity
        ->getCacheContexts());
}