function MigrateSourceTest::testCount

Tests that the source count is correct.

File

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

Class

MigrateSourceTest

Tests Drupal\migrate\Plugin\migrate\source\SourcePluginBase.

Namespace

Code

public function testCount() : void {
  // Mock the cache to validate set() receives appropriate arguments.
  $container = new ContainerBuilder();
  $cache = $this->createMock(CacheBackendInterface::class);
  $cache->expects($this->any())
    ->method('set')
    ->with($this->isString(), $this->isInt(), $this->isInt());
  $container->set('cache.migrate', $cache);
  \Drupal::setContainer($container);
  // Test that the basic count works.
  $source = $this->getSource();
  $this->assertEquals(1, $source->count());
  // Test caching the count works.
  $source = $this->getSource([
    'cache_counts' => TRUE,
  ]);
  $this->assertEquals(1, $source->count());
  // Test the skip argument.
  $source = $this->getSource([
    'skip_count' => TRUE,
  ]);
  $this->assertEquals(MigrateSourceInterface::NOT_COUNTABLE, $source->count());
  $this->migrationConfiguration['id'] = 'test_migration';
  $migration = $this->getMigration();
  $source = new StubSourceGeneratorPlugin([], '', [], $migration);
  // Test the skipCount property's default value.
  $this->assertEquals(MigrateSourceInterface::NOT_COUNTABLE, $source->count());
  // Test the count value using a generator.
  $source = new StubSourceGeneratorPlugin([
    'skip_count' => FALSE,
  ], '', [], $migration);
  $this->assertEquals(3, $source->count());
}