Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\Tests\migrate\Kernel;
Chris@0: 
Chris@0: use Drupal\migrate\Event\MigratePostRowSaveEvent;
Chris@0: use Drupal\migrate\Plugin\MigrationInterface;
Chris@0: use Drupal\migrate\Event\MigrateEvents;
Chris@0: use Drupal\migrate\MigrateExecutable;
Chris@0: use Drupal\KernelTests\KernelTestBase;
Chris@0: 
Chris@0: /**
Chris@0:  * Tests interruptions triggered during migrations.
Chris@0:  *
Chris@0:  * @group migrate
Chris@0:  */
Chris@0: class MigrateInterruptionTest extends KernelTestBase {
Chris@0: 
Chris@0:   /**
Chris@0:    * Modules to enable.
Chris@0:    *
Chris@0:    * @var array
Chris@0:    */
Chris@0:   public static $modules = ['migrate', 'migrate_events_test'];
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function setUp() {
Chris@0:     parent::setUp();
Chris@0:     \Drupal::service('event_dispatcher')->addListener(MigrateEvents::POST_ROW_SAVE,
Chris@0:       [$this, 'postRowSaveEventRecorder']);
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Tests migration interruptions.
Chris@0:    */
Chris@0:   public function testMigrateEvents() {
Chris@0:     // Run a simple little migration, which should trigger one of each event
Chris@0:     // other than map_delete.
Chris@0:     $definition = [
Chris@0:       'migration_tags' => ['Interruption test'],
Chris@0:       'source' => [
Chris@0:         'plugin' => 'embedded_data',
Chris@0:         'data_rows' => [
Chris@0:           ['data' => 'dummy value'],
Chris@0:           ['data' => 'dummy value2'],
Chris@0:         ],
Chris@0:         'ids' => [
Chris@0:           'data' => ['type' => 'string'],
Chris@0:         ],
Chris@0:       ],
Chris@0:       'process' => ['value' => 'data'],
Chris@0:       'destination' => ['plugin' => 'dummy'],
Chris@0:     ];
Chris@0: 
Chris@0:     $migration = \Drupal::service('plugin.manager.migration')->createStubMigration($definition);
Chris@0: 
Chris@14:     $executable = new MigrateExecutable($migration);
Chris@0:     // When the import runs, the first row imported will trigger an
Chris@0:     // interruption.
Chris@0:     $result = $executable->import();
Chris@0: 
Chris@0:     $this->assertEqual($result, MigrationInterface::RESULT_INCOMPLETE);
Chris@0: 
Chris@0:     // The status should have been reset to IDLE.
Chris@0:     $this->assertEqual($migration->getStatus(), MigrationInterface::STATUS_IDLE);
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Reacts to post-row-save event.
Chris@0:    *
Chris@18:    * @param \Drupal\migrate\Event\MigratePostRowSaveEvent $event
Chris@0:    *   The migration event.
Chris@0:    * @param string $name
Chris@0:    *   The event name.
Chris@0:    */
Chris@0:   public function postRowSaveEventRecorder(MigratePostRowSaveEvent $event, $name) {
Chris@0:     $event->getMigration()->interruptMigration(MigrationInterface::RESULT_INCOMPLETE);
Chris@0:   }
Chris@0: 
Chris@0: }