You are here

public function MigrateRollbackTest::testRollback in Drupal 9

Same name and namespace in other branches
  1. 8 core/modules/migrate/tests/src/Kernel/MigrateRollbackTest.php \Drupal\Tests\migrate\Kernel\MigrateRollbackTest::testRollback()

Tests rolling back configuration and content entities.

File

core/modules/migrate/tests/src/Kernel/MigrateRollbackTest.php, line 39

Class

MigrateRollbackTest
Tests rolling back of imports.

Namespace

Drupal\Tests\migrate\Kernel

Code

public function testRollback() {

  // We use vocabularies to demonstrate importing and rolling back
  // configuration entities.
  $vocabulary_data_rows = [
    [
      'id' => '1',
      'name' => 'categories',
      'weight' => '2',
    ],
    [
      'id' => '2',
      'name' => 'tags',
      'weight' => '1',
    ],
  ];
  $ids = [
    'id' => [
      'type' => 'integer',
    ],
  ];
  $definition = [
    'id' => 'vocabularies',
    'migration_tags' => [
      'Import and rollback test',
    ],
    'source' => [
      'plugin' => 'embedded_data',
      'data_rows' => $vocabulary_data_rows,
      'ids' => $ids,
    ],
    'process' => [
      'vid' => 'id',
      'name' => 'name',
      'weight' => 'weight',
    ],
    'destination' => [
      'plugin' => 'entity:taxonomy_vocabulary',
    ],
  ];
  $vocabulary_migration = \Drupal::service('plugin.manager.migration')
    ->createStubMigration($definition);
  $vocabulary_id_map = $vocabulary_migration
    ->getIdMap();
  $this
    ->assertTrue($vocabulary_migration
    ->getDestinationPlugin()
    ->supportsRollback());

  // Import and validate vocabulary config entities were created.
  $vocabulary_executable = new MigrateExecutable($vocabulary_migration, $this);
  $vocabulary_executable
    ->import();
  foreach ($vocabulary_data_rows as $row) {

    /** @var \Drupal\taxonomy\Entity\Vocabulary $vocabulary */
    $vocabulary = Vocabulary::load($row['id']);
    $this
      ->assertNotEmpty($vocabulary);
    $map_row = $vocabulary_id_map
      ->getRowBySource([
      'id' => $row['id'],
    ]);
    $this
      ->assertNotNull($map_row['destid1']);
  }

  // We use taxonomy terms to demonstrate importing and rolling back content
  // entities.
  $term_data_rows = [
    [
      'id' => '1',
      'vocab' => '1',
      'name' => 'music',
    ],
    [
      'id' => '2',
      'vocab' => '2',
      'name' => 'Bach',
    ],
    [
      'id' => '3',
      'vocab' => '2',
      'name' => 'Beethoven',
    ],
  ];
  $ids = [
    'id' => [
      'type' => 'integer',
    ],
  ];
  $definition = [
    'id' => 'terms',
    'migration_tags' => [
      'Import and rollback test',
    ],
    'source' => [
      'plugin' => 'embedded_data',
      'data_rows' => $term_data_rows,
      'ids' => $ids,
    ],
    'process' => [
      'tid' => 'id',
      'vid' => 'vocab',
      'name' => 'name',
    ],
    'destination' => [
      'plugin' => 'entity:taxonomy_term',
    ],
    'migration_dependencies' => [
      'required' => [
        'vocabularies',
      ],
    ],
  ];
  $term_migration = \Drupal::service('plugin.manager.migration')
    ->createStubMigration($definition);
  $term_id_map = $term_migration
    ->getIdMap();
  $this
    ->assertTrue($term_migration
    ->getDestinationPlugin()
    ->supportsRollback());

  // Pre-create a term, to make sure it isn't deleted on rollback.
  $preserved_term_ids[] = 1;
  $new_term = Term::create([
    'tid' => 1,
    'vid' => 1,
    'name' => 'music',
  ]);
  $new_term
    ->save();

  // Import and validate term entities were created.
  $term_executable = new MigrateExecutable($term_migration, $this);
  $term_executable
    ->import();

  // Also explicitly mark one row to be preserved on rollback.
  $preserved_term_ids[] = 2;
  $map_row = $term_id_map
    ->getRowBySource([
    'id' => 2,
  ]);
  $dummy_row = new Row([
    'id' => 2,
  ], $ids);
  $term_id_map
    ->saveIdMapping($dummy_row, [
    $map_row['destid1'],
  ], $map_row['source_row_status'], MigrateIdMapInterface::ROLLBACK_PRESERVE);
  foreach ($term_data_rows as $row) {

    /** @var \Drupal\taxonomy\Entity\Term $term */
    $term = Term::load($row['id']);
    $this
      ->assertNotEmpty($term);
    $map_row = $term_id_map
      ->getRowBySource([
      'id' => $row['id'],
    ]);
    $this
      ->assertNotNull($map_row['destid1']);
  }

  // Add a failed row to test if this can be rolled back without errors.
  $this
    ->mockFailure($term_migration, [
    'id' => '4',
    'vocab' => '2',
    'name' => 'FAIL',
  ]);

  // Rollback and verify the entities are gone.
  $term_executable
    ->rollback();
  foreach ($term_data_rows as $row) {
    $term = Term::load($row['id']);
    if (in_array($row['id'], $preserved_term_ids)) {
      $this
        ->assertNotNull($term);
    }
    else {
      $this
        ->assertNull($term);
    }
    $map_row = $term_id_map
      ->getRowBySource([
      'id' => $row['id'],
    ]);
    $this
      ->assertFalse($map_row);
  }
  $vocabulary_executable
    ->rollback();
  foreach ($vocabulary_data_rows as $row) {
    $term = Vocabulary::load($row['id']);
    $this
      ->assertNull($term);
    $map_row = $vocabulary_id_map
      ->getRowBySource([
      'id' => $row['id'],
    ]);
    $this
      ->assertFalse($map_row);
  }

  // Test that simple configuration is not rollbackable.
  $term_setting_rows = [
    [
      'id' => 1,
      'override_selector' => '0',
      'terms_per_page_admin' => '10',
    ],
  ];
  $ids = [
    'id' => [
      'type' => 'integer',
    ],
  ];
  $definition = [
    'id' => 'taxonomy_settings',
    'migration_tags' => [
      'Import and rollback test',
    ],
    'source' => [
      'plugin' => 'embedded_data',
      'data_rows' => $term_setting_rows,
      'ids' => $ids,
    ],
    'process' => [
      'override_selector' => 'override_selector',
      'terms_per_page_admin' => 'terms_per_page_admin',
    ],
    'destination' => [
      'plugin' => 'config',
      'config_name' => 'taxonomy.settings',
    ],
    'migration_dependencies' => [
      'required' => [
        'vocabularies',
      ],
    ],
  ];
  $settings_migration = \Drupal::service('plugin.manager.migration')
    ->createStubMigration($definition);
  $this
    ->assertFalse($settings_migration
    ->getDestinationPlugin()
    ->supportsRollback());
}