MigrationTest.php in Zircon Profile 8

Contains \Drupal\Tests\migrate\Unit\MigrationTest.

Namespace

Drupal\Tests\migrate\Unit

File

core/modules/migrate/tests/src/Unit/MigrationTest.php

<?php

/**
 * @file
 * Contains \Drupal\Tests\migrate\Unit\MigrationTest.
 */
namespace Drupal\Tests\migrate\Unit;

use Drupal\Core\Entity\EntityManagerInterface;
use Drupal\migrate\Entity\Migration;
use Drupal\migrate\Exception\RequirementsException;
use Drupal\migrate\Plugin\MigrateDestinationInterface;
use Drupal\migrate\Plugin\MigrateSourceInterface;
use Drupal\migrate\Plugin\RequirementsInterface;
use Drupal\Tests\UnitTestCase;

/**
 * @coversDefaultClass \Drupal\migrate\Entity\Migration
 * @group Migration
 */
class MigrationTest extends UnitTestCase {

  /**
   * Tests checking requirements for source plugins.
   *
   * @covers ::checkRequirements
   *
   * @expectedException \Drupal\migrate\Exception\RequirementsException
   * @expectedExceptionMessage Missing source requirement
   */
  public function testRequirementsForSourcePlugin() {
    $migration = new TestMigration();
    $source_plugin = $this
      ->getMock('Drupal\\Tests\\migrate\\Unit\\RequirementsAwareSourceInterface');
    $source_plugin
      ->expects($this
      ->once())
      ->method('checkRequirements')
      ->willThrowException(new RequirementsException('Missing source requirement', [
      'key' => 'value',
    ]));
    $destination_plugin = $this
      ->getMock('Drupal\\Tests\\migrate\\Unit\\RequirementsAwareDestinationInterface');
    $migration
      ->setSourcePlugin($source_plugin);
    $migration
      ->setDestinationPlugin($destination_plugin);
    $migration
      ->checkRequirements();
  }

  /**
   * Tests checking requirements for destination plugins.
   *
   * @covers ::checkRequirements
   *
   * @expectedException \Drupal\migrate\Exception\RequirementsException
   * @expectedExceptionMessage Missing destination requirement
   */
  public function testRequirementsForDestinationPlugin() {
    $migration = new TestMigration();
    $source_plugin = $this
      ->getMock('Drupal\\migrate\\Plugin\\MigrateSourceInterface');
    $destination_plugin = $this
      ->getMock('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);
    $migration
      ->checkRequirements();
  }

  /**
   * Tests checking requirements for destination plugins.
   *
   * @covers ::checkRequirements
   *
   * @expectedException \Drupal\migrate\Exception\RequirementsException
   * @expectedExceptionMessage Missing migrations test_a, test_c
   */
  public function testRequirementsForMigrations() {
    $migration = new TestMigration();

    // Setup source and destination plugins without any requirements.
    $source_plugin = $this
      ->getMock('Drupal\\migrate\\Plugin\\MigrateSourceInterface');
    $destination_plugin = $this
      ->getMock('Drupal\\migrate\\Plugin\\MigrateDestinationInterface');
    $migration
      ->setSourcePlugin($source_plugin);
    $migration
      ->setDestinationPlugin($destination_plugin);
    $entity_manager = $this
      ->getMock('Drupal\\Core\\Entity\\EntityManagerInterface');
    $migration
      ->setEntityManager($entity_manager);

    // We setup the requirements that test_a doesn't exist and test_c is not
    // completed yet.
    $migration
      ->setRequirements([
      'test_a',
      'test_b',
      'test_c',
      'test_d',
    ]);
    $migration_b = $this
      ->getMock('Drupal\\migrate\\Entity\\MigrationInterface');
    $migration_c = $this
      ->getMock('Drupal\\migrate\\Entity\\MigrationInterface');
    $migration_d = $this
      ->getMock('Drupal\\migrate\\Entity\\MigrationInterface');
    $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);
    $migration_storage = $this
      ->getMock('Drupal\\Core\\Entity\\EntityStorageInterface');
    $migration_storage
      ->expects($this
      ->once())
      ->method('loadMultiple')
      ->with([
      'test_a',
      'test_b',
      'test_c',
      'test_d',
    ])
      ->willReturn([
      'test_b' => $migration_b,
      'test_c' => $migration_c,
      'test_d' => $migration_d,
    ]);
    $entity_manager
      ->expects($this
      ->once())
      ->method('getStorage')
      ->with('migration')
      ->willReturn($migration_storage);
    $migration
      ->checkRequirements();
  }

}
class TestMigration extends Migration {
  public function __construct() {
  }
  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 setEntityManager(EntityManagerInterface $entity_manager) {
    $this->entityManager = $entity_manager;
  }

}
interface RequirementsAwareSourceInterface extends MigrateSourceInterface, RequirementsInterface {

}
interface RequirementsAwareDestinationInterface extends MigrateDestinationInterface, RequirementsInterface {

}

Classes

Name	Description
MigrationTest	@coversDefaultClass \Drupal\migrate\Entity\Migration @group Migration
TestMigration

Interfaces

Name	Description
RequirementsAwareDestinationInterface
RequirementsAwareSourceInterface