function MigrateSourceTest::testPrepareRowPrepareException

Tests that a skip exception during prepare hooks correctly skips.

@covers ::prepareRow

File

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

Class

MigrateSourceTest

@coversDefaultClass \Drupal\migrate\Plugin\migrate\source\SourcePluginBase[[api-linebreak]] @group migrate

Namespace

Code

public function testPrepareRowPrepareException() : void {
  $this->migrationConfiguration['id'] = 'test_migration';
  $migration = $this->getMigration();
  $source = new StubSourcePlugin([], '', [], $migration);
  $row = new Row();
  $module_handler = $this->prophesize(ModuleHandlerInterface::class);
  // Return a failure from a prepare row hook.
  $module_handler->invokeAll('migrate_prepare_row', [
    $row,
    $source,
    $migration,
  ])
    ->willReturn([
    TRUE,
    TRUE,
  ])
    ->shouldBeCalled();
  $module_handler->invokeAll('migrate_' . $migration->id() . '_prepare_row', [
    $row,
    $source,
    $migration,
  ])
    ->willThrow(new MigrateSkipRowException())
    ->shouldBeCalled();
  $source->setModuleHandler($module_handler->reveal());
  // This will only be called on the first prepare because the second
  // explicitly avoids it.
  $this->idMap
    ->expects($this->once())
    ->method('saveIdMapping')
    ->with($row, [], MigrateIdMapInterface::STATUS_IGNORED);
  $this->assertFalse($source->prepareRow($row));
  // Throw an exception the second time that avoids mapping.
  $e = new MigrateSkipRowException('', FALSE);
  $module_handler->invokeAll('migrate_' . $migration->id() . '_prepare_row', [
    $row,
    $source,
    $migration,
  ])
    ->willThrow($e)
    ->shouldBeCalled();
  $this->assertFalse($source->prepareRow($row));
}