class MigrateExecutableTest

class MigrateExecutableTest extends MigrateTestCase {
    
    /**
     * Stores ID map records of the ID map plugin from ::getTestRollbackIdMap.
     *
     * @var string[][]
     */
    protected static $idMapRecords;
    
    /**
     * The mocked migration entity.
     *
     * @var \Drupal\migrate\Plugin\MigrationInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $migration;
    
    /**
     * The mocked migrate message.
     *
     * @var \Drupal\migrate\MigrateMessageInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $message;
    
    /**
     * The tested migrate executable.
     *
     * @var \Drupal\Tests\migrate\Unit\TestMigrateExecutable
     */
    protected $executable;
    
    /**
     * A mocked event dispatcher.
     *
     * @var \Symfony\Contracts\EventDispatcher\EventDispatcherInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $eventDispatcher;
    
    /**
     * The migration's configuration values.
     *
     * @var array
     */
    protected $migrationConfiguration = [
        'id' => 'test',
    ];
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        static::$idMapRecords = [];
        $this->migration = $this->getMigration();
        $this->message = $this->createMock('Drupal\\migrate\\MigrateMessageInterface');
        $this->eventDispatcher = $this->createMock('Symfony\\Contracts\\EventDispatcher\\EventDispatcherInterface');
        $this->executable = new TestMigrateExecutable($this->migration, $this->message, $this->eventDispatcher);
        $this->executable
            ->setStringTranslation($this->getStringTranslationStub());
    }
    
    /**
     * Tests an import with an incomplete rewinding.
     */
    public function testImportWithFailingRewind() : void {
        $exception_message = $this->getRandomGenerator()
            ->string();
        $source = $this->createMock('Drupal\\migrate\\Plugin\\MigrateSourceInterface');
        $source->expects($this->once())
            ->method('rewind')
            ->will($this->throwException(new \Exception($exception_message)));
        // The exception message contains the line number where it is thrown. Save
        // it for the testing the exception message.
        $line = __LINE__ - 3;
        $this->migration
            ->expects($this->any())
            ->method('getSourcePlugin')
            ->willReturn($source);
        // Ensure that a message with the proper message was added.
        $exception_message .= " in " . __FILE__ . " line {$line}";
        $this->message
            ->expects($this->once())
            ->method('display')
            ->with("Migration failed with source plugin exception: " . Html::escape($exception_message));
        $result = $this->executable
            ->import();
        $this->assertEquals(MigrationInterface::RESULT_FAILED, $result);
    }
    
    /**
     * Tests the import method with a valid row.
     */
    public function testImportWithValidRow() : void {
        $source = $this->getMockSource();
        $this->executable
            ->setSource($source);
        $this->migration
            ->expects($this->once())
            ->method('getProcessPlugins')
            ->willReturn([]);
        $destination = $this->createMock('Drupal\\migrate\\Plugin\\MigrateDestinationInterface');
        $this->migration
            ->method('getDestinationPlugin')
            ->willReturn($destination);
        $this->assertSame(MigrationInterface::RESULT_COMPLETED, $this->executable
            ->import());
    }
    
    /**
     * Tests the import method with a valid row.
     */
    public function testImportWithValidRowWithoutDestinationId() : void {
        $source = $this->getMockSource();
        $this->executable
            ->setSource($source);
        $this->migration
            ->expects($this->once())
            ->method('getProcessPlugins')
            ->willReturn([]);
        $destination = $this->createMock('Drupal\\migrate\\Plugin\\MigrateDestinationInterface');
        $this->migration
            ->method('getDestinationPlugin')
            ->willReturn($destination);
        $this->idMap
            ->expects($this->never())
            ->method('saveIdMapping');
        $this->assertSame(MigrationInterface::RESULT_COMPLETED, $this->executable
            ->import());
    }
    
    /**
     * Tests the import method with a valid row.
     */
    public function testImportWithValidRowNoDestinationValues() : void {
        $source = $this->getMockSource();
        $this->executable
            ->setSource($source);
        $this->migration
            ->expects($this->once())
            ->method('getProcessPlugins')
            ->willReturn([]);
        $destination = $this->createMock('Drupal\\migrate\\Plugin\\MigrateDestinationInterface');
        $this->migration
            ->method('getDestinationPlugin')
            ->willReturn($destination);
        $this->assertSame(MigrationInterface::RESULT_COMPLETED, $this->executable
            ->import());
    }
    
    /**
     * Tests the import method with a thrown MigrateException.
     *
     * The MigrationException in this case is being thrown from the destination.
     */
    public function testImportWithValidRowWithDestinationMigrateException() : void {
        $source = $this->getMockSource();
        $this->executable
            ->setSource($source);
        $this->migration
            ->expects($this->once())
            ->method('getProcessPlugins')
            ->willReturn([]);
        $destination = $this->createMock('Drupal\\migrate\\Plugin\\MigrateDestinationInterface');
        $this->migration
            ->method('getDestinationPlugin')
            ->willReturn($destination);
        $this->assertSame(MigrationInterface::RESULT_COMPLETED, $this->executable
            ->import());
    }
    
    /**
     * Tests the import method with a thrown MigrateException.
     *
     * The MigrationException in this case is being thrown from a process plugin.
     */
    public function testImportWithValidRowWithProcesMigrateException() : void {
        $exception_message = $this->getRandomGenerator()
            ->string();
        $source = $this->getMockSource();
        $row = $this->getMockBuilder('Drupal\\migrate\\Row')
            ->disableOriginalConstructor()
            ->getMock();
        $row->expects($this->once())
            ->method('getSourceIdValues')
            ->willReturn([
            'id' => 'test',
        ]);
        $source->expects($this->once())
            ->method('current')
            ->willReturn($row);
        $this->executable
            ->setSource($source);
        $this->migration
            ->expects($this->once())
            ->method('getProcessPlugins')
            ->willThrowException(new MigrateException($exception_message));
        $destination = $this->createMock('Drupal\\migrate\\Plugin\\MigrateDestinationInterface');
        $destination->expects($this->never())
            ->method('import');
        $this->migration
            ->method('getDestinationPlugin')
            ->willReturn($destination);
        $this->idMap
            ->expects($this->once())
            ->method('saveIdMapping')
            ->with($row, [], MigrateIdMapInterface::STATUS_FAILED, NULL);
        $this->idMap
            ->expects($this->once())
            ->method('saveMessage');
        $this->idMap
            ->expects($this->never())
            ->method('lookupDestinationIds');
        $this->assertSame(MigrationInterface::RESULT_COMPLETED, $this->executable
            ->import());
    }
    
    /**
     * Tests the import method with a regular Exception being thrown.
     */
    public function testImportWithValidRowWithException() : void {
        $source = $this->getMockSource();
        $this->executable
            ->setSource($source);
        $this->migration
            ->expects($this->once())
            ->method('getProcessPlugins')
            ->willReturn([]);
        $destination = $this->createMock('Drupal\\migrate\\Plugin\\MigrateDestinationInterface');
        $this->migration
            ->method('getDestinationPlugin')
            ->willReturn($destination);
        $this->assertSame(MigrationInterface::RESULT_COMPLETED, $this->executable
            ->import());
    }
    
    /**
     * Tests the processRow method.
     */
    public function testProcessRow() : void {
        $expected = [
            'test' => 'test destination',
            'test1' => 'test1 destination',
        ];
        foreach ($expected as $key => $value) {
            $plugins[$key][0] = $this->createMock('Drupal\\migrate\\Plugin\\MigrateProcessInterface');
            $plugins[$key][0]->expects($this->once())
                ->method('getPluginDefinition')
                ->willReturn([]);
            $plugins[$key][0]->expects($this->once())
                ->method('transform')
                ->willReturn($value);
        }
        $this->migration
            ->expects($this->once())
            ->method('getProcessPlugins')
            ->with(NULL)
            ->willReturn($plugins);
        $row = new Row();
        $this->executable
            ->processRow($row);
        foreach ($expected as $key => $value) {
            $this->assertSame($row->getDestinationProperty($key), $value);
        }
        $this->assertSameSize($expected, $row->getDestination());
    }
    
    /**
     * Tests the processRow method with an empty pipeline.
     */
    public function testProcessRowEmptyPipeline() : void {
        $this->migration
            ->expects($this->once())
            ->method('getProcessPlugins')
            ->with(NULL)
            ->willReturn([
            'test' => [],
        ]);
        $row = new Row();
        $this->executable
            ->processRow($row);
        $this->assertSame($row->getDestination(), []);
    }
    
    /**
     * Tests the processRow pipeline exception.
     */
    public function testProcessRowPipelineException() : void {
        $row = new Row();
        $plugin = $this->prophesize(MigrateProcessInterface::class);
        $plugin->getPluginDefinition()
            ->willReturn([
            'handle_multiples' => FALSE,
        ]);
        $plugin->transform(NULL, $this->executable, $row, 'destination_id')
            ->willReturn('transform_return_string');
        $plugin->multiple()
            ->willReturn(TRUE);
        $plugin->getPluginId()
            ->willReturn('plugin_id');
        $plugin->reset()
            ->shouldBeCalled();
        $plugin->isPipelineStopped()
            ->willReturn(FALSE);
        $plugin = $plugin->reveal();
        $plugins['destination_id'] = [
            $plugin,
            $plugin,
        ];
        $this->migration
            ->method('getProcessPlugins')
            ->willReturn($plugins);
        $this->expectException(MigrateException::class);
        $this->expectExceptionMessage('Pipeline failed at plugin_id plugin for destination destination_id: transform_return_string received instead of an array,');
        $this->executable
            ->processRow($row);
    }
    
    /**
     * Tests a plugin which stops the pipeline.
     */
    public function testStopPipeline() : void {
        $row = new Row();
        // Prophesize a plugin that stops the pipeline and returns 'first_plugin'.
        $stop_plugin = $this->prophesize(MigrateProcessInterface::class);
        $stop_plugin->getPluginDefinition()
            ->willReturn([
            'handle_multiples' => FALSE,
        ]);
        $stop_plugin->transform(NULL, $this->executable, $row, 'destination_id')
            ->willReturn('first_plugin');
        $stop_plugin->multiple()
            ->willReturn(FALSE);
        $stop_plugin->reset()
            ->shouldBeCalled();
        $stop_plugin->isPipelineStopped()
            ->willReturn(TRUE);
        // Prophesize a plugin that transforms 'first_plugin' to 'final_plugin'.
        $final_plugin = $this->prophesize(MigrateProcessInterface::class);
        $final_plugin->getPluginDefinition()
            ->willReturn([
            'handle_multiples' => FALSE,
        ]);
        $final_plugin->transform('first_plugin', $this->executable, $row, 'destination_id')
            ->willReturn('final_plugin');
        $plugins['destination_id'] = [
            $stop_plugin->reveal(),
            $final_plugin->reveal(),
        ];
        $this->migration
            ->method('getProcessPlugins')
            ->willReturn($plugins);
        // Process the row and confirm that destination value is 'first_plugin'.
        $this->executable
            ->processRow($row);
        $this->assertEquals('first_plugin', $row->getDestinationProperty('destination_id'));
    }
    
    /**
     * Tests a plugin which does not stop the pipeline.
     */
    public function testContinuePipeline() : void {
        $row = new Row();
        // Prophesize a plugin that does not stop the pipeline.
        $continue_plugin = $this->prophesize(MigrateProcessInterface::class);
        $continue_plugin->getPluginDefinition()
            ->willReturn([
            'handle_multiples' => FALSE,
        ]);
        $continue_plugin->transform(NULL, $this->executable, $row, 'destination_id')
            ->willReturn('first_plugin');
        $continue_plugin->multiple()
            ->willReturn(FALSE);
        $continue_plugin->reset()
            ->shouldBeCalled();
        $continue_plugin->isPipelineStopped()
            ->willReturn(FALSE);
        // Prophesize a plugin that transforms 'first_plugin' to 'final_plugin'.
        $final_plugin = $this->prophesize(MigrateProcessInterface::class);
        $final_plugin->getPluginDefinition()
            ->willReturn([
            'handle_multiples' => FALSE,
        ]);
        $final_plugin->transform('first_plugin', $this->executable, $row, 'destination_id')
            ->willReturn('final_plugin');
        $final_plugin->multiple()
            ->willReturn(FALSE);
        $final_plugin->reset()
            ->shouldBeCalled();
        $final_plugin->isPipelineStopped()
            ->willReturn(FALSE);
        $plugins['destination_id'] = [
            $continue_plugin->reveal(),
            $final_plugin->reveal(),
        ];
        $this->migration
            ->method('getProcessPlugins')
            ->willReturn($plugins);
        // Process the row and confirm that the destination value is 'final_plugin'.
        $this->executable
            ->processRow($row);
        $this->assertEquals('final_plugin', $row->getDestinationProperty('destination_id'));
    }
    
    /**
     * Tests the processRow method.
     */
    public function testProcessRowEmptyDestination() : void {
        $expected = [
            'test' => 'test destination',
            'test1' => 'test1 destination',
            'test2' => NULL,
        ];
        $row = new Row();
        $plugins = [];
        foreach ($expected as $key => $value) {
            $plugin = $this->prophesize(MigrateProcessInterface::class);
            $plugin->getPluginDefinition()
                ->willReturn([]);
            $plugin->transform(NULL, $this->executable, $row, $key)
                ->willReturn($value);
            $plugin->multiple()
                ->willReturn(TRUE);
            $plugin->reset()
                ->shouldBeCalled();
            $plugin->isPipelineStopped()
                ->willReturn(FALSE);
            $plugins[$key][0] = $plugin->reveal();
        }
        $this->migration
            ->method('getProcessPlugins')
            ->willReturn($plugins);
        $this->executable
            ->processRow($row);
        foreach ($expected as $key => $value) {
            $this->assertSame($value, $row->getDestinationProperty($key));
        }
        $this->assertCount(2, $row->getDestination());
        $this->assertSame([
            'test2',
        ], $row->getEmptyDestinationProperties());
    }
    
    /**
     * Returns a mock migration source instance.
     *
     * @return \Drupal\migrate\Plugin\MigrateSourceInterface|\PHPUnit\Framework\MockObject\MockObject
     *   The mocked migration source.
     */
    protected function getMockSource() {
        $source = $this->createMock(StubSourcePlugin::class);
        $source->expects($this->once())
            ->method('rewind');
        $source->expects($this->any())
            ->method('valid')
            ->willReturn(TRUE, FALSE);
        return $source;
    }
    
    /**
     * Tests rollback.
     *
     * @param array[] $id_map_records
     *   The ID map records to test with.
     * @param bool $rollback_called
     *   Sets an expectation that the destination's rollback() will or will not be
     *   called.
     * @param string[] $source_id_keys
     *   The keys of the source IDs. The provided source ID keys must be defined
     *   in the $id_map_records parameter. Optional, defaults to ['source'].
     * @param string[] $destination_id_keys
     *   The keys of the destination IDs. The provided keys must be defined in the
     *   $id_map_records parameter. Optional, defaults to ['destination'].
     * @param int $expected_result
     *   The expected result of the rollback action. Optional, defaults to
     *   MigrationInterface::RESULT_COMPLETED.
     *
     * @dataProvider providerTestRollback
     *
     * @covers ::rollback
     */
    public function testRollback(array $id_map_records, bool $rollback_called = TRUE, array $source_id_keys = [
        'source',
    ], array $destination_id_keys = [
        'destination',
    ], int $expected_result = MigrationInterface::RESULT_COMPLETED) : void {
        $id_map = $this->getTestRollbackIdMap($id_map_records, $source_id_keys, $destination_id_keys)
            ->reveal();
        $migration = $this->getMigration($id_map);
        $destination = $this->prophesize(MigrateDestinationInterface::class);
        if ($rollback_called) {
            $destination->rollback($id_map->currentDestination())
                ->shouldBeCalled();
        }
        else {
            $destination->rollback()
                ->shouldNotBeCalled();
        }
        $migration->method('getDestinationPlugin')
            ->willReturn($destination->reveal());
        $executable = new TestMigrateExecutable($migration, $this->message, $this->eventDispatcher);
        $this->assertEquals($expected_result, $executable->rollback());
    }
    
    /**
     * Data provider for ::testRollback.
     *
     * @return array
     *   The test cases.
     */
    public static function providerTestRollback() {
        return [
            'Rollback delete' => [
                'id_map_records' => [
                    [
                        'source' => '1',
                        'destination' => '1',
                        'rollback_action' => MigrateIdMapInterface::ROLLBACK_DELETE,
                    ],
                ],
            ],
            'Rollback preserve' => [
                'id_map_records' => [
                    [
                        'source' => '1',
                        'destination' => '1',
                        'rollback_action' => MigrateIdMapInterface::ROLLBACK_PRESERVE,
                    ],
                ],
                'rollback_called' => FALSE,
            ],
            'Rolling back a failed row' => [
                'id_map_records' => [
                    [
                        'source' => '1',
                        'destination' => NULL,
                        'source_row_status' => '2',
                        'rollback_action' => MigrateIdMapInterface::ROLLBACK_DELETE,
                    ],
                ],
                'rollback_called' => FALSE,
            ],
            'Rolling back with ID map having records with duplicated destination ID' => [
                'id_map_records' => [
                    [
                        'source_1' => '1',
                        'source_2' => '1',
                        'destination' => '1',
                        'rollback_action' => MigrateIdMapInterface::ROLLBACK_DELETE,
                    ],
                    [
                        'source_1' => '2',
                        'source_2' => '2',
                        'destination' => '2',
                        'rollback_action' => MigrateIdMapInterface::ROLLBACK_PRESERVE,
                    ],
                    [
                        'source_1' => '3',
                        'source_2' => '3',
                        'destination' => '1',
                        'rollback_action' => MigrateIdMapInterface::ROLLBACK_DELETE,
                    ],
                ],
                'rollback_called' => TRUE,
                'source_id_keys' => [
                    'source_1',
                    'source_2',
                ],
            ],
            'Rollback NULL' => [
                'id_map_records' => [
                    [
                        'source' => '1',
                        'destination' => '1',
                        'rollback_action' => NULL,
                    ],
                ],
            ],
            'Rollback missing' => [
                'id_map_records' => [
                    [
                        'source' => '1',
                        'destination' => '1',
                    ],
                ],
            ],
        ];
    }
    
    /**
     * Returns an ID map object prophecy used in ::testRollback.
     *
     * @return \Prophecy\Prophecy\ObjectProphecy<\Drupal\migrate\Plugin\MigrateIdMapInterface>
     *   An ID map object prophecy.
     */
    public function getTestRollbackIdMap(array $items, array $source_id_keys, array $destination_id_keys) : ObjectProphecy {
        static::$idMapRecords = array_map(function (array $item) {
            return $item + [
                'source_row_status' => '0',
                'rollback_action' => '0',
                'last_imported' => '0',
                'hash' => '',
            ];
        }, $items);
        $array_iterator = new \ArrayIterator(static::$idMapRecords);
        $id_map = $this->prophesize(MigrateIdMapInterface::class);
        $id_map->setMessage(Argument::cetera())
            ->willReturn(NULL);
        $id_map->rewind()
            ->will(function () use ($array_iterator) {
            $array_iterator->rewind();
        });
        $id_map->valid()
            ->will(function () use ($array_iterator) {
            return $array_iterator->valid();
        });
        $id_map->next()
            ->will(function () use ($array_iterator) {
            $array_iterator->next();
        });
        $id_map->currentDestination()
            ->will(function () use ($array_iterator, $destination_id_keys) {
            $current = $array_iterator->current();
            $destination_values = array_filter($current, function ($key) use ($destination_id_keys) {
                return in_array($key, $destination_id_keys, TRUE);
            }, ARRAY_FILTER_USE_KEY);
            return empty(array_filter($destination_values, 'is_null')) ? array_combine($destination_id_keys, array_values($destination_values)) : NULL;
        });
        $id_map->currentSource()
            ->will(function () use ($array_iterator, $source_id_keys) {
            $current = $array_iterator->current();
            $source_values = array_filter($current, function ($key) use ($source_id_keys) {
                return in_array($key, $source_id_keys, TRUE);
            }, ARRAY_FILTER_USE_KEY);
            return empty(array_filter($source_values, 'is_null')) ? array_combine($source_id_keys, array_values($source_values)) : NULL;
        });
        $id_map->getRowByDestination(Argument::type('array'))
            ->will(function () {
            $destination_ids = func_get_args()[0][0];
            $return = array_reduce(self::$idMapRecords, function (array $carry, array $record) use ($destination_ids) {
                if (array_merge($record, $destination_ids) === $record) {
                    $carry = $record;
                }
                return $carry;
            }, []);
            return $return;
        });
        $id_map->deleteDestination(Argument::type('array'))
            ->will(function () {
            $destination_ids = func_get_args()[0][0];
            $matching_records = array_filter(self::$idMapRecords, function (array $record) use ($destination_ids) {
                return array_merge($record, $destination_ids) === $record;
            });
            foreach (array_keys($matching_records) as $record_key) {
                unset(self::$idMapRecords[$record_key]);
            }
        });
        $id_map->delete(Argument::type('array'))
            ->will(function () {
            $source_ids = func_get_args()[0][0];
            $matching_records = array_filter(self::$idMapRecords, function (array $record) use ($source_ids) {
                return array_merge($record, $source_ids) === $record;
            });
            foreach (array_keys($matching_records) as $record_key) {
                unset(self::$idMapRecords[$record_key]);
            }
        });
        return $id_map;
    }

}