You are here

public function MigrateRollbackTest::testRollback in Zircon Profile 8

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

Tests rolling back configuration and content entities.

File

core/modules/migrate/src/Tests/MigrateRollbackTest.php, line 44
Contains \Drupal\migrate\Tests\MigrateRollbackTest.

Class

MigrateRollbackTest
Tests rolling back of imports.

Namespace

Drupal\migrate\Tests

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',
    ],
  ];
  $config = [
    '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 = Migration::create($config);
  $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 Vocabulary $vocabulary */
    $vocabulary = Vocabulary::load($row['id']);
    $this
      ->assertTrue($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',
    ],
  ];
  $config = [
    '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 = Migration::create($config);
  $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 Term $term */
    $term = Term::load($row['id']);
    $this
      ->assertTrue($term);
    $map_row = $term_id_map
      ->getRowBySource([
      'id' => $row['id'],
    ]);
    $this
      ->assertNotNull($map_row['destid1']);
  }

  // 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',
    ],
  ];
  $config = [
    '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 = Migration::create($config);
  $this
    ->assertFalse($settings_migration
    ->getDestinationPlugin()
    ->supportsRollback());
}