View source
<?php
namespace Drupal\Tests\migrate\Unit;
use Drupal\Component\Plugin\Exception\InvalidPluginDefinitionException;
use Drupal\migrate\Plugin\MigrationInterface;
use Drupal\migrate\Plugin\Migration;
use Drupal\migrate\Exception\RequirementsException;
use Drupal\migrate\Plugin\MigrateDestinationInterface;
use Drupal\migrate\Plugin\MigrateSourceInterface;
use Drupal\migrate\Plugin\MigrationPluginManagerInterface;
use Drupal\migrate\Plugin\RequirementsInterface;
use Drupal\Tests\UnitTestCase;
class MigrationTest extends UnitTestCase {
public function testRequirementsForSourcePlugin() {
$migration = new TestMigration();
$source_plugin = $this
->createMock('Drupal\\Tests\\migrate\\Unit\\RequirementsAwareSourceInterface');
$source_plugin
->expects($this
->once())
->method('checkRequirements')
->willThrowException(new RequirementsException('Missing source requirement', [
'key' => 'value',
]));
$destination_plugin = $this
->createMock('Drupal\\Tests\\migrate\\Unit\\RequirementsAwareDestinationInterface');
$migration
->setSourcePlugin($source_plugin);
$migration
->setDestinationPlugin($destination_plugin);
$this
->expectException(RequirementsException::class);
$this
->expectExceptionMessage('Missing source requirement');
$migration
->checkRequirements();
}
public function testRequirementsForDestinationPlugin() {
$migration = new TestMigration();
$source_plugin = $this
->createMock('Drupal\\migrate\\Plugin\\MigrateSourceInterface');
$destination_plugin = $this
->createMock('Drupal\\Tests\\migrate\\Unit\\RequirementsAwareDestinationInterface');
$destination_plugin
->expects($this
->once())
->method('checkRequirements')
->willThrowException(new RequirementsException('Missing destination requirement', [
'key' => 'value',
]));
$migration
->setSourcePlugin($source_plugin);
$migration
->setDestinationPlugin($destination_plugin);
$this
->expectException(RequirementsException::class);
$this
->expectExceptionMessage('Missing destination requirement');
$migration
->checkRequirements();
}
public function testRequirementsForMigrations() {
$migration = new TestMigration();
$source_plugin = $this
->createMock('Drupal\\migrate\\Plugin\\MigrateSourceInterface');
$destination_plugin = $this
->createMock('Drupal\\migrate\\Plugin\\MigrateDestinationInterface');
$migration
->setSourcePlugin($source_plugin);
$migration
->setDestinationPlugin($destination_plugin);
$plugin_manager = $this
->createMock('Drupal\\migrate\\Plugin\\MigrationPluginManagerInterface');
$migration
->setMigrationPluginManager($plugin_manager);
$migration
->setRequirements([
'test_a',
'test_b',
'test_c',
'test_d',
]);
$migration_b = $this
->createMock(MigrationInterface::class);
$migration_c = $this
->createMock(MigrationInterface::class);
$migration_d = $this
->createMock(MigrationInterface::class);
$migration_b
->expects($this
->once())
->method('allRowsProcessed')
->willReturn(TRUE);
$migration_c
->expects($this
->once())
->method('allRowsProcessed')
->willReturn(FALSE);
$migration_d
->expects($this
->once())
->method('allRowsProcessed')
->willReturn(TRUE);
$plugin_manager
->expects($this
->once())
->method('createInstances')
->with([
'test_a',
'test_b',
'test_c',
'test_d',
])
->willReturn([
'test_b' => $migration_b,
'test_c' => $migration_c,
'test_d' => $migration_d,
]);
$this
->expectException(RequirementsException::class);
$this
->expectExceptionMessage('Missing migrations test_a, test_c');
$migration
->checkRequirements();
}
public function testMigrationDependenciesWithValidConfig($source, array $expected_value) {
$migration = new TestMigration();
if (!is_null($source)) {
$migration
->set('migration_dependencies', $source);
}
$this
->assertSame($migration
->getMigrationDependencies(), $expected_value);
}
public function testMigrationDependenciesWithInvalidConfig() {
$migration = new TestMigration();
$plugin_id = 'test_migration';
$migration
->setPluginId($plugin_id);
$migration
->set('migration_dependencies', [
'test_migration_dependency',
]);
$this
->expectException(InvalidPluginDefinitionException::class);
$this
->expectExceptionMessage("Invalid migration dependencies configuration for migration {$plugin_id}");
$migration
->getMigrationDependencies();
}
public function getValidMigrationDependenciesProvider() {
return [
[
'source' => NULL,
'expected_value' => [
'required' => [],
'optional' => [],
],
],
[
'source' => [],
'expected_value' => [
'required' => [],
'optional' => [],
],
],
[
'source' => [
'required' => [
'test_migration',
],
],
'expected_value' => [
'required' => [
'test_migration',
],
'optional' => [],
],
],
[
'source' => [
'optional' => [
'test_migration',
],
],
'expected_value' => [
'optional' => [
'test_migration',
],
'required' => [],
],
],
[
'source' => [
'required' => [
'req_test_migration',
],
'optional' => [
'opt_test_migration',
],
],
'expected_value' => [
'required' => [
'req_test_migration',
],
'optional' => [
'opt_test_migration',
],
],
],
];
}
}
class TestMigration extends Migration {
public function __construct() {
}
public function setPluginId($plugin_id) {
$this->pluginId = $plugin_id;
}
public function setRequirements(array $requirements) {
$this->requirements = $requirements;
}
public function setSourcePlugin(MigrateSourceInterface $source_plugin) {
$this->sourcePlugin = $source_plugin;
}
public function setDestinationPlugin(MigrateDestinationInterface $destination_plugin) {
$this->destinationPlugin = $destination_plugin;
}
public function setMigrationPluginManager(MigrationPluginManagerInterface $plugin_manager) {
$this->migrationPluginManager = $plugin_manager;
}
}
interface RequirementsAwareSourceInterface extends MigrateSourceInterface, RequirementsInterface {
}
interface RequirementsAwareDestinationInterface extends MigrateDestinationInterface, RequirementsInterface {
}