function MigrateSourceTest::testCountCacheKey

Tests that the key can be set for the count cache.

@covers ::count

File

core/modules/migrate/tests/src/Unit/MigrateSourceTest.php, line 209

Class

MigrateSourceTest

@coversDefaultClass \Drupal\migrate\Plugin\migrate\source\SourcePluginBase @group migrate

Namespace

Code

public function testCountCacheKey() {
    // Mock the cache to validate set() receives appropriate arguments.
    $container = new ContainerBuilder();
    $cache = $this->createMock(CacheBackendInterface::class);
    $cache->expects($this->any())
        ->method('set')
        ->with('test_key', $this->isType('int'), $this->isType('int'));
    $container->set('cache.migrate', $cache);
    \Drupal::setContainer($container);
    // Test caching the count with a configured key works.
    $source = $this->getSource([
        'cache_counts' => TRUE,
        'cache_key' => 'test_key',
    ]);
    $this->assertEquals(1, $source->count());
}