public function MigrationTest::testEnableWithExistingContent in Multiversion 8

File

tests/src/Functional/MigrationTest.php, line 73

Class

MigrationTest

Test the MigrationTest class.

Namespace

Drupal\Tests\multiversion\Functional

Code

public function testEnableWithExistingContent() {
  foreach ($this->entityTypes as $entity_type_id => $values) {
    $storage = \Drupal::entityTypeManager()
      ->getStorage($entity_type_id);
    $count = 2;
    for ($i = 0; $i < $count; $i++) {
      if ($entity_type_id == 'file') {
        $values['filename'] = "test{$i}.txt";
        $values['uri'] = "public://test{$i}.txt";
        $this
          ->assertTrue($values['uri'], t('The test file has been created.'));
        $file = $storage
          ->create($values);
        file_put_contents($file
          ->getFileUri(), 'Hello world!');
        $file
          ->save();
        continue;
      }
      $storage
        ->create($values)
        ->save();
    }
    $count_before[$entity_type_id] = $count;
  }

  // Installing Multiversion will trigger the migration of existing content.
  $this->moduleInstaller
    ->install([
    'multiversion',
  ]);
  $this->multiversionManager = \Drupal::service('multiversion.manager');

  // Check if all updates have been applied.
  $update_manager = \Drupal::service('entity.definition_update_manager');
  $this
    ->assertFalse($update_manager
    ->needsUpdates(), 'All compatible entity types have been updated.');
  $ids_after = [];

  // Now check that the previously created entities still exist, have the
  // right IDs and are multiversion enabled. That means profit. Big profit.
  foreach ($this->entityTypes as $entity_type_id => $values) {
    $manager = \Drupal::entityTypeManager();
    $entity_type = $manager
      ->getDefinition($entity_type_id);
    $storage = $manager
      ->getStorage($entity_type_id);
    $id_key = $entity_type
      ->getKey('id');
    $this
      ->assertTrue($this->multiversionManager
      ->isEnabledEntityType($entity_type), "{$entity_type_id} was enabled for Multiversion.");
    $this
      ->assertTrue($storage instanceof ContentEntityStorageInterface, "{$entity_type_id} got the correct storage handler assigned.");
    $this
      ->assertTrue($storage
      ->getQuery() instanceof QueryInterface, "{$entity_type_id} got the correct query handler assigned.");
    $ids_after[$entity_type_id] = $storage
      ->getQuery()
      ->execute();
    $this
      ->assertEqual($count_before[$entity_type_id], count($ids_after[$entity_type_id]), "All {$entity_type_id}s were migrated.");
    foreach ($ids_after[$entity_type_id] as $revision_id => $entity_id) {
      $rev = (int) $storage
        ->getQuery()
        ->condition($id_key, $entity_id)
        ->condition('_rev', 'NULL', '<>')
        ->count()
        ->execute();
      $this
        ->assertEqual($rev, 1, "{$entity_type_id} {$entity_id} has a revision hash in database");
      if ($entity_type
        ->get('workspace') !== FALSE) {
        $workspace = (int) $storage
          ->getQuery()
          ->condition($id_key, $entity_id)
          ->condition('workspace', 1)
          ->count()
          ->execute();
        $this
          ->assertEqual($workspace, 1, "{$entity_type_id} {$entity_id} has correct workspace in database");
      }
      $deleted = (int) $storage
        ->getQuery()
        ->condition($id_key, $entity_id)
        ->condition('_deleted', 0)
        ->count()
        ->execute();
      $this
        ->assertEqual($deleted, 1, "{$entity_type_id} {$entity_id} is not marked as deleted in database");
    }
  }

  // Now install a module with an entity type AFTER the migration and assert
  // that is being returned as supported and enabled.
  $this->moduleInstaller
    ->install([
    'taxonomy',
  ]);
  $entity_type = \Drupal::entityTypeManager()
    ->getDefinition('taxonomy_term');
  $this
    ->assertTrue($this->multiversionManager
    ->isEnabledEntityType($entity_type), 'Newly installed entity types got enabled as well.');
  $this
    ->assertFalse($update_manager
    ->needsUpdates(), 'There are not new updates to apply.');
}