public function UninstallTest::testDisableWithExistingContent in Multiversion 8.2

File

src/Tests/UninstallTest.php, line 75

Class

UninstallTest

Test the UninstallTest class.

Namespace

Drupal\multiversion\Tests

Code

public function testDisableWithExistingContent() {
  $entity_type_manager = $this->container
    ->get('entity_type.manager');
  foreach ($this->entityTypes as $entity_type_id => $values) {
    $storage = $entity_type_manager
      ->getStorage($entity_type_id);
    $count = 2;
    for ($i = 0; $i < $count; $i++) {
      $storage
        ->create($values)
        ->save();
    }
    $count_before[$entity_type_id] = $count;
  }

  /** @var \Drupal\multiversion\MultiversionManagerInterface $manager */
  $manager = $this->container
    ->get('multiversion.manager');

  // Disable entity types.
  $manager
    ->disableEntityTypes();

  // Delete workspace entities before uninstall.
  $storage = $entity_type_manager
    ->getStorage('workspace');
  $entities = $storage
    ->loadMultiple();
  $storage
    ->delete($entities);

  // Uninstall Multiversion.
  $this->container
    ->get('module_installer')
    ->uninstall([
    'multiversion',
  ]);

  /** @var \Drupal\Core\Entity\EntityDefinitionUpdateManagerInterface $update_manager */
  $update_manager = \Drupal::entityDefinitionUpdateManager();

  // The field class for the UUID field that Multiversion provides will now
  // be gone. So we need to apply updates.
  $update_manager
    ->applyUpdates();

  // Check that applying updates worked.
  $this
    ->assertFalse($update_manager
    ->needsUpdates(), 'There are not new updates to apply.');
  $ids_after = [];

  // Now check that the previously created entities still exist, have the
  // right IDs and are multiversion enabled.
  foreach ($this->entityTypes as $entity_type_id => $values) {
    $storage = $entity_type_manager
      ->getStorage($entity_type_id);
    $storage_class = $storage
      ->getEntityType($entity_type_id)
      ->getStorageClass();
    $this
      ->assertFalse(strpos($storage_class, 'Drupal\\multiversion\\Entity\\Storage'), "{$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.");
  }
}