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 @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));
}